浙江农业学报 ›› 2023, Vol. 35 ›› Issue (9): 2250-2264.DOI: 10.3969/j.issn.1004-1524.20221193

• 生物系统工程 • 上一篇    下一篇

基于知识蒸馏和模型剪枝的轻量化模型植物病害识别

刘媛媛(), 王定坤, 邬雷, 黄德昌, 朱路()   

  1. 华东交通大学 信息工程学院,江西 南昌 330013
  • 收稿日期:2022-08-14 出版日期:2023-09-25 发布日期:2023-10-09
  • 作者简介:刘媛媛(1978—),女,江西永新人,硕士,副教授,研究方向为农业物联网与机器学习。E-mail:lyy.78@163.com
  • 通讯作者: 朱路,E-mail:lzhu@ecjtu.edu.cn
  • 基金资助:
    国家自然科学基金(61967007);国家自然科学基金(61963016);江西省重点研发计划重点项目(20201BBF61012)

A light-weight model for plant disease identification based on model pruning and knowledge distillation

LIU Yuanyuan(), WANG Dingkun, WU Lei, HUANG Dechang, ZHU Lu()   

  1. School of Information Engineering, East China Jiaotong University, Nanchang 330013, China
  • Received:2022-08-14 Online:2023-09-25 Published:2023-10-09

摘要:

深度学习为植物病害识别提供了新方法,但是目前大多数深度学习模型的参数众多,难以在存储和计算资源受限的智能手机或嵌入式传感器节点等边缘设备上使用。为此,以植物叶片为研究对象,基于知识蒸馏和模型剪枝方法开展基于轻量化模型的植物病害识别研究。首先,改进ResNet模型,在知识蒸馏中引入一个或多个助教网络训练模型;然后,经过稀疏化训练后,利用模型剪枝获得轻量化的学生网络模型;接着,使用助教网络和学习率倒带重训练该学生网络模型,在减小模型规模的同时保证模型的性能。结果表明:在包含14种植物共38个类别的数据集上,将模型剪枝90%后,模型准确率为97.78%,比原模型提高1.49百分点;在包含5个类别苹果叶的数据集上,将模型剪枝70%后,模型准确率为91.94%,比原模型提高4.85百分点。提出的轻量化模型能够移植在Android平台上并有效运行,可为嵌入式终端精准识别植物病害提供新方案。

关键词: 病害识别, 模型剪枝, 知识蒸馏, 学习率倒带, 残差网络

Abstract:

The emergence of deep learning has provided a new method for plant disease identification, but the current deep learning models have many parameters, which are difficult to use on edge devices such as smartphones or embedded sensor nodes with limited storage and computing resources. In the present study, plant leaves are taken as the research objects, and the methods based on knowledge distillation and model pruning are used to construct a light-weight model for plant disease identification. By improving the ResNet model, one or more teaching assistant network training models are introduced into the knowledge distillation. After sparse training, a light-weight student network model is obtained by model pruning; and the student network is retrained by using the teaching assistant network and learning rate rewinding, which can reduce the size of the model and effectively ensure the performance of the model. The experimental results show that, on a dataset including 38 categories of 14 plants, after pruning the model by 90%, the accuracy of the model is 97.78%, with an increase of 1.49 percentage points over the original model. On the dataset including 5 categories of apple leaves, after pruning the model by 70%, the accuracy of the model is 91.94%, which is 4.85 percentage points higher than the original model. The proposed light-weight model can be transplanted on Android platform and run effectively, which provides a new solution for the embedded terminal to accurately identify plant diseases.

Key words: disease identification, model pruning, knowledge distillation, learning rate rewinding, residual network

中图分类号: