PAC-Bayes Generalisation Bounds for Heavy-Tailed Losses through Supermartingales
Haddouche, Maxime, Guedj, Benjamin
–arXiv.org Artificial Intelligence
While PAC-Bayes is now an established learning framework for light-tailed losses (\emph{e.g.}, subgaussian or subexponential), its extension to the case of heavy-tailed losses remains largely uncharted and has attracted a growing interest in recent years. We contribute PAC-Bayes generalisation bounds for heavy-tailed losses under the sole assumption of bounded variance of the loss function. Under that assumption, we extend previous results from \citet{kuzborskij2019efron}. Our key technical contribution is exploiting an extention of Markov's inequality for supermartingales. Our proof technique unifies and extends different PAC-Bayesian frameworks by providing bounds for unbounded martingales as well as bounds for batch and online learning with heavy-tailed losses.
arXiv.org Artificial Intelligence
Apr-24-2023
- Country:
- North America > United States
- Washington > King County
- Bellevue (0.04)
- New York > New York County
- New York City (0.04)
- Washington > King County
- Europe > United Kingdom
- England > Cambridgeshire > Cambridge (0.04)
- North America > United States
- Genre:
- Research Report (0.64)
- Industry:
- Education (0.34)