Why you don't overfit, and don't need Bayes if you only train for one epoch

Fuente: arXiv
Saved in:
Bibliographic Details
Main Author: Aitchison, Laurence
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929600225869824
author Aitchison, Laurence
author_facet Aitchison, Laurence
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2411_14478
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Why you don't overfit, and don't need Bayes if you only train for one epoch
Aitchison, Laurence
Machine Learning
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.
title Why you don't overfit, and don't need Bayes if you only train for one epoch
topic Machine Learning
url https://arxiv.org/abs/2411.14478