提升ViT抗噪声标签能力,显式熵正则化有效增强模型鲁棒性
Implicit to Explicit Entropy Regularization: Benchmarking ViT Fine-tuning under Noisy Labels
- 设计显式熵正则化,主动降低模型预测熵以增强抗噪能力
- 在六个数据集上验证,显式熵正则化显著提升ViT在噪声标签下的准确率
- 适用于需要高鲁棒性的视觉Transformer微调场景,如真实世界图像数据
大规模数据集的自动标注会引入噪声标签,严重影响深度神经网络的学习效果。尽管卷积神经网络(CNN)在噪声标签学习(NLL)方面已有较多研究,但视觉变换器(ViT)仍缺乏系统评估。本文评估了ViT微调对噪声标签的敏感性,并与CNN对比其鲁棒性;同时检验为CNN设计的六种NLL方法在ViT上的适用性。采用线性探测和MLP-K微调,测试了两个ViT骨干网络(ViT-B/16和ViT-L/16),以及交叉熵(CE)、焦点损失(FL)、平均绝对误差(MAE)三种常用损失函数,搭配六种鲁棒方法:GCE、SCE、NLNL、APL、NCE+AGCE、ANL-CE。实验覆盖MNIST、CIFAR-10/100、WebVision、Clothing1M、Food-101N共六个数据集。观察到多数NLL方法均伴随预测熵下降,据此提出显式熵最小化策略。结果表明,引入熵正则化可提升CE、FL等传统损失及六种方法在双骨干模型上的性能,显著增强ViT在噪声标签下的泛化能力。
原文摘要 · Abstract (English)
Automatic annotation of large-scale datasets can introduce noisy training data labels, which adversely affect the learning process of deep neural networks (DNNs). Consequently, Noisy Labels Learning (NLL) has become a critical research field for Convolutional Neural Networks (CNNs), though it remains less explored for Vision Transformers (ViTs). In this study, we evaluate the vulnerability of ViT fine-tuning to noisy labels and compare its robustness with CNNs. We also investigate whether NLL methods developed for CNNs are equally effective for ViTs. Using linear probing and MLP-K fine-tuning, we benchmark two ViT backbones (ViT-B/16 and ViT-L/16) using three commonly used classification losses: Cross Entropy (CE), Focal Loss (FL), and Mean Absolute Error (MAE), alongside six robust NLL methods: GCE, SCE, NLNL, APL, NCE+AGCE, and ANL-CE. The evaluation is conducted across six datasets including MNIST, CIFAR-10/100, WebVision, Clothing1M, and Food-101N. Furthermore, we explore whether implicit prediction entropy minimization contributes to ViT robustness against noisy labels, noting a general trend of prediction entropy reduction across most NLL methods. Building on this observation, we examine whether explicit entropy minimization could enhance ViT resilience to noisy labels. Our findings indicate that incorporating entropy regularization enhances the performance of established loss functions such as CE and FL, as well as the robustness of the six studied NLL methods across both ViT backbones.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。