An Optimal Framework for Rice Leaf Disease Prediction Using Mixed Precision and Gradient-Weighted Class Activation Mapping
摘要
In this work, we offered ideas for an effective and generalizable framework, the mixed precision-based EfficientNet-B0 (MPEfficientNet-B0) model, to classify rice leaf diseases using the mixed precision technique in the base model EfficientNet-B0. The mixed precision training helps boost computational speed while conserving resources. Moreover, the proposed framework uses focal loss to tackle the issue of class disparity, leading to enhanced predictive performance. The dataset used to test the efficacy of this model is the rice leaf disease. The experimental study consists of state-of-the-art models such as XceptionNet, VGG16, ResNet50, InceptionV3, CNN-Based, SVM, Inception-ResNet-V2, ADSNN-BO, EfficientNet-B0. The proposed framework shows the best testing accuracy of 98.86% and low computational cost (measured in terms of Floating Point Operations per Second (FLOPs)) of 27.02M FLOPs. Furthermore, the heatmaps are used to present the attention map using Grad-CAM (Gradient-weighted Class Activation Mapping) showcasing how a model makes decisions, and which features of input are important while prediction.