用物理先验指导掩码,让雷达图像识别模型更高效准确
SAMBA: A Scatter-Guided Masked Bidirectional Mamba Foundation Model for SAR Target Recognition

- 基于雷达散射特性设计分层掩码策略,匹配成像机理
- 引入线性复杂度Mamba结构,参数量比传统模型少得多
- 在7个数据集上实现顶尖性能,适合小样本场景
合成孔径雷达自动目标识别(SAR ATR)在地球观测与国防领域至关重要,但受限于标注数据稀缺。自监督预训练可缓解标签瓶颈,但现有Transformer架构计算复杂度达二次方,且通用掩码策略忽视了雷达图像独特的电磁散射特性。为此,我们提出SAMBA(Scattering-Guided Bidirectional Mamba),一种面向SAR目标理解的高效自监督预训练基础模型。框架包含三大创新:(i) 基于Mamba的线性复杂度编码器结合序列中间类标记,缓解计算瓶颈;(ii) 三层分层散射引导掩码自编码器(SG-MAE),利用雷达物理先验,使预训练任务契合成像机制;(iii) 轻量级SpatialMix模块,增强跨区域特征融合。同时设计两阶段跨域预训练流程优化整体训练效率。大量实验表明,SAMBA在所有预训练配置下均表现优异,参数量显著低于CNN与Transformer基线。相较于标准MAE的默认掩码策略,所提SG-MAE进一步提升模型少样本迁移能力。在涵盖分类与检测任务的七个下游数据集上,SAMBA多数指标达到当前最优(SOTA),充分验证其在多样化SAR解析任务中的鲁棒泛化能力。源代码与预训练权重已公开于https://github.com/mynswkk/SAMBA。
原文摘要 · Abstract (English)
Synthetic aperture radar automatic target recognition (SAR ATR) is critical for Earth observation and defense, but its practical deployment is constrained by scarce annotated training data. Self-supervised pre-training alleviates this label bottleneck, yet prevailing Transformer architectures incur prohibitive quadratic computational complexity, and conventional universal masking neglects the unique electromagnetic scattering properties intrinsic to SAR imagery. To address these limitations, we propose SAMBA (Scattering-Guided Bidirectional Mamba), an efficient self-supervised pre-training foundation model for SAR target interpretation. Our framework features three core innovations: (i) a linear-complexity Mamba encoder with a mid-sequence class token to mitigate computational bottlenecks; (ii) a three-level hierarchical Scattering-Guided Masked Autoencoder (SG-MAE) masking strategy guided by SAR physical priors, aligning the pretext task with SAR's intrinsic imaging mechanism; (iii) a lightweight SpatialMix feature interaction module to enhance cross-region feature fusion. We also design a two-stage cross-domain pre-training pipeline to optimize the overall pre-training process. Extensive evaluations demonstrate that SAMBA consistently delivers superior performance across all pre-training configurations, with substantially fewer parameters than both CNN and Transformer baselines. Compared with the default masking strategy in standard MAE, the proposed SG-MAE strategy further boosts the model's few-shot transfer capability. Benchmarking on seven downstream datasets covering classification and detection tasks shows SAMBA achieves state-of-the-art (SOTA) performance on most metrics, fully validating its robust generalizability across diverse SAR interpretation tasks. Source code and pre-trained weights are publicly available at https://github.com/mynswkk/SAMBA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。