Why you don't overfit, and don't need Bayes if you only train for one epoch
Not on the map. Why?
Abstract (arXiv)
Here, we show that in the data-rich setting where you only train on each datapoint once (or equivalently, you only train for one epoch), standard "maximum likelihood" training optimizes the true data generating process (DGP) loss, which is equivalent to the test loss. Further, we show that the Bayesian model average optimizes the same objective, albeit while taking the expectation over uncertainty induced by finite data. As standard maximum likelihood training in the single-epoch setting optimizes the same objective as Bayesian inference, we argue that we do not expect Bayesian inference to offer any advantages in terms of overfitting or calibration in these settings. This explains the diminishing importance of Bayes in areas such as LLMs, which are often trained with one (or very few) epochs.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 16 of its 35 references that are in the map. First posted on arXiv 19 Nov 2024.
Builds on
16 of 35 references.
References · 16
- 2023GPT-4 Technical Reportcited by 18,396
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 533
- 2019Fantastic Generalization Measures and Where to Find Themcited by 513
- 2021Documenting Large Webtext Corpora: A Case Study on the Colossal Clean Crawled Corpuscited by 492
- 2019Deep Ensembles: A Loss Landscape Perspectivecited by 443
- 2023What's In My Big Data?cited by 129
- 2023LoRA ensembles for large language model fine-tuningcited by 67
4 more
- 2024Position: Bayesian Deep Learning is Needed in the Age of Large-Scale AIcited by 62
- 2021Is MC Dropout Bayesian?cited by 35
- 2024Bayesian Reward Models for LLM Alignmentcited by 35
- 2020Variational Laplace for Bayesian neural networkscited by 1
Built on by
Citing papers per year: 2025: 2.
2 papers.