PeerJ Computer Science (Apr 2024)

Learning label smoothing for text classification

  • Han Ren,
  • Yajie Zhao,
  • Yong Zhang,
  • Wei Sun

DOI
https://doi.org/10.7717/peerj-cs.2005
Journal volume & issue
Vol. 10
p. e2005

Abstract

Read online Read online

Training with soft labels instead of hard labels can effectively improve the robustness and generalization of deep learning models. Label smoothing often provides uniformly distributed soft labels during the training process, whereas it does not take the semantic difference of labels into account. This article introduces discrimination-aware label smoothing, an adaptive label smoothing approach that learns appropriate distributions of labels for iterative optimization objectives. In this approach, positive and negative samples are employed to provide experience from both sides, and the performances of regularization and model calibration are improved through an iterative learning method. Experiments on five text classification datasets demonstrate the effectiveness of the proposed method.

Keywords