通过高置信度伪标签提升标签比例学习的准确性
Forming Auxiliary High-confident Instance-level Loss to Promote Learning from Label Proportions
- 用双熵权重动态评估伪标签置信度,防止过平滑
- 在大包尺寸下性能提升显著,超越现有基线方法
- 适合标签比例标注场景,尤其适用于大规模数据
标签比例学习(LLP)是一种弱监督学习任务,旨在仅使用包含多个实例的包及其类别比例来训练分类器,而非每个实例的标注标签。主流方法通常引入基于预测生成的伪标签的辅助实例级损失,但实验发现,由于过平滑现象,伪标签在大包尺寸下往往不准确,影响分类器性能。为此,本文提出一种新方法L^2P-AHIL,通过双熵权重(DEW)自适应地衡量伪标签置信度,同时兼顾包级准确性和避免过度平滑。进而构建高置信度实例级损失,并与包级损失联合优化,采用自训练方式提升模型。在基准数据集上的实验表明,该方法优于现有基线,且随着包尺寸增大,性能增益更加明显。
原文摘要 · Abstract (English)
Learning from label proportions (LLP), i.e., a challenging weakly-supervised learning task, aims to train a classifier by using bags of instances and the proportions of classes within bags, rather than annotated labels for each instance. Beyond the traditional bag-level loss, the mainstream methodology of LLP is to incorporate an auxiliary instance-level loss with pseudo-labels formed by predictions. Unfortunately, we empirically observed that the pseudo-labels are are often inaccurate due to over-smoothing, especially for the scenarios with large bag sizes, hurting the classifier induction. To alleviate this problem, we suggest a novel LLP method, namely Learning from Label Proportions with Auxiliary High-confident Instance-level Loss (L^2P-AHIL). Specifically, we propose a dual entropy-based weight (DEW) method to adaptively measure the confidences of pseudo-labels. It simultaneously emphasizes accurate predictions at the bag level and avoids overly smoothed predictions. We then form high-confident instance-level loss with DEW, and jointly optimize it with the bag-level loss in a self-training manner. The experimental results on benchmark datasets show that L^2P-AHIL can surpass the existing baseline methods, and the performance gain can be more significant as the bag size increases. The implementation of our method is available at https://github.com/TianhaoMa5/LLP-AHIL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。