针对卫星图像分割中的类别不平衡问题,提出动态关注薄弱类的主动学习方法。
Dynamic Class-Aware Active Learning for Unbiased Satellite Image Segmentation

- 根据每类实时表现差距动态调整采样权重,聚焦难分或稀有类别。
- 在OpenEarth数据集上显著提升小类的交并比,标注效率更高。
- 适合资源受限、类别不均的遥感图像标注场景。
卫星影像语义分割在土地覆盖制图与环境监测中至关重要,但大规模高分辨率数据标注成本高昂。主动学习(AL)通过人机协同智能选择最有信息量的样本进行标注,可显著降低标注成本并保持高模型性能,尤其适用于大范围或资源受限的卫星应用。然而,传统AL方法多依赖全局不确定度或多样性度量,难以随训练过程动态关注表现较差或稀有类别,导致系统性偏差。为此,本文提出一种新型自适应获取函数——动态类别感知不确定性主动学习(DCAU-AL),依据各分类别实时性能差距优先选择样本,有效缓解类别不平衡问题。该机制持续追踪每类分割表现,并动态调整采样权重,始终聚焦于表现不佳或样本不足的类别。在OpenEarth土地覆盖数据集上的大量实验表明,DCAU-AL显著优于现有AL方法,尤其在严重类别不平衡条件下,实现了更高的逐类交并比(IoU)和更优的标注效率。
原文摘要 · Abstract (English)
Semantic segmentation of satellite imagery plays a vital role in land cover mapping and environmental monitoring. However, annotating large-scale, high-resolution satellite datasets is costly and time consuming, especially when covering vast geographic regions. Instead of randomly labeling data or exhaustively annotating entire datasets, Active Learning (AL) offers an efficient alternative by intelligently selecting the most informative samples for annotation with the help of Human-in-the-loop (HITL), thereby reducing labeling costs while maintaining high model performance. AL is particularly beneficial for large-scale or resource-constrained satellite applications, as it enables high segmentation accuracy with significantly fewer labeled samples. Despite these advantages, standard AL strategies typically rely on global uncertainty or diversity measures and lack the adaptability to target underperforming or rare classes as training progresses, leading to bias in the system. To overcome these limitations, we propose a novel adaptive acquisition function, Dynamic Class-Aware Uncertainty based Active learning (DCAU-AL) that prioritizes sample selection based on real-time class-wise performance gaps, thereby overcoming class-imbalance issue. The proposed DCAU-AL mechanism continuously tracks the performance of the segmentation per class and dynamically adjusts the sampling weights to focus on poorly performing or underrepresented classes throughout the active learning process. Extensive experiments on the OpenEarth land cover dataset show that DCAU-AL significantly outperforms existing AL methods, especially under severe class imbalance, delivering superior per-class IoU and improved annotation efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。