用新型自监督模型提升植物病害检测效率
ConMamba: Contrastive Vision Mamba for Plant Disease Detection
- 基于双向状态空间模型捕捉图像长程依赖关系
- 双层级对比损失动态调整,优化局部全局特征对齐
- 在3个数据集上超越现有方法,适合农业视觉任务
植物病害检测(PDD)是精准农业的关键。现有深度学习方法常依赖大量标注数据,生成成本高。自监督学习(SSL)可利用海量无标签数据,但多数方法因使用卷积神经网络或基于Transformer的架构导致计算开销大,且难以有效捕捉视觉表示中的长程依赖,静态损失函数也难以实现局部与全局特征的良好对齐。为此,我们提出ConMamba,一种专为PDD设计的新型自监督框架。ConMamba融合视觉马尔可夫编码器(VME),采用双向状态空间模型(SSM)高效捕获长程依赖。此外,引入具有动态权重调整的双层级对比损失,优化局部-全局特征对齐。在三个基准数据集上的实验表明,ConMamba在多个评估指标上显著优于现有最先进方法,为植物病害检测提供了高效且鲁棒的解决方案。
原文摘要 · Abstract (English)
Plant Disease Detection (PDD) is a key aspect of precision agriculture. However, existing deep learning methods often rely on extensively annotated datasets, which are time-consuming and costly to generate. Self-supervised Learning (SSL) offers a promising alternative by exploiting the abundance of unlabeled data. However, most existing SSL approaches suffer from high computational costs due to convolutional neural networks or transformer-based architectures. Additionally, they struggle to capture long-range dependencies in visual representation and rely on static loss functions that fail to align local and global features effectively. To address these challenges, we propose ConMamba, a novel SSL framework specially designed for PDD. ConMamba integrates the Vision Mamba Encoder (VME), which employs a bidirectional State Space Model (SSM) to capture long-range dependencies efficiently. Furthermore, we introduce a dual-level contrastive loss with dynamic weight adjustment to optimize local-global feature alignment. Experimental results on three benchmark datasets demonstrate that ConMamba significantly outperforms state-of-the-art methods across multiple evaluation metrics. This provides an efficient and robust solution for PDD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。