Zhang Xinwen, Feng Quan, Zhu Dongqin, Liang Xue, Zhang Jianhua
School of Mechanical and Electrical Engineering, Gansu Agricultural University, Lanzhou, China.
Agricultural Information Institute, Chinese Academy of Agricultural Sciences, Beijing, China.
Front Plant Sci. 2024 Sep 26;15:1433543. doi: 10.3389/fpls.2024.1433543. eCollection 2024.
Deep networks play a crucial role in the recognition of agricultural diseases. However, these networks often come with numerous parameters and large sizes, posing a challenge for direct deployment on resource-limited edge computing devices for plant protection robots. To tackle this challenge for recognizing cotton diseases on the edge device, we adopt knowledge distillation to compress the big networks, aiming to reduce the number of parameters and the computational complexity of the networks. In order to get excellent performance, we conduct combined comparison experiments from three aspects: teacher network, student network and distillation algorithm. The teacher networks contain three classical convolutional neural networks, while the student networks include six lightweight networks in two categories of homogeneous and heterogeneous structures. In addition, we investigate nine distillation algorithms using spot-adaptive strategy. The results demonstrate that the combination of DenseNet40 as the teacher and ShuffleNetV2 as the student show best performance when using NST algorithm, yielding a recognition accuracy of 90.59% and reducing from 0.29 G to 0.045 G. The proposed method can facilitate the lightweighting of the model for recognizing cotton diseases while maintaining high recognition accuracy and offer a practical solution for deploying deep models on edge computing devices.
深度网络在农业病害识别中起着至关重要的作用。然而,这些网络通常具有大量参数和较大规模,这给直接部署在资源有限的用于植保机器人的边缘计算设备上带来了挑战。为了应对在边缘设备上识别棉花病害这一挑战,我们采用知识蒸馏来压缩大型网络,旨在减少网络的参数数量和计算复杂度。为了获得优异的性能,我们从教师网络、学生网络和蒸馏算法三个方面进行了综合比较实验。教师网络包含三个经典卷积神经网络,而学生网络包括两类同构和异构结构中的六个轻量级网络。此外,我们使用点自适应策略研究了九种蒸馏算法。结果表明,当使用NST算法时,以DenseNet40作为教师网络和以ShuffleNetV2作为学生网络的组合表现最佳,识别准确率达到90.59%,并且参数从0.29 G减少到0.045 G。所提出的方法可以在保持高识别准确率的同时促进棉花病害识别模型的轻量化,并为在边缘计算设备上部署深度模型提供了一种实用的解决方案。