深度学习中的标签噪声学习进展
近年来,深度学习在计算机视觉、自然语言处理等领域取得了巨大成功。然而,深度神经网络的训练往往需要大量的标注数据,而在实际应用中,获取大规模高质量的标注数据是一项耗时耗力的工作。因此,如何从带有噪声标签的数据中学习,成为了深度学习领域的一个重要研究方向。本文将对标签噪声学习的最新进展进行全面综述。
标签噪声问题定义
标签噪声学习(Learning with Noisy Labels)研究的是如何从存在错误标注的数据集中学习出高性能的模型。具体来说,给定一个带有噪声标签的训练集 D = {(x1, y1), (x2, y2), ..., (xn, yn)},其中xi是输入样本,yi是对应的标签,但yi可能是错误的。我们的目标是设计一个鲁棒的学习算法,使得在这样的带噪声数据集上训练出的模型,能够在干净的测试集上取得良好的泛化性能。
标签噪声可以分为以下几种类型:
- 随机噪声(Random Noise):标签被随机翻转为其他类别。
- 对称噪声(Symmetric Noise):标签以相等的概率被翻转为其他类别。
- 非对称噪声(Asymmetric Noise):标签以不同的概率被翻转为语义相近的类别。
- 实例相关噪声(Instance-dependent Noise):噪声与输入样本相关。
其中,实例相关噪声更接近真实场景,也是目前研究的重点和难点。
主要研究方法
针对标签噪声学习问题,研究人员提出了多种方法,大致可以分为以下几类:
1. 噪声转移矩阵估计
这类方法通过估计从真实标签到观测标签的转移概率矩阵,来建模标签噪声。代表性工作包括:
-
Estimating the Noise Transition Matrix with Label Correlations for Noisy Multi-Label Learning(NeurIPS 2022)提出了一种利用标签相关性来估计噪声转移矩阵的方法。
-
Beyond Images: Label Noise Transition Matrix Estimation for Tasks with Lower-Quality Features(ICML 2022)研究了如何在特征质量较差的任务中估计噪声转移矩阵。
2. 样本选择
这类方法通过选择可能是干净样本的子集进行训练。代表性工作有:
-
Selective-Supervised Contrastive Learning with Noisy Labels(CVPR 2022)提出了一种选择性监督的对比学习方法。
-
Sample Selection with Uncertainty of Losses for Learning with Noisy Labels(ICLR 2022)利用损失的不确定性来进行样本选择。
3. 损失修正
这类方法通过修正损失函数来减轻噪声标签的影响。代表性工作包括:
-
To Smooth or Not? When Label Smoothing Meets Noisy Labels(ICML 2022)研究了标签平滑在噪声标签场景下的作用。
-
Mitigating Memorization of Noisy Labels via Regularization between Representations(ICLR 2023)通过表示之间的正则化来缓解对噪声标签的记忆。
4. 元学习
这类方法利用元学习的思想来学习噪声环境下的学习策略。代表性工作有:
- Robust Meta-learning with Sampling Noise and Label Noise via Eigen-Reptile(ICML 2022)提出了一种基于特征分解的鲁棒元学习方法。
5. 半监督学习
这类方法将噪声标签学习问题转化为半监督学习问题。代表性工作包括:
- Which is Better for Learning with Noisy Labels: The Semi-supervised Method or Modeling Label Noise?(ICML 2023)比较了半监督方法和噪声建模方法在噪声标签学习中的效果。
评估基准
为了公平比较各种方法的性能,研究人员构建了一些标准的评估基准数据集:
-
CIFAR-10N/CIFAR-100N:基于CIFAR-10/100数据集,通过众包标注获得的真实噪声标签数据集。
-
Clothing1M:包含100万张服装图片的大规模真实噪声数据集。
-
WebVision:从网络收集的包含1600万张图片的大规模噪声数据集。
-
Animal-10N:包含10类动物的5万张图片,标签噪声率约8%。
-
Food-101N:包含101类食物的31万张图片,估计噪声率约20%。
这些数据集涵盖了不同规模、不同领域、不同噪声类型,为相关研究提供了全面的评估环境。
未来研究方向
尽管标签噪声学习已经取得了显著进展,但仍然存在一些重要的开放问题值得进一步研究:
-
实例相关噪声:如何有效建模和处理与输入相关的复杂噪声模式?
-
噪声估计:如何在无监督的情况下准确估计数据集的噪声分布?
-
可解释性:如何提高噪声标签学习方法的可解释性?
-
跨域泛化:如何提高噪声标签学习方法在跨域场景下的泛化能力?
-
大规模预训练:如何利用大规模预训练模型来提升噪声标签学习的性能?
-
数据高效学习:如何在小样本、弱监督等数据受限场景下进行有效的噪声标签学习?
结论
标签噪声学习是深度学习领域一个重要的研究方向,对于提高深度学习模型在实际应用中的鲁棒性具有重要意义。本文综述了该领域的最新进展,包括问题定义、主要方法、评估基准以及未来研究方向。随着研究的不断深入,我们相信会有更多创新性的方法被提出,进一步推动深度学习在噪声环境下的应用。
参考文献
-
Song, H., Kim, M., Park, D., Shin, Y., & Lee, J. G. (2022). Learning from noisy labels with deep neural networks: A survey. IEEE Transactions on Neural Networks and Learning Systems.
-
Wei, J., Liu, Y., Luo, Y., Shen, C., & Yang, Y. (2022). To aggregate or not? Learning with separate noisy labels. In Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining (pp. 1814-1824).
-
Liu, Y., & Guo, H. (2020). Peer loss functions: Learning from noisy labels without knowing noise rates. In International Conference on Machine Learning (pp. 6226-6236). PMLR.
-
Jiang, L., Zhou, Z., Leung, T., Li, L. J., & Fei-Fei, L. (2018). Mentornet: Learning data-driven curriculum for very deep neural networks on corrupted labels. In International Conference on Machine Learning (pp. 2304-2313). PMLR.
-
Han, B., Yao, Q., Yu, X., Niu, G., Xu, M., Hu, W., ... & Sugiyama, M. (2018). Co-teaching: Robust training of deep neural networks with extremely noisy labels. Advances in neural information processing systems, 31.