arXiv:2501.11053cs.LGcs.CV2025-01AAAI被引 6

新方法应对真实场景中的未知噪声标签,提升模型鲁棒性。

Learning with Open-world Noisy Data via Class-independent Margin in Dual Representation Space

  • 构建双空间结构,分别学习共享与独立语义表示
  • 在CIFAR80N上实现4.55%准确率提升和6.17% AUROC提升
  • 适合处理含未知类噪声的真实数据场景

学习带噪声标签(LNL)旨在提升面对噪声标签数据时的模型泛化能力。现有方法通常假设噪声标签来自已知类别(封闭集噪声),但真实场景中可能存在来自相似未知类别的开放集噪声,严重影响现有LNL方法性能。本文提出一种新型双空间联合学习方法,以鲁棒应对开放世界噪声。通过两个网络构建双表示空间:一个投影网络学习原型空间中的共享表示,另一个One-Vs-All(OVA)网络在类无关空间中使用独特语义表示进行预测。在两个空间中引入双层次对比学习与一致性正则化,增强对未知类别样本的检测能力。设计类无关边界准则,利用样本记忆效应,有效识别干净样本、加权封闭集噪声并过滤开放集噪声。大量实验表明,本方法优于现有最先进方法,在CIFAR80N上平均准确率提升4.55%,AUROC提升6.17%。

原文摘要 · Abstract (English)

Learning with Noisy Labels (LNL) aims to improve the model generalization when facing data with noisy labels, and existing methods generally assume that noisy labels come from known classes, called closed-set noise. However, in real-world scenarios, noisy labels from similar unknown classes, i.e., open-set noise, may occur during the training and inference stage. Such open-world noisy labels may significantly impact the performance of LNL methods. In this study, we propose a novel dual-space joint learning method to robustly handle the open-world noise. To mitigate model overfitting on closed-set and open-set noises, a dual representation space is constructed by two networks. One is a projection network that learns shared representations in the prototype space, while the other is a One-Vs-All (OVA) network that makes predictions using unique semantic representations in the class-independent space. Then, bi-level contrastive learning and consistency regularization are introduced in two spaces to enhance the detection capability for data with unknown classes. To benefit from the memorization effects across different types of samples, class-independent margin criteria are designed for sample identification, which selects clean samples, weights closed-set noise, and filters open-set noise effectively. Extensive experiments demonstrate that our method outperforms the state-of-the-art methods and achieves an average accuracy improvement of 4.55\% and an AUROC improvement of 6.17\% on CIFAR80N.

噪声标签开放集双空间鲁棒学习

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