大模型更抗噪声,小图块未必更好,主动学习需权衡精度与校准。
Balancing Accuracy, Calibration, and Efficiency in Active Learning with Vision Transformers Under Label Noise
- 对比4种ViT和3种Swin在噪声标签下的表现,发现大模型更稳定。
- ViTl32在高噪声下仍优于小模型,而小图块增加开销却无收益。
- 主动学习在中等噪声下提升精度,但会降低模型校准性。
将预训练的卷积神经网络在ImageNet上微调以用于下游任务已成常规,但视觉变换器(Vision Transformers)在类似场景下,尤其是存在标签噪声时,其模型规模对性能的影响尚不明确。鉴于变换器架构的实用性与通用性,本研究探讨了其在低预算和标签噪声条件下的适用性。我们在CIFAR10和CIFAR100数据集上评估了四种视觉变换器配置(基线与大型,16×16和32×32图像块大小)及三种Swin变换器配置(微型、小型、基线),在不同标签噪声率下的分类准确率与校准能力。结果表明,更大的ViT模型(尤其是ViTl32)在中等至高噪声环境下,持续优于较小模型,而Swin变换器在所有噪声水平下均表现出较弱鲁棒性。较小图像块并未总带来更好性能:尽管ViTl16计算成本更高,其表现始终不如ViTl32。此外,基于信息的主动学习策略仅在中等噪声下显著提升准确率,但在高噪声下导致模型校准性下降,远低于随机采样训练的模型。这些发现为资源受限环境下部署视觉变换器提供了实用指导,强调在模型复杂度、标签噪声与计算效率之间取得平衡的重要性。
原文摘要 · Abstract (English)
Fine-tuning pre-trained convolutional neural networks on ImageNet for downstream tasks is well-established. Still, the impact of model size on the performance of vision transformers in similar scenarios, particularly under label noise, remains largely unexplored. Given the utility and versatility of transformer architectures, this study investigates their practicality under low-budget constraints and noisy labels. We explore how classification accuracy and calibration are affected by symmetric label noise in active learning settings, evaluating four vision transformer configurations (Base and Large with 16x16 and 32x32 patch sizes) and three Swin Transformer configurations (Tiny, Small, and Base) on CIFAR10 and CIFAR100 datasets, under varying label noise rates. Our findings show that larger ViT models (ViTl32 in particular) consistently outperform their smaller counterparts in both accuracy and calibration, even under moderate to high label noise, while Swin Transformers exhibit weaker robustness across all noise levels. We find that smaller patch sizes do not always lead to better performance, as ViTl16 performs consistently worse than ViTl32 while incurring a higher computational cost. We also find that information-based Active Learning strategies only provide meaningful accuracy improvements at moderate label noise rates, but they result in poorer calibration compared to models trained on randomly acquired labels, especially at high label noise rates. We hope these insights provide actionable guidance for practitioners looking to deploy vision transformers in resource-constrained environments, where balancing model complexity, label noise, and compute efficiency is critical in model fine-tuning or distillation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。