arXiv:2502.15798cs.LGcs.AI2025-02NeurIPS被引 3

提出新方法缓解标签平滑导致的特征坍缩问题

MaxSup: Overcoming Representation Collapse in Label Smoothing

  • 用最大置信度惩罚替代真实类别惩罚,统一正误样本正则化
  • 在多个数据集上显著提升类内多样性,改善分类边界清晰度
  • 适合追求鲁棒性与泛化能力的图像分类研究者使用

标签平滑(Label Smoothing, LS)虽能降低神经网络预测的过度自信并提升泛化性能,但近期研究揭示其存在两大问题:一是对错误分类样本产生过度自信;二是使特征表示过于紧密聚集,削弱类内多样性。本文通过解析LS损失函数,发现两个关键项:(i) 仅在预测正确时起作用的正则化项;(ii) 在误分类时引发误差放大的项,促使模型对错误预测赋予过高置信度,加剧特征坍缩。为此,我们提出最大抑制(MaxSup),通过对顶部置信度而非真实类别置信度施加统一正则化,同时约束正确与错误预测。大量特征空间分析表明,MaxSup恢复了类内差异并锐化了类间边界。在大规模图像分类及多个下游任务上的实验验证了其优于传统LS的鲁棒性。代码已公开于https://github.com/ZhouYuxuanYX/Maximum-Suppression-Regularization。

原文摘要 · Abstract (English)

Label Smoothing (LS) is widely adopted to reduce overconfidence in neural network predictions and improve generalization. Despite these benefits, recent studies reveal two critical issues with LS. First, LS induces overconfidence in misclassified samples. Second, it compacts feature representations into overly tight clusters, diluting intra-class diversity, although the precise cause of this phenomenon remained elusive. In this paper, we analytically decompose the LS-induced loss, exposing two key terms: (i) a regularization term that dampens overconfidence only when the prediction is correct, and (ii) an error-amplification term that arises under misclassifications. This latter term compels the network to reinforce incorrect predictions with undue certainty, exacerbating representation collapse. To address these shortcomings, we propose Max Suppression (MaxSup), which applies uniform regularization to both correct and incorrect predictions by penalizing the top-1 logit rather than the ground-truth logit. Through extensive feature-space analyses, we show that MaxSup restores intra-class variation and sharpens inter-class boundaries. Experiments on large-scale image classification and multiple downstream tasks confirm that MaxSup is a more robust alternative to LS. Code is available at: https://github.com/ZhouYuxuanYX/Maximum-Suppression-Regularization

标签平滑特征坍缩正则化

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