针对遥感图像标签不全问题,提出自适应梯度校准方法提升多标签分类精度
Adaptive Gradient Calibration for Single-Positive Multi-Label Learning in Remote Sensing Image Scene Classification
- 通过双指数移动平均机制生成稳定伪标签
- 在两种噪声场景下均达到当前最优性能
- 适合遥感图像多标签学习且对噪声鲁棒
多标签分类(MLC)相比传统单标签分类(SLC)能更全面理解遥感(RS)图像语义。但完整标注MLC成本高、难度大。为此,单正例多标签学习(SPML)被提出:每张图像仅标注一个相关标签,模型需推断全部标签。尽管可扩展性强,但监督信号模糊,需特殊训练方案。现有计算机视觉的SPML方法在遥感领域仍较少。本文提出面向遥感图像的自适应梯度校准(AdaGC)框架,采用梯度校准(GC)机制结合双指数移动平均(EMA)模块,实现稳健伪标签生成。设计基于训练动态的理论指标,自适应触发GC,避免模型因欠拟合或过拟合噪声而失效。在两个基准遥感数据集上,针对两种不同标签噪声类型进行实验,结果表明AdaGC在多种设置下均达到当前最优(SOTA)表现,且具备强鲁棒性。
原文摘要 · Abstract (English)
Multi-label classification (MLC) offers a more comprehensive semantic understanding of Remote Sensing (RS) imagery compared to traditional single-label classification (SLC). However, obtaining complete annotations for MLC is particularly challenging due to the complexity and high cost of the labeling process. As a practical alternative, single-positive multi-label learning (SPML) has emerged, where each image is annotated with only one relevant label, and the model is expected to recover the full set of labels. While scalable, SPML introduces significant supervision ambiguity, demanding specialized solutions for model training. Although various SPML methods have been proposed in the computer vision domain, research in the RS context remains limited. To bridge this gap, we propose Adaptive Gradient Calibration (AdaGC), a novel and generalizable SPML framework tailored to RS imagery. AdaGC adopts a gradient calibration (GC) mechanism with a dual exponential moving average (EMA) module for robust pseudo-label generation. We introduce a theoretically grounded, training-dynamics-based indicator to adaptively trigger GC, which ensures GC's effectiveness by preventing it from being affected by model underfitting or overfitting to label noise. Extensive experiments on two benchmark RS datasets under two distinct label noise types demonstrate that AdaGC achieves state-of-the-art (SOTA) performance while maintaining strong robustness across diverse settings. The codes and data will be released at https://github.com/rslab-unitrento/AdaGC.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。