Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

When Does Label Smoothing Help?

About

The generalization and learning speed of a multi-class neural network can often be significantly improved by using soft targets that are a weighted average of the hard targets and the uniform distribution over labels. Smoothing the labels in this way prevents the network from becoming over-confident and label smoothing has been used in many state-of-the-art models, including image classification, language translation and speech recognition. Despite its widespread use, label smoothing is still poorly understood. Here we show empirically that in addition to improving generalization, label smoothing improves model calibration which can significantly improve beam-search. However, we also observe that if a teacher network is trained with label smoothing, knowledge distillation into a student network is much less effective. To explain these observations, we visualize how label smoothing changes the representations learned by the penultimate layer of the network. We show that label smoothing encourages the representations of training examples from the same class to group in tight clusters. This results in loss of information in the logits about resemblances between instances of different classes, which is necessary for distillation, but does not hurt generalization or calibration of the model's predictions.

Rafael M\"uller, Simon Kornblith, Geoffrey Hinton• 2019

Related benchmarks

TaskDatasetResultRank
Fine-grained Image ClassificationCUB200 2011 (test)
Accuracy82.74
585
Image ClassificationCIFAR-10--
564
Out-of-Distribution DetectionCIFAR-10 vs SVHN (test)
AUROC0.7423
146
Out-of-Distribution DetectionCIFAR-10 vs CIFAR-100 (test)
AUROC75.2
119
Image ClassificationCIFAR-100 (test)
Accuracy72.54
93
Long-tailed classificationCIFAR-100 long-tail (test)
Accuracy (All)56
77
Long-tailed classificationCIFAR-10 long-tail (test)
Accuracy (All)86.8
77
Model CalibrationTiny ImageNet (test)
ECE2.04
57
Skin lesion classificationISIC Asymmetric Noise 2019
Balanced Accuracy66.3
32
Skin lesion classificationISIC Symmetric Noise 2019
Balanced Accuracy59.9
32
Showing 10 of 39 rows

Other info

Follow for update