提出DI3CL框架,用动态实例和轮廓一致性提升雷达地物分类模型泛化能力。
DI3CL: Contrastive Learning With Dynamic Instances and Contour Consistency for SAR Land-Cover Classification Foundation Model
- 通过动态实例与轮廓一致性对比学习,增强模型对地物结构的感知。
- 在46万张SAR图像上预训练,显著提升下游任务性能。
- 适合需要少标注数据的遥感地物分类场景,开源代码可复现。
尽管合成孔径雷达(SAR)地物分类已取得显著进展,但现有方法仍主要依赖大量标注数据的监督学习,限制了模型的可扩展性与泛化能力。本文提出一种通用型基础模型,用于加速下游模型的开发与部署。该模型基于动态实例与轮廓一致性对比学习(DI3CL)框架,包含动态实例(DI)模块和轮廓一致性(CC)模块:DI模块通过同一区域不同视角间的局部一致性,提升全局上下文感知;CC模块利用浅层特征图引导模型关注地物几何轮廓,增强结构判别力。为提升预训练鲁棒性,构建了一个包含460,532张SAR图像的大规模多样化数据集SARSense,使模型能捕捉全面且具代表性的特征。在多类任务(如地物制图、水体检测、道路提取)上的实验表明,所提DI3CL模型性能优于现有方法。代码与预训练权重已公开:https://github.com/SARpre-train/DI3CL。
原文摘要 · Abstract (English)
Although significant advances have been achieved in SAR land-cover classification, recent methods remain predominantly focused on supervised learning, which relies heavily on extensive labeled datasets. This dependency not only limits scalability and generalization but also restricts adaptability to diverse application scenarios. In this paper, a general-purpose foundation model for SAR land-cover classification is developed, serving as a robust cornerstone to accelerate the development and deployment of various downstream models. Specifically, a Dynamic Instance and Contour Consistency Contrastive Learning (DI3CL) pre-training framework is presented, which incorporates a Dynamic Instance (DI) module and a Contour Consistency (CC) module. DI module enhances global contextual awareness by enforcing local consistency across different views of the same region. CC module leverages shallow feature maps to guide the model to focus on the geometric contours of SAR land-cover objects, thereby improving structural discrimination. Additionally, to enhance robustness and generalization during pre-training, a large-scale and diverse dataset named SARSense, comprising 460,532 SAR images, is constructed to enable the model to capture comprehensive and representative features. To evaluate the generalization capability of our foundation model, we conducted extensive experiments across a variety of SAR land-cover classification tasks, including SAR land-cover mapping, water body detection, and road extraction. The results consistently demonstrate that the proposed DI3CL outperforms existing methods. Our code and pre-trained weights are publicly available at: https://github.com/SARpre-train/DI3CL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。