轻量级模型精准识别7种棉叶病,准确率达98.42%
Improved Cotton Leaf Disease Classification Using Parameter-Efficient Deep Learning Framework
- 仅激活MobileNet部分可训练层,结合迁移学习与数据增强
- 7类病害整体准确率98.42%,单类精确率96%~100%
- 模型轻量高效,适合智能农业实际部署
棉花作物常被称为“白金”,因叶部疾病导致产量严重受损。作为全球主要纤维来源,及时准确识别病害对保障产量和作物健康至关重要。尽管已有深度学习与机器学习方法被尝试,但缺乏参数少、计算高效的轻量模型以满足农业实践需求。为此,本文提出一种创新的深度学习框架,整合MobileNet的部分可训练层、迁移学习、数据增强、学习率衰减策略、模型检查点与早停机制。模型在七类棉叶病分类任务中表现优异,整体准确率达98.42%,各类别精确率介于96%至100%之间。相比现有方法,该模型在准确率与模型复杂度上均有显著提升。即使在更少病种的数据集上测试,现有文献模型也未达到此水平。性能优越且模型轻量,使其具备在智慧农业中实际应用的潜力,有助于推动可持续棉花种植。
原文摘要 · Abstract (English)
Cotton crops, often called "white gold," face significant production challenges, primarily due to various leaf-affecting diseases. As a major global source of fiber, timely and accurate disease identification is crucial to ensure optimal yields and maintain crop health. While deep learning and machine learning techniques have been explored to address this challenge, there remains a gap in developing lightweight models with fewer parameters which could be computationally effective for agricultural practitioners. To address this, we propose an innovative deep learning framework integrating a subset of trainable layers from MobileNet, transfer learning, data augmentation, a learning rate decay schedule, model checkpoints, and early stopping mechanisms. Our model demonstrates exceptional performance, accurately classifying seven cotton disease types with an overall accuracy of 98.42% and class-wise precision ranging from 96% to 100%. This results in significantly enhanced efficiency, surpassing recent approaches in accuracy and model complexity. The existing models in the literature have yet to attain such high accuracy, even when tested on data sets with fewer disease types. The substantial performance improvement, combined with the lightweight nature of the model, makes it practically suitable for real-world applications in smart farming. By offering a high-performing and efficient solution, our framework can potentially address challenges in cotton cultivation, contributing to sustainable agricultural practices.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。