LLM2D
针对移动应用中稻叶疾病图像分类的CNN算法增强
An Enhancement of CNN Algorithm for Rice Leaf Disease Image Classification in Mobile Applications
作者: Kayne Uriel K. Rodrigo, Jerriane Hillary Heart S. Marcial, Samuel C. Brillo, Khatalyn E. Mata, Jonathan C. Morano
发布日期: 2/18/2025
arXiv ID: oai:arXiv.org:2412.07182v2

摘要

arXiv:2412.07182v2 宣布类型: replace-cross 摘要:本研究致力于改进传统的基于卷积神经网络(CNN)模型的水稻叶片疾病图像分类算法。我们使用了MobileViTV2_050模型进行迁移学习,该模型通过可分离自注意机制整合了CNN的局部特征提取和视觉变换器的全局上下文学习,使用的是ImageNet-1k权重。我们的方法使得MobileViTV2_050-A,基于基线数据集的第一个增强模型,分类准确率提高了15.66%,达到了93.14%。此外,我们的第二个增强模型MobileViTV2_050-B,在更广泛的水稻叶片数据集上训练,展示了22.12%的改进,测试准确率达到99.6%。此外,MobileViTV2-A在四个水稻标签上的F1分数达到了93%,并在接收到的操作特征(ROC)曲线下面积范围为87%-97%之间。在资源消耗方面,我们的增强模型将基线CNN模型的总参数量最多减少了92.50%,从1400万减少到110万。这些结果表明,MobileViTV2_050不仅通过其可分离自注意机制提高了计算效率,还增强了全局上下文学习。因此,它提供了一个轻量级且稳健的解决方案,适合移动部署,提升了这些模型在精准农业中的可解释性和实用性。