Mislabel-Detection

此代码库用于测试实验。深度模型的性能可能受到训练集中标签错误或标签模糊样本的负面影响，而诊断数据集并识别错误标签的样本有助于提升模型的泛化能力。近年来，基于手工设计特征的训练动态（即优化算法迭代留下的痕迹）已被证明能有效定位错误标签的样本。本文进一步提出一种新颖的基于学习的解决方案，即通过一个由LSTM网络实现的噪声检测器，利用原始训练动态作为输入，学习预测样本是否被错误标注。具体而言，该方法通过带有合成标签噪声的数据集以监督方式训练噪声检测器，并能适应各种数据集（无论是自然标签噪声还是合成标签噪声）而无需重新训练。我们通过大量实验评估该方法，在基于合成标签噪声的CIFAR数据集上训练噪声检测器，并在Tiny ImageNet、CUB-200、Caltech-256、WebVision和Clothing1M等数据集上测试。结果表明，该方法在不进行额外适配的情况下能够精准检测各种数据集中的错误标签样本，并超越现有最先进的方法。此外，进一步实验表明，错误标签识别可以指导标签校正（即数据调试），从数据层面为基于算法的最先进技术提供互补提升。