arXiv:2502.02257cs.CV2025-02ICLR被引 7

针对红外语义分割,提出统一预训练框架UNIP提升模型性能。

UNIP: Rethinking Pre-trained Attention Patterns for Infrared Semantic Segmentation

  • 通过分析注意力模式,发现混合型关注对分割最有效。
  • 在三个红外任务上,平均mIoU提升最高达13.5%。
  • 计算成本仅为MAE-L的1/10,适合资源受限场景。

预训练技术能显著提升小样本语义分割性能,但其在跨域(如从RGB预训练到红外微调)下的效果仍不明确。本研究首次对多种预训练方法在红外语义分割上的表现进行基准测试,揭示了与可见光域不同的现象。层间注意力图分析发现:(1) 存在局部、混合和全局三类典型注意力模式;(2) 预训练任务显著影响各层注意力分布;(3) 混合模式对分割至关重要,能同时关注邻近区域与前景元素;(4) 纹理偏差会阻碍红外任务的泛化能力。基于此,提出统一红外预训练框架UNIP,以混合注意力蒸馏(NMI-HAD)为预训练目标,使用大规模混合数据集InfMix进行预训练,并采用最后一层特征金字塔网络(LL-FPN)进行微调。实验表明,UNIP在三个红外分割任务上,通过微调和线性探测评估,平均mIoU相较其他预训练方法最高提升13.5%。UNIP-S性能媲美MAE-L,但计算成本仅为后者的1/10。此外,其性能显著超越当前最优的红外或可见光分割方法,并展现出在其他模态(如RGB、深度)中的广泛应用潜力。代码已开源。

原文摘要 · Abstract (English)

Pre-training techniques significantly enhance the performance of semantic segmentation tasks with limited training data. However, the efficacy under a large domain gap between pre-training (e.g. RGB) and fine-tuning (e.g. infrared) remains underexplored. In this study, we first benchmark the infrared semantic segmentation performance of various pre-training methods and reveal several phenomena distinct from the RGB domain. Next, our layerwise analysis of pre-trained attention maps uncovers that: (1) There are three typical attention patterns (local, hybrid, and global); (2) Pre-training tasks notably influence the pattern distribution across layers; (3) The hybrid pattern is crucial for semantic segmentation as it attends to both nearby and foreground elements; (4) The texture bias impedes model generalization in infrared tasks. Building on these insights, we propose UNIP, a UNified Infrared Pre-training framework, to enhance the pre-trained model performance. This framework uses the hybrid-attention distillation NMI-HAD as the pre-training target, a large-scale mixed dataset InfMix for pre-training, and a last-layer feature pyramid network LL-FPN for fine-tuning. Experimental results show that UNIP outperforms various pre-training methods by up to 13.5\% in average mIoU on three infrared segmentation tasks, evaluated using fine-tuning and linear probing metrics. UNIP-S achieves performance on par with MAE-L while requiring only 1/10 of the computational cost. Furthermore, UNIP significantly surpasses state-of-the-art (SOTA) infrared or RGB segmentation methods and demonstrates broad potential for application in other modalities, such as RGB and depth. Our code is available at https://github.com/casiatao/UNIP.

红外分割预训练注意力机制轻量化

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