用知识迁移+领域自适应提升遥感图像细粒度分割效果
Knowledge Transfer and Domain Adaptation for Fine-Grained Remote Sensing Image Segmentation
- 通过特征对齐与调制模块,融合预训练视觉变换器与卷积网络优势
- 在草地图和云图上分别提升2.57和3.73 mIoU,显著改善分割精度
- 适合遥感图像细粒度分割任务,尤其数据稀缺场景下表现优异
细粒度遥感图像分割对精准识别遥感影像中的细节物体至关重要。近期,基于大规模数据预训练的视觉变换器(VTMs)展现出强大的零样本泛化能力,但直接应用于特定任务可能引发领域偏移。本文提出一种结合知识引导与领域精炼的端到端学习范式,包含两个核心组件:特征对齐模块(FAM)和特征调制模块(FMM)。FAM通过通道变换与空间插值,将基于CNN的主干特征与预训练VTM编码器特征对齐,并利用KL散度与L2归一化约束实现知识迁移;FMM进一步将迁移的知识适配至具体领域,缓解领域偏移。我们还构建了一个细粒度草地分割数据集。实验在两个数据集上验证了该方法的有效性,分别在草地图和云图上实现2.57 mIoU和3.73 mIoU的显著提升。结果表明,结合知识迁移与领域自适应可有效应对领域相关挑战与数据限制。项目页面见 https://xavierjiezou.github.io/KTDA/。
原文摘要 · Abstract (English)
Fine-grained remote sensing image segmentation is essential for accurately identifying detailed objects in remote sensing images. Recently, vision transformer models (VTMs) pre-trained on large-scale datasets have demonstrated strong zero-shot generalization. However, directly applying them to specific tasks may lead to domain shift. We introduce a novel end-to-end learning paradigm combining knowledge guidance with domain refinement to enhance performance. We present two key components: the Feature Alignment Module (FAM) and the Feature Modulation Module (FMM). FAM aligns features from a CNN-based backbone with those from the pretrained VTM's encoder using channel transformation and spatial interpolation, and transfers knowledge via KL divergence and L2 normalization constraint. FMM further adapts the knowledge to the specific domain to address domain shift. We also introduce a fine-grained grass segmentation dataset and demonstrate, through experiments on two datasets, that our method achieves a significant improvement of 2.57 mIoU on the grass dataset and 3.73 mIoU on the cloud dataset. The results highlight the potential of combining knowledge transfer and domain adaptation to overcome domain-related challenges and data limitations. The project page is available at https://xavierjiezou.github.io/KTDA/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。