arXiv:2506.01040cs.CV2025-06被引 7

用状态空间模型提升极化SAR图像分类效率与精度

ECP-Mamba: An Efficient Multi-scale Self-supervised Contrastive Learning Method with State Space Model for PolSAR Image Classification

  • 结合多尺度自监督对比学习与Mamba架构,减少对标注数据依赖
  • 在Flevoland 1989数据集上达99.70%整体准确率,性能领先
  • 专为像素级分类设计螺旋扫描策略,计算更高效,适合资源受限场景

极化合成孔径雷达(PolSAR)图像分类近年来受益于深度神经网络的发展。然而,现有基于深度学习的方法仍面临标注数据依赖性强及Transformer类架构计算效率低的问题。本文提出ECP-Mamba框架,融合多尺度自监督对比学习与状态空间模型(SSM)主干网络。通过基于局部到全局特征对应关系的多尺度预测预训练任务,采用简化自蒸馏范式,无需负样本对,缓解标注稀缺问题。为提升计算效率,首次将Mamba架构(一种选择性状态空间模型)适配于像素级PolSAR分类任务,设计螺旋扫描策略,优先捕捉中心像素附近因果相关特征,契合像素分类的局部特性。此外,提出轻量级Cross Mamba模块,实现多尺度特征互补交互且开销极小。在四个基准数据集上的实验表明,ECP-Mamba在高精度与资源效率间取得良好平衡。在Flevoland 1989数据集上,整体准确率达99.70%,平均准确率为99.64%,Kappa系数为99.62e-2。代码将开源。

原文摘要 · Abstract (English)

Recently, polarimetric synthetic aperture radar (PolSAR) image classification has been greatly promoted by deep neural networks. However,current deep learning-based PolSAR classification methods encounter difficulties due to its dependence on extensive labeled data and the computational inefficiency of architectures like Transformers. This paper presents ECP-Mamba, an efficient framework integrating multi-scale self-supervised contrastive learning with a state space model (SSM) backbone. Specifically, ECP-Mamba addresses annotation scarcity through a multi-scale predictive pretext task based on local-to-global feature correspondences, which uses a simplified self-distillation paradigm without negative sample pairs. To enhance computational efficiency,the Mamba architecture (a selective SSM) is first tailored for pixel-wise PolSAR classification task by designing a spiral scan strategy. This strategy prioritizes causally relevant features near the central pixel, leveraging the localized nature of pixel-wise classification tasks. Additionally, the lightweight Cross Mamba module is proposed to facilitates complementary multi-scale feature interaction with minimal overhead. Extensive experiments across four benchmark datasets demonstrate ECP-Mamba's effectiveness in balancing high accuracy with resource efficiency. On the Flevoland 1989 dataset, ECP-Mamba achieves state-of-the-art performance with an overall accuracy of 99.70%, average accuracy of 99.64% and Kappa coefficient of 99.62e-2. Our code will be available at https://github.com/HaixiaBi1982/ECP_Mamba.

PolSAR分类Mamba自监督学习状态空间模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。