注意力增强的轻量级卷积网络在植物叶片细粒度病害分类中的创新应用

《Frontiers in Plant Science》:An attention-augmented lightweight convolutional framework for fine-grained plant leaf disease classification

【字体: 时间:2026年02月09日 来源:Frontiers in Plant Science 4.8

编辑推荐:

  本文提出了一种名为ALNet的注意力增强轻量级卷积框架,通过融合Squeeze-and-Excitation(SE)模块、空间注意力机制和残差连接等先进技术,在仅使用0.17M参数的情况下,于葡萄、苹果和樱桃叶片病害分类中分别达到99.78%、99.95%和100%的准确率。该模型具有677.20KB的极小体积和151.98 MFLOPs的低计算复杂度,为边缘设备部署提供了突破性解决方案。

  
引言:农业病理学与轻量化AI的融合挑战
全球粮食安全正持续受到植物病原体的威胁,据联合国粮农组织(FAO)统计,每年因植物病害导致的农作物减产高达20-40%,经济损失超过2200亿美元。传统依赖专家目视检测的方法存在效率低、误判率高等局限性。虽然深度学习方法如卷积神经网络(CNN)和视觉Transformer(ViT)在植物病害分类中表现出色,但参数量庞大的模型(如VGG、ResNet)难以在资源受限的边缘设备上部署。这一矛盾催生了对兼具高精度与轻量化特性的新型网络架构的迫切需求。
ALNet架构设计:多模块协同的轻量化创新
ALNet采用三模块式设计:茎干模块(Stem Block)通过3×3卷积(步长为2)实现特征提取与尺寸缩减,结合批归一化(Batch Normalization)和ReLU激活函数完成初始处理;核心模块(Core Block)包含四个级联的混合块(Hybrid Block),每个混合块创新性地整合了五种经典模型的优势:
  1. 1.
    改进型Fire模块借鉴SqueezeNet的压缩-扩展结构,引入通道洗牌(Channel Shuffle)和深度可分离卷积(Depthwise Separable Convolution)技术,在1×1卷积压缩通道后,通过并行1×1与3×3卷积路径增强特征多样性
  2. 2.
    挤压激励模块(SE Block)通过全局平均池化(Global Average Pooling)生成通道描述符,经两个全连接层和Sigmoid激活函数生成权重向量,实现通道注意力机制
  3. 3.
    空间注意力模块(Spatial Attention)对通道维度执行平均池化与最大池化,经7×7卷积生成空间权重图,突出病害区域空间特征
  4. 4.
    残差路径(Residual Path)通过1×1卷积实现维度匹配,结合批归一化层稳定梯度流动
头部模块(Head Block)采用全局平均池化、Dropout层和全连接层完成分类。该设计使ALNet参数量降至0.17M,模型体积仅677.20 KB,较最轻量基准模型SqueezeNet缩小18倍。
实验设计与数据集验证
研究采用PlantVillage公开数据集,聚焦葡萄叶片病害(健康、黑腐病、ESCA病、叶枯病四类,共9027张图像)、苹果病害(健康、雪松锈病、黑腐病、黑星病四类,9714张图像)及樱桃病害(健康、白粉病二类,1906张图像)。为模拟真实环境,额外构建包含高斯噪声(σ=25)、运动模糊(核尺寸15)和低对比度(α=0.5)的增强数据集(共36108张图像)。实验在Kaggle平台使用P100 GPU,采用五折交叉验证,以Adam优化器(初始学习率0.01)、稀疏分类交叉熵损失函数训练30个epoch。
结果分析:精度与轻量化的双重突破
在葡萄叶片多分类任务中,ALNet达到99.78%准确率,较最佳预训练模型SENet提升0.2%,混淆矩阵显示仅2例黑腐病误判。对于二分类任务,葡萄与樱桃数据集分别实现100%和99.69%准确率。在噪声数据集测试中,ALNet保持98.74%的稳健性能,显著优于SqueezeNet(97.46%)和ShuffleNet(97.21%)。模型单轮训练时间仅14秒,较预训练模型(17-31秒)效率提升显著。
消融实验与统计学验证
通过系统性消融实验发现:移除通道洗牌操作导致准确率下降10.85%,证明该设计对特征交互的关键作用;缺失空间注意力和SE模块分别造成7.70%和4.43%的性能损失;残差路径的移除导致2.99%的精度下降。统计学分析显示,ALNet与ResNet50、SqueezeNet的差异具有显著性(p<0.05),与EfficientNetB0、ShuffleNet和SENet处于统计等效水平。与现有先进技术对比,ALNet在三个数据集上均达到最优性能。
结论与展望:迈向田间部署的轻量化AI
ALNet通过异构模块的协同设计,实现了植物叶片病害分类中精度与效率的最佳平衡。其677.20KB的极小体积和151.98 MFLOPs的低计算需求,为手机、嵌入式设备等边缘端部署提供了可行性。未来研究将聚焦于复杂田间环境的适应性验证,包括多变光照、遮挡物干扰等实际挑战,同时开展在真实边缘硬件上的延迟与能耗评估。通过与农业基地的合作推进,该技术有望成为智慧农业诊断工具的关键核心,实现从实验室原型到田间应用的跨越。
相关新闻
生物通微信公众号
微信
新浪微博

知名企业招聘

热点排行

    今日动态 | 人才市场 | 新技术专栏 | 中国科学人 | 云展台 | BioHot | 云讲堂直播 | 会展中心 | 特价专栏 | 技术快讯 | 免费试用

    版权所有 生物通

    Copyright© eBiotrade.com, All Rights Reserved

    联系信箱:

    粤ICP备09063491号