arXiv:2412.07182cs.CVcs.AI2024-12被引 6

用轻量模型提升水稻病叶识别准确率,适合手机端部署。

An Enhancement of CNN Algorithm for Rice Leaf Disease Image Classification in Mobile Applications

  • 结合卷积与视觉变换器,用可分离自注意力提升特征捕捉能力。
  • 在扩展数据集上达到99.6%准确率,较基线提升22.12%。
  • 参数量减少92.5%,适合移动端实时应用,适合农业智能检测。

本研究针对传统基于卷积神经网络(CNN)的水稻叶病图像分类算法进行改进。采用MobileViTV2_050与ImageNet-1k预训练权重进行迁移学习,该模型通过可分离自注意力机制融合了CNN的局部特征提取与视觉变换器的全局上下文感知能力。首个增强模型MobileViTV2_050-A在基准数据集上分类准确率提升15.66%,达93.14%;第二个增强模型MobileViTV2_050-B在更广义水稻叶病数据集上进一步提升22.12%,测试准确率达99.6%。此外,MobileViTV2-A在四个水稻病类上取得93%的F1-score,ROC曲线范围为87%至97%。在资源消耗方面,相比基线CNN模型(1400万参数),参数量最多降低92.50%,降至110万。结果表明,MobileViTV2_050不仅通过可分离自注意力提升计算效率,还强化了全局上下文学习,为移动部署提供轻量且鲁棒的解决方案,推动精准农业中模型的可解释性与实用性。

原文摘要 · Abstract (English)

This study focuses on enhancing rice leaf disease image classification algorithms, which have traditionally relied on Convolutional Neural Network (CNN) models. We employed transfer learning with MobileViTV2_050 using ImageNet-1k weights, a lightweight model that integrates CNN's local feature extraction with Vision Transformers' global context learning through a separable self-attention mechanism. Our approach resulted in a significant 15.66% improvement in classification accuracy for MobileViTV2_050-A, our first enhanced model trained on the baseline dataset, achieving 93.14%. Furthermore, MobileViTV2_050-B, our second enhanced model trained on a broader rice leaf dataset, demonstrated a 22.12% improvement, reaching 99.6% test accuracy. Additionally, MobileViTV2-A attained an F1-score of 93% across four rice labels and a Receiver Operating Characteristic (ROC) curve ranging from 87% to 97%. In terms of resource consumption, our enhanced models reduced the total parameters of the baseline CNN model by up to 92.50%, from 14 million to 1.1 million. These results indicate that MobileViTV2_050 not only improves computational efficiency through its separable self-attention mechanism but also enhances global context learning. Consequently, it offers a lightweight and robust solution suitable for mobile deployment, advancing the interpretability and practicality of models in precision agriculture.

图像分类轻量模型农业AI移动端

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。