北方农业学报 ›› 2025, Vol. 53 ›› Issue (5): 117-126.doi: 10.12190/j.issn.2096-1197.2025.05.12

• 农业信息技术 • 上一篇    下一篇

基于知识蒸馏的轻量级马铃薯病害识别神经网络研究

侯丁一1, 张鹏2, 孙梦媛2, 张洋1, 刘瑞芳1, 李利平1, 侯建伟2   

  1. 1.呼和浩特市农牧技术推广中心,内蒙古 呼和浩特 010010;
    2.内蒙古自治区农牧业科学院,内蒙古 呼和浩特 010031
  • 收稿日期:2025-04-14 出版日期:2025-10-20 发布日期:2026-01-20
  • 通讯作者: 侯建伟(1986—),男,教授,博士,主要从事土壤肥力与植物营养方面的研究工作。
  • 作者简介:侯丁一(1994—),女,农艺师,硕士,主要从事农业技术推广方面的研究工作。
  • 基金资助:
    农牧业科技转移转化资金项目(2024TG01-4); 农田智慧施肥项目(05)

Research on a lightweight potato disease recognition neural network based on knowledge distillation

HOU Dingyi1, ZHANG Peng2, SUN Mengyuan2, ZHANG Yang1, LIU Ruifang1, LI Liping1, HOU Jianwei2   

  1. 1. Hohhot Agricultural and Animal Husbandry Technology Promotion Center,Hohhot 010010,China;
    2. Inner Mongolia Academy of Agricultural & Animal Husbandry Sciences,Hohhot 010031,China
  • Received:2025-04-14 Online:2025-10-20 Published:2026-01-20

摘要: 【目的】 探究基于知识蒸馏的轻量级神经网络在马铃薯病害识别中的应用效果,为智慧农业中的病害防控提供参考。【方法】 使用PlantVillage公开数据集和自主采集数据构建包含健康、早疫病和晚疫病3类样本的马铃薯病害数据集(12 000张),按7∶1∶2划分为训练集、验证集和测试集。首先训练EfficientNet-B3作为教师网络,然后构建双路知识迁移框架,并采用MobileNetV3-Small作为学生网络。通过特征选择机制提取关键判别特征,采用自适应知识蒸馏策略在特征层和预测层同时进行知识迁移,并根据训练状态动态调整损失权重。在教师网络指导下训练学生网络,得到兼具轻量化和高识别性能的改进模型,并与未经蒸馏的MobileNetV3-Small基线模型进行对比。【结果】 改进模型在测试集上对马铃薯病害的检测达到97.80%的准确率,较基线模型提升3.20个百分点,接近教师网络98.50%的性能水平。模型参数量仅为1.60 M,浮点运算次数为62.03 M,分别比教师网络减少85.05%和93.75%。Grad-CAM可视化分析证实,模型能够精确定位马铃薯早疫病的同心轮纹和晚疫病的水渍状病变等关键诊断特征,热力图与病变区域高度吻合。【结论】 基于自适应知识蒸馏的轻量级马铃薯病害识别模型兼具高准确率和低计算复杂度,能有效识别马铃薯早疫病和晚疫病特征,适用于资源受限环境下的田间场景。

关键词: 轻量级神经网络, 马铃薯病害识别, 知识蒸馏, 深度学习, 智慧农业

Abstract: 【Objective】 To explore the application effectiveness of a lightweight deep neural network based on knowledge distillation for potato disease recognition,providing a reference for disease prevention and control in smart agriculture.【Methods】 A potato disease dataset containing 12 000 images across three categories(healthy,early blight,and late blight) was constructed using the public PlantVillage dataset and self-collected data. This dataset was divided into training,validation,and test sets in a ratio of 7∶1∶2. First,EfficientNet-B3 was trained as the teacher network. Subsequently,a dual-path knowledge transfer framework was constructed,with MobileNetV3-Small serving as the student network. Key discriminative features were extracted through a feature selection mechanism,and an adaptive knowledge distillation strategy was employed for simultaneous knowledge transfer at the feature and prediction layers. The loss weights were dynamically adjusted based on the training state. The student network was trained under the guidance of the teacher network to obtain an improved model that achieves both lightweight architecture and high recognition performance,and was compared with the un-distilled MobileNetV3-Small baseline model.【Results】 The improved model achieved an accuracy of 97.80% for potato disease detection on the test set,representing a 3.20 percentage point improvement over the baseline model and closely approaching the teacher network′s performance level of 98.50%. The model′s parameter count was only 1.60 M and its floating-point operations(FLOPs)were 62.03 M,which were 85.05% and 93.75% less than the teacher network,respectively. Grad-CAM visualization analysis confirmed that the model precisely locates key diagnostic features,such as the concentric rings of potato early blight and the water-soaked lesions of potato late blight,with the heatmap highly matching with the diseased regions.【Conclusion】 The lightweight potato disease recognition model based on adaptive knowledge distillation achieves both high accuracy and low computational complexity. It can effectively recognize the features of potato early blight and late blight,making it suitable for resource-constrained field environments.

Key words: Lightweight deep neural network, Potato disease recognition, Knowledge distillation, Deep learning, Smart agriculture

中图分类号: 

  • S435.32