English

Locally Adaptive Label Smoothing for Predictive Churn

Machine Learning 2021-06-15 v2 Artificial Intelligence

Abstract

Training modern neural networks is an inherently noisy process that can lead to high \emph{prediction churn} -- disagreements between re-trainings of the same model due to factors such as randomization in the parameter initialization and mini-batches -- even when the trained models all attain similar accuracies. Such prediction churn can be very undesirable in practice. In this paper, we present several baselines for reducing churn and show that training on soft labels obtained by adaptively smoothing each example's label based on the example's neighboring labels often outperforms the baselines on churn while improving accuracy on a variety of benchmark classification tasks and model architectures.

Keywords

Cite

@article{arxiv.2102.05140,
  title  = {Locally Adaptive Label Smoothing for Predictive Churn},
  author = {Dara Bahri and Heinrich Jiang},
  journal= {arXiv preprint arXiv:2102.05140},
  year   = {2021}
}

Comments

ICML 2021

R2 v1 2026-06-23T22:59:59.814Z