混合卷积与Transformer,精准识别木薯叶病。
DenseSwinV2: Channel Attentive Dual Branch CNN Transformer Learning for Cassava Leaf Disease Classification
- 双分支结构融合局部细节与全局上下文特征。
- 准确率达98.02%,F1分数97.81%,优于主流模型。
- 适合田间复杂环境下病害诊断,抗遮挡和噪声。
本文提出一种新型混合密集SwinV2架构,采用双分支框架,同时利用密集连接卷积网络的局部特征与定制化Swin Transformer V2的层次化全局表示进行木薯病害分类。DenseNet分支通过高分辨率特征保留细粒度结构线索,并促进有效梯度传播;定制SwinV2则通过移位窗口自注意力机制捕捉长程依赖关系,有助于区分视觉相似的病斑。每个分支独立引入注意力通道压缩模块,强化疾病相关响应并抑制冗余或背景驱动激活。最终,将判别性通道融合,生成由密集局部与全局相关增强特征图构成的精细化表征。该模型在包含31000张图像的公开木薯叶病数据集上测试,涵盖褐条纹、花叶、绿脉、细菌性萎蔫及健康叶片五类病害,实现98.02%的分类准确率与97.81%的F1分数,显著超越现有卷积与Transformer模型。结果表明,该方法在田间病害诊断中具备鲁棒性与实用性,可应对遮挡、噪声与复杂背景等现实挑战。
原文摘要 · Abstract (English)
This work presents a new Hybrid Dense SwinV2, a two-branch framework that jointly leverages densely connected convolutional features and hierarchical customized Swin Transformer V2 (SwinV2) representations for cassava disease classification. The proposed framework captures high resolution local features through its DenseNet branch, preserving the fine structural cues and also allowing for effective gradient flow. Concurrently, the customized SwinV2 models global contextual dependencies through the idea of shifted-window self attention, which enables the capture of long range interactions critical in distinguishing between visually similar lesions. Moreover, an attention channel-squeeze module is employed for each CNN Transformer stream independently to emphasize discriminative disease related responses and suppress redundant or background driven activations. Finally, these discriminative channels are fused to achieve refined representations from the dense local and SwinV2 global correlated strengthened feature maps, respectively. The proposed Dense SwinV2 utilized a public cassava leaf disease dataset of 31000 images, comprised of five diseases, including brown streak, mosaic, green mottle, bacterial blight, and normal leaf conditions. The proposed Dense SwinV2 demonstrates a significant classification accuracy of 98.02 percent with an F1 score of 97.81 percent, outperforming well-established convolutional and transformer models. These results underline the fact that Hybrid Dense SwinV2 offers robustness and practicality in the field level diagnosis of cassava disease and real world challenges related to occlusion, noise, and complex backgrounds.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。