English

Using Pre-Training Can Improve Model Robustness and Uncertainty

Machine Learning 2019-10-22 v5 Computer Vision and Pattern Recognition Machine Learning

Abstract

He et al. (2018) have called into question the utility of pre-training by showing that training from scratch can often yield similar performance to pre-training. We show that although pre-training may not improve performance on traditional classification metrics, it improves model robustness and uncertainty estimates. Through extensive experiments on adversarial examples, label corruption, class imbalance, out-of-distribution detection, and confidence calibration, we demonstrate large gains from pre-training and complementary effects with task-specific methods. We introduce adversarial pre-training and show approximately a 10% absolute improvement over the previous state-of-the-art in adversarial robustness. In some cases, using pre-training without task-specific methods also surpasses the state-of-the-art, highlighting the need for pre-training when evaluating future methods on robustness and uncertainty tasks.

Keywords

Cite

@article{arxiv.1901.09960,
  title  = {Using Pre-Training Can Improve Model Robustness and Uncertainty},
  author = {Dan Hendrycks and Kimin Lee and Mantas Mazeika},
  journal= {arXiv preprint arXiv:1901.09960},
  year   = {2019}
}

Comments

ICML 2019. PyTorch code here: https://github.com/hendrycks/pre-training Figure 3 updated

R2 v1 2026-06-23T07:24:43.691Z