English

PAC-Bayes Un-Expected Bernstein Inequality

Machine Learning 2021-12-16 v2 Machine Learning

Abstract

We present a new PAC-Bayesian generalization bound. Standard bounds contain a Ln\KL/n\sqrt{L_n \cdot \KL/n} complexity term which dominates unless LnL_n, the empirical error of the learning algorithm's randomized predictions, vanishes. We manage to replace LnL_n by a term which vanishes in many more situations, essentially whenever the employed learning algorithm is sufficiently stable on the dataset at hand. Our new bound consistently beats state-of-the-art bounds both on a toy example and on UCI datasets (with large enough nn). Theoretically, unlike existing bounds, our new bound can be expected to converge to 00 faster whenever a Bernstein/Tsybakov condition holds, thus connecting PAC-Bayesian generalization and {\em excess risk\/} bounds---for the latter it has long been known that faster convergence can be obtained under Bernstein conditions. Our main technical tool is a new concentration inequality which is like Bernstein's but with X2X^2 taken outside its expectation.

Keywords

Cite

@article{arxiv.1905.13367,
  title  = {PAC-Bayes Un-Expected Bernstein Inequality},
  author = {Zakaria Mhammedi and Peter D. Grunwald and Benjamin Guedj},
  journal= {arXiv preprint arXiv:1905.13367},
  year   = {2021}
}

Comments

24 pages, 6 figures. To Appear in NeurIPS2019

R2 v1 2026-06-23T09:34:20.171Z