改进的FasterViT模型用于类病的诊断
Jiyang Chen1, Shuai Wang1, Jianwen Guo1
1School of Mechanical Engineering, Dongguan University of Technology, Dongguan, 523808, China.
Heliyon
|September 9, 2024
概括
这项研究引入了一种改进的FasterViT模型用于类疾病识别,通过结合CNN和Vision Transformer的优势,在小型数据集上实现快速学习. 改进后的模型提高了准确性,并降低了用于检测植物疾病的培训成本.
科学领域:
- 计算机视觉 计算机视觉
- 植物病理学 植物病理学
- 机器学习 机器学习
背景情况:
- 深度学习模型因环境变化和融合本地/全球图像信息的挑战而难以识别植物疾病,特别是在小数据集中.
- 传统方法往往会导致植物疾病识别任务的性能受损和高培训成本.
研究的目的:
- 提出一个改进的FasterViT模型来准确识别类疾病,重点是从小规模数据集有效学习.
- 提高模型的稳定性,概括性,并降低植物疾病检测方面的培训成本.
主要方法:
- 开发了一种改进的FasterViT模型,一种混合卷积神经网络 (CNN) -视觉转换器 (ViT) 框架,将CNN的本地特征提取与ViT的全球上下文理解相结合.
- 实现跨阶段交替混合和切割用于数据增强,以及三重注意力和自适应AvgPool机制以优化性能和降低计算负载.
- 在定制的现场小型类疾病数据集和PlantVillage数据集上验证了模型.
主要成果:
- 改进的FasterViT模型展示了快速学习能力和有效适应用于检测植物疾病的小样本培训.
- 与基线模型相比,实现了模型准确度的显著改善和培训成本的显著降低.
- 在转移学习场景中表现出色,突出了适应性和广泛适用性.
结论:
- 提议的改进的FasterViT模型有效地解决了植物疾病图像识别的复杂性,特别是对于小规模数据集.
- 这种方法为植物疾病检测提供了一个高效,可扩展和强大的分类系统.
- 这项研究开创了在农业应用中开发先进深度学习解决方案的新范式.


