arXiv:2504.06544cs.CV2025-04AAAI被引 6

提出新方法缓解类别不平衡下的半监督学习偏差问题。

LCGC: Learning from Consistency Gradient Conflicting for Class-Imbalanced Semi-Supervised Debiasing

  • 利用图像梯度冲突机制识别并修正偏差预测。
  • 在多个公开数据集上显著提升现有模型准确率。
  • 适合解决类别不平衡场景下的半监督学习任务。

分类器在类别不平衡的数据集上,尤其在半监督学习(SSL)设定下,容易产生偏差。以往方法通过减去与类别无关图像的logit来重新平衡分类器,但缺乏坚实的理论基础。本文从理论上分析为何使用基准图像可优化伪标签,并证明黑图是最佳选择;同时发现随着训练深入,伪标签在精炼前后的差异逐渐缩小。基于此,提出名为LCGC的去偏方案——通过鼓励训练过程中对偏差类别预测的更新,主动调整与去偏逻辑相冲突的伪标签梯度方向。测试时,通过减去基准图像的logit实现去偏。大量实验表明,LCGC能显著提升现有类别不平衡半监督学习(CISSL)模型在公开基准上的预测精度。

原文摘要 · Abstract (English)

Classifiers often learn to be biased corresponding to the class-imbalanced dataset, especially under the semi-supervised learning (SSL) set. While previous work tries to appropriately re-balance the classifiers by subtracting a class-irrelevant image's logit, but lacks a firm theoretical basis. We theoretically analyze why exploiting a baseline image can refine pseudo-labels and prove that the black image is the best choice. We also indicated that as the training process deepens, the pseudo-labels before and after refinement become closer. Based on this observation, we propose a debiasing scheme dubbed LCGC, which Learning from Consistency Gradient Conflicting, by encouraging biased class predictions during training. We intentionally update the pseudo-labels whose gradient conflicts with the debiased logits, representing the optimization direction offered by the over-imbalanced classifier predictions. Then, we debiased the predictions by subtracting the baseline image logits during testing. Extensive experiments demonstrate that LCGC can significantly improve the prediction accuracy of existing CISSL models on public benchmarks.

半监督学习去偏类别不平衡

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。