轻量级注意力模型精准识别槟榔叶病,支持可解释性分析。
CBAM Integrated Attention Driven Model For Betel Leaf Diseases Classification With Explainable AI
- 融合CBAM注意力模块的轻量CNN,仅213万参数
- 95.58%准确率,召回率达94%,优于传统模型
- 用Grad-CAM可视化关注区域,结果可解释
槟榔叶是具有经济价值的重要作物,其藤蔓易受多种病害侵袭,植物病害严重威胁粮食安全,及时识别困难且易造成经济损失。人工智能在预测病害、提升产量方面潜力巨大。本文提出一种轻量级CBAM-CNN模型,仅含213万参数(8.13 MB),通过集成CBAM模块增强特征聚焦能力,无需依赖大型预训练网络。模型利用包含10,185张图像的扩充数据集,分为健康叶、叶腐和叶斑三类,实现对细微病害差异的精准区分。测试集上达到97%精确率、94%召回率、95% F1值及95.58%准确率,性能显著优于传统预训练CNN模型。采用Grad-CAM技术可视化模型关注区域,实现可解释性分析。
原文摘要 · Abstract (English)
Betel leaf is an important crop because of its economic advantages and widespread use. Its betel vines are susceptible to a number of illnesses that are commonly referred to as betel leaf disease. Plant diseases are the largest threat to the food supply's security, and they are challenging to identify in time to stop possible financial damage. Interestingly, artificial intelligence can leave a big mark on the betel leaf industry since it helps with output growth by forecasting sickness. This paper presents a lightweight CBAM-CNN model with just 2.13 million parameters (8.13 MB), incorporating CBAM (Convolutional Block Attention Module) to improve feature emphasis without depending on heavy pre-trained networks. The model's capacity to discern minute variations among leaf disease classes is improved by the integrated attention mechanism, which allows it to adaptively focus on significant spatial and channel-wise information. In order to ensure class balance and diversity for efficient model training and validation, this work makes use of an enriched dataset of 10,185 images divided into three categories: Healthy Leaf, Leaf Rot, and Leaf Spot. The proposed model achieved a precision of 97%, recall of 94%, and F1 score of 95%, and 95.58% accuracy on the test set demonstrating strong and balanced classification performance outperforming traditional pre trained CNN models. The model's focus regions were visualized and interpreted using Grad-CAM (Gradient-weighted Class Activation Mapping), an explainable AI technique.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。