Cotton disease identification method based on pruning

文献类型： 外文期刊

第一作者： Zhu, Dongqin

作者： Zhu, Dongqin;Feng, Quan;Yang, Wanxia;Zhang, Jianhua;Zhang, Jianhua

作者机构：

关键词： convolutional neural network; pruning; cotton diseases; transfer learning; compact model

期刊名称：FRONTIERS IN PLANT SCIENCE （ 影响因子：6.627； 五年影响因子：7.255 ）

ISSN： 1664-462X

年卷期： 2022 年 13 卷

页码：

收录情况： SCI

摘要： Deep convolutional neural networks (DCNN) have shown promising performance in plant disease recognition. However, these networks cannot be deployed on resource-limited smart devices due to their vast parameters and computations. To address the issue of deployability when developing cotton disease identification applications for mobile/smart devices, we compress the disease recognition models employing the pruning algorithm. The algorithm uses the gamma coefficient in the Batch Normalization layer to prune the channels to realize the compression of DCNN. To further improve the accuracy of the model, we suggest two strategies in combination with transfer learning: compression after transfer learning or transfer learning after compression. In our experiments, the source dataset is famous PlantVillage while the target dataset is the cotton disease image set which contains images collected from the Internet and taken from the fields. We select VGG16, ResNet164 and DenseNet40 as compressed models for comparison. The experimental results show that transfer learning after compression overall surpass its counterpart. When compression rate is set to 80% the accuracies of compressed version of VGG16, ResNet164 and DenseNet40 are 90.77%, 96.31% and 97.23%, respectively, and the parameters are only 0.30M, 0.43M and 0.26M, respectively. Among the compressed models, DenseNet40 has the highest accuracy and the smallest parameters. The best model (DenseNet40-80%-T) is pruned 75.70% of the parameters and cut off 65.52% of the computations, with the model size being only 2.2 MB. Compared with the version of compression after transfer learning, the accuracy of the model is improved by 0.74%. We further develop a cotton disease recognition APP on the Android platform based on the model and on the test phone, the average time to identify a single image is just 87ms.

分类号：

  • 相关文献
作者其他论文 更多>>

微信小程序