采用改进的EfficientNet识别苹果叶片病害

    Identifying apple leaf diseases using improved EfficientNet

    • 摘要: 该研究针对当前自然环境下的苹果叶片病害识别中病害病斑小、空间分布特征不同以及特征相近病害识别困难的问题,设计DEFL (DenseNet121+EfficientNet with focal loss and label smoothing)模型。首先,该模型以并行的EfficientNet-B0网络和DenseNet121网络为特征提取网络,以提升模型特征提取能力,其次引入结合标签平滑策略的焦点损失函数以加强模型对识别困难样本的关注。经测试,所提模型的识别准确率为99.13%,平均精度均值为98.47%。消融试验表明两项改进分别使模型平均精度均值提高了7.99和3.15个百分点。对比试验结果表明,DEFL模型平均精度均值较于ResNet50、Inception V3、ResNeXt模型以及分别融合这3种模型的EfficientNet-B0模型分别高出14.53、13.17、14.61、6.4、7.71以及8.91个百分点,模型规模分别小18.73、7.7、12.2、83.62、69.6以及60.09 MB。Grad-CAM(gradient-weighted class activation mapping)热力图可视化结果表明所提模型重点关注了叶片病变区域。UMAP(uniform manifold approximation and projection)特征降维可视化结果表明所提模型提取的特征更具区分度。实际应用验证取得了97.73%的总体准确率以及95.82%的平均精度均值。综上,该研究提出的DEFL模型能够为苹果病害防治提供有效参考。

       

      Abstract: The traditional manual control measures for apple disease primarily rely on subjective experience, which can lead to arbitrariness and bias. This approach is also prone to pesticide waste and misuse. As apple leaves are one of the high-incidence areas of diseases, achieving natural environment-based identification of apple leaf diseases can provide effective guidance for disease prevention and control. To address the challenge of accurate identification of apple leaf diseases in natural settings, a DenseNet121+EfficientNet model with Focal Loss and Label Smoothing (DEFL) is designed. Given that apple leaf disease lesions are small and exhibit varying spatial characteristics for different diseases, accurately localizing and identifying disease regions is challenging. The DEFL model utilizes two parallel networks: EfficientNet-B0 and DenseNet121 for feature extraction. It combines the semantic and positional information extracted from these networks to enhance the model's fine-grained feature extraction capability. To overcome difficulties in identifying apple leaf samples due to the similarity in features among different diseases, a Focal Loss function, combined with label smoothing strategies, is introduced. This prevents the model from being overly confident about the class of apple leaf samples, focusing on those samples with challenging decision boundaries during training, thereby improving the model's ability to recognize boundary cases and enhancing its robustness and generalization. The DEFL model proposed in this study was tested with 3906 apple leaf disease images in natural scenes. The results show that the overall recognition accuracy of the model proposed in this study is 99.13%, and the mean average accuracy is 98.47%. The results of ablation experiments show that the mean average precision of the model is improved by 7.99% after adding DenseNet121 feature extraction branch to EfficientNet-B0. And, after introducing the focus loss function combined with label smoothing strategy to EfficientNet-B0, the mean average precision of the model is improved by 3.15%. After the combination of the two improvements, the mean average precision of the model increased by 12.29%. Visualization results using Gradient-weighted class activation mapping (Grad-CAM) heatmaps show that the model accurately localizes and focuses on disease regions. The image feature information used by the model for apple leaf disease recognition is reliable. Uniform manifold approximation and projection (UMAP) feature dimension reduction visualization results indicate that the Focal Loss function with label smoothing effectively enhances the model's feature extraction capability, resulting in more distinctive feature information and improved disease recognition. The comparative experimental results show that compared with mainstream recognition models such as ResNet50, Inception V3, and ResNeXt, as well as EfficientNet-B0 models that fuse these three models, the mean average precision of the proposed model has increased by 14.53, 13.17, 14.61, 6.4, 7.71, and 8.91 percentage points, respectively, and the model size has decreased by 18.73, 7.7, 12.2, 83.62, 69.6, and 60.09 MB, respectively. Comparative experiments have shown that the proposed model has the best recognition performance. In practical applications, the DEFL model achieved an overall recognition accuracy of 97.73% and a mean average precision of 95.82% when tested with 1501 apple leaf images collected from apple plantations. In conclusion, the DEFL model proposed in this study enables accurate and rapid identification of apple leaf diseases, providing valuable guidance for apple disease prevention and control.

       

    /

    返回文章
    返回