通过自适应评估伪标签,提升遥感变化检测的精度与稳定性。
AdaSemiCD: An Adaptive Semi-Supervised Change Detection Method Based on Pseudo-Label Evaluation
- 设计可衡量的伪标签评估指标,增强混淆区域和变化目标的权重。
- 引入动态融合模块,用可信内容替换不确定区域的伪标签。
- 采用自适应指数移动平均更新教师模型,仅使用高置信度样本训练。
变化检测(CD)是遥感领域的关键任务,旨在识别同一地区在不同时期由卫星拍摄的双时相图像对中的变化区域。该任务的数据标注过程耗时且费力。为更高效利用少量标注数据和大量未标注数据,本文提出一种自适应半监督学习方法AdaSemiCD,以优化伪标签使用并改进训练流程。针对变化检测中固有的极端类别不平衡问题,模型易偏向背景类且难以区分目标边界,我们设计了一种可度量的伪标签评估机制,通过类别平衡与混淆区域放大,提升信息熵表征能力,赋予变化目标更高权重。为进一步提高样本级伪标签可靠性,提出AdaFusion模块,可动态识别最不确定区域,并以更可信内容进行替换。最后,为保障训练稳定性,引入AdaEMA模块,仅使用高置信度批次更新教师模型。在LEVIR-CD、WHU-CD和CDD数据集上的实验验证了所提框架的有效性与通用性。
原文摘要 · Abstract (English)
Change Detection (CD) is an essential field in remote sensing, with a primary focus on identifying areas of change in bi-temporal image pairs captured at varying intervals of the same region by a satellite. The data annotation process for the CD task is both time-consuming and labor-intensive. To make better use of the scarce labeled data and abundant unlabeled data, we present an adaptive dynamic semi-supervised learning method, AdaSemiCD, to improve the use of pseudo-labels and optimize the training process. Initially, due to the extreme class imbalance inherent in CD, the model is more inclined to focus on the background class, and it is easy to confuse the boundary of the target object. Considering these two points, we develop a measurable evaluation metric for pseudo-labels that enhances the representation of information entropy by class rebalancing and amplification of confusing areas to give a larger weight to prospects change objects. Subsequently, to enhance the reliability of sample-wise pseudo-labels, we introduce the AdaFusion module, which is capable of dynamically identifying the most uncertain region and substituting it with more trustworthy content. Lastly, to ensure better training stability, we introduce the AdaEMA module, which updates the teacher model using only batches of trusted samples. Experimental results from LEVIR-CD, WHU-CD, and CDD datasets validate the efficacy and universality of our proposed adaptive training framework.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。