Improved ShuffleNet V2 for Lightweight Crop Disease Identification

doi:10.3778/j.issn.1002-8331.2111-0457

Abstract

Abstract: Aiming at the problems of current crop disease deep learning identification methods are more complex, poor adaptability for deployment on edge mobile devices with limited computing resources and poor real-time accurate identification, a lightweight crop disease identification method based on improved ShuffleNet V2 is proposed. Based on ShuffleNet V2 Unit, the efficient channel attention（ECA） attention module is introduced, using H-Swish activation function inorder to reduce ShuffleNet V2 Unit numbers in each Stage module of network structure, and using a lightweight network design component with depthwise separable convolution. Finally, experiments are conducted on the PlantVillage disease dataset. Experimental results show that model parameter is 2.95×105, computational volume is 3.388×107（FLOPs） and 6.674×107（MAdd）, and crop disease identification average accuracy reaches 99.24%, provides a reference for the deployment of crop diseases identification methods on resource-constrained devices such as mobile terminals.

Key words: crop disease identification, ShuffleNet V2, lightweight, efficient channel attention（ECA） attention module

摘要： 针对目前有关深度学习的农作物病害识别方法中存在模型较为复杂，部署在计算资源有限的边缘设备和移动终端上适应性不强，实时准确识别作物病害较差的问题，提出一种改进ShuffleNet V2的轻量级农作物病害识别方法。以ShuffleNet V2单元为基础，引入ECA（efficient channel attention）注意力模块，使用H-Swish激活函数以便减少网络结构每个Stage模块中ShuffleNet V2单元使用个数，使用轻量化网络设计组件深度可分离卷积。在PlantVillage病害数据集上进行实验。结果表明，模型的参数量约为2.95×105，计算量为3.388×107（FLOPs）和6.674×107（MAdd），病害识别平均准确率达到了99.24%，为农作物病害识别方法在移动终端等资源受限设备上部署应用提供参考。

关键词: 农作物病害识别, ShuffleNet V2, 轻量级, ECA注意力模块

LI Hao, QIU Weigen, ZHANG Lichen. Improved ShuffleNet V2 for Lightweight Crop Disease Identification[J]. Computer Engineering and Applications, 2022, 58(12): 260-268.

李好, 邱卫根, 张立臣. 改进ShuffleNet V2的轻量级农作物病害识别方法[J]. 计算机工程与应用, 2022, 58(12): 260-268.

References

[1] 翟肇裕，曹益飞，徐焕良，等.农作物病虫害识别关键技术研究综述[J].农业机械学报，2021，52（7）：1-18.
ZHAI Z Y，CAO Y F，XU H L，et al.Review of key techniques for crop disease and pest detection[J].Transactions of the Chinese Society for Agricultural Machinery，2021，52（7）：1-18.
[2] LI L，ZHANG S，WANG B.Plant disease detection and classification by deep learning—a review[J].IEEE Access，2021，9：56683-56698.
[3] LEE S H，GOAU H，BONNET P，et al.New perspectives on plant disease characterization based on deep learning[J].Computers and Electronics in Agriculture，2020，170：1-12.
[4] THANGARAJ R，ANANDAMURUGAN S，PANDIYAN P，et al.Artificial intelligence in tomato leaf disease detection：a comprehensive review and discussion[J].Journal of Plant Diseases and Protection，2021：1-20.
[5] AGARWAL D，CHAWLA M，TIWARI N.Plant leaf disease classification using deep learning：a survey[C]//2021 Third International Conference on Inventive Research in Computing Applications（ICIRCA），2021：643-650.
[6] MOHANTY S P，HUGHES D P，SALATHé M.Using deep learning for image-based plant disease detection[J].Frontiers in Plant Science，2016，7：1-10.
[7] KRIZHEVSKY A，SUTSKEVER I，HINTON G E.Imagenet classification with deep convolutional neural networks[J].Advances in Neural Information Processing Systems，2012，25：1097-1105.
[8] SZEGEDY C，LIU W，JIA Y，et al.Going deeper with convolutions[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2015：1-9.
[9] TOO E C，YUJIAN L，NJUKI S，et al.A comparative study of fine-tuning deep learning models for plant disease identification[J].Computers and Electronics in Agriculture，2019，161：272-279.
[10] SIMONYAN K，ZISSERMAN A.Very deep convolutional networks for large-scale image recognition[J].arXiv：1409.
1556，2014.
[11] SZEGEDY C，IOFFE S，VANHOUCKE V，et al.Inception-v4，inception-resnet and the impact of residual connections on learning[C]//Thirty-First AAAI Conference on Artificial Intelligence，2017：4278-4284.
[12] HE K，ZHANG X，REN S，et al.Deep residual learning for image recognition[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2016：770-778.
[13] HUANG G，LIU Z，VAN DER MAATEN L，et al.Densely connected convolutional networks[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2017：4700-4708.
[14] HUGHES D，SALATHE M.An open access repository of images on plant health to enable the development of mobiledisease diagnostics[J].arXiv：1511.08060，2015.
[15] 葛道辉，李洪升，张亮，等.轻量级神经网络架构综述[J].软件学报，2020，31（9）：2627-2653.
GE D H，LI H S，ZHANG L，et al.Survey of lightweight neural network[J].Journal of Software，2020，31（9）：2627-2653.
[16] 马金林，张裕，马自萍，等.轻量化神经网络卷积设计研究进展[J].计算机科学与探索，2022，16（3）：512-528.
MA J L，ZHANG Y，MA Z P，et al.Research progress of lightweight neural network convolution design[J].Journal of Frontiers of Computer Science and Technology，2022，16（3）：512-528.
[17] CHOLLET F.Xception：deep learning with depthwise separable convolutions[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2017：1251-1258.
[18] HOWARD A G，ZHU M，CHEN B，et al.MobileNets：efficient convolutional neural networks for mobile vision applications[J].arXiv：1704.04861，2017.
[19] HOWARD A，SANDLER M，CHU G，et al.Searching for mobilenetv3[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision，2019：1314-1324.
[20] ZHANG X，ZHOU X，LIN M，et al.Shufflenet：an extremely efficient convolutional neural network for mobile devices[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2018：6848-6856.
[21] MA N，ZHANG X，ZHENG H T，et al.Shufflenet v2：practical guidelinesfor efficient cnn architecture design[C]//Proceedings of the European Conference on Computer Vision（ECCV），2018：116-131.
[22] BOUGUETTAYA A，ZARZOUR H，KECHIDA A，et al.Recent advances on UAV and deep learning for early crop diseases identification：a short review[C]//2021 International Conference on Information Technology（ICIT），2021：334-339.
[23] XIE G，WANG J，ZHANG T，et al.Interleaved structured sparse convolutional neural networks[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2018：8847-8856.
[24] WANG Q，WU B，ZHU P，et al.ECA-Net：efficient channel attention for deep convolutional neural networks[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition（CVPR），2020：11531-11539.
[25] GEETHARAMANI G，PANDIAN A.Identification of plant leaf diseases using a nine-layer deep convolutional neural network[J].Computers & Electrical Engineering，2019，76：323-338.