A unified framework for information-theoretic generalization bounds
Yifeng Chu, Maxim Raginsky
Abstract
This paper presents a general methodology for deriving information-theoretic generalization bounds for learning algorithms. The main technical tool is a probabilistic decorrelation lemma based on a change of measure and a relaxation of Young's inequality in Orlicz spaces. Using the decorrelation lemma in combination with other techniques, such as symmetrization, couplings, and chaining in the space of probability measures, we obtain new upper bounds on the generalization error, both in expectation and in high probability, and recover as special cases many of the existing generalization bounds, including the ones based on mutual information, conditional mutual information, stochastic chaining, and PAC-Bayes inequalities. In addition, the Fernique-Talagrand upper bound on the expected supremum of a subgaussian process emerges as a special case.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 2857cf2c-acfe-4fd8-bd42-c06a25aa88c8Cited by top-tier papers1
Ask how each one uses itBuilds on6
- Sharpened Generalization Bounds based on Conditional Mutual Information and an Application to Noisy, Iterative AlgorithmsMahdi Haghifam, Jeffrey Negrea, Ashish Khisti, Daniel M. Roy et al.NeurIPS 2020 · 124 citations
- Conditioning and Processing: Techniques to Improve Information-Theoretic Generalization BoundsHassan Hafez-Kolahi, Zeinab Golgooni, Shohreh Kasaei, Mahdieh SoleymaniNeurIPS 2020 · 63 citations
- Information-theoretic generalization bounds for black-box learning algorithmsHrayr Harutyunyan, Maxim Raginsky, Greg Ver Steeg, Aram GalstyanNeurIPS 2021 · 61 citations
- Tighter Expected Generalization Error Bounds via Wasserstein DistanceBorja Rodríguez Gálvez, Germán Bassi, Ragnar Thobaben, Mikael SkoglundNeurIPS 2021 · 52 citations
- Towards a Unified Information-Theoretic Framework for GeneralizationMahdi Haghifam, Gintare Karolina Dziugaite, Shay Moran, Daniel M. RoyNeurIPS 2021 · 38 citations
Related papers
- Tighter Information-Theoretic Generalization Bounds from SupersamplesZiqiao Wang, Yongyi MaoICML 2023 · 23 citations
- Integral Probability Metrics PAC-Bayes BoundsRon Amit, Baruch Epstein, Shay Moran, Ron MeirNeurIPS 2022 · 25 citations
- Generalization of Hamiltonian algorithmsAndreas MaurerNeurIPS 2024 · 2 citations
- Generalization Bounds via Conditional f-InformationZiqiao Wang, Yongyi MaoNeurIPS 2024 · 4 citations
- An Exact Characterization of the Generalization Error for the Gibbs AlgorithmGholamali Aminian, Yuheng Bu, Laura Toni, Miguel R. D. Rodrigues et al.NeurIPS 2021 · 75 citations
