Sharpened Generalization Bounds based on Conditional Mutual Information and an Application to Noisy, Iterative Algorithms
Not on the map. Why?
Abstract (arXiv)
The information-theoretic framework of Russo and J. Zou (2016) and Xu and Raginsky (2017) provides bounds on the generalization error of a learning algorithm in terms of the mutual information between the algorithm's output and the training sample. In this work, we study the proposal, by Steinke and Zakynthinou (2020), to reason about the generalization error of a learning algorithm by introducing a super sample that contains the training sample as a random subset and computing mutual information conditional on the super sample. We first show that these new bounds based on the conditional mutual information are tighter than those based on the unconditional mutual information. We then introduce yet tighter bounds, building on the "individual sample" idea of Bu, S. Zou, and Veeravalli (2019) and the "data dependent" ideas of Negrea et al. (2019), using disintegrated mutual information. Finally, we apply these bounds to the study of Langevin dynamics algorithm, showing that conditioning on the super sample allows us to exploit information in the optimization trajectory to obtain tighter bounds based on hypothesis tests.
Key facts. Cited by 89 papers in the map (data through 7 Oct 2026). First posted on arXiv 27 Apr 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 20; 2022: 27; 2023: 15; 2024: 8; 2025: 14; 2026: 5.
89 papers.
Sort:
By area:
- 2021Information-Theoretic Generalization Bounds for Stochastic Gradient Descentcited by 84
- 2025How much do language models memorize?cited by 53
- 2021Information-theoretic generalization bounds for black-box learning algorithmscited by 49
- 2021Tighter expected generalization error bounds via Wasserstein distancecited by 43
- 2021PAC-Bayes, MAC-Bayes and Conditional Mutual Information: Fast rate bounds that handle general VC classescited by 31
- 2021On the Generalization of Models Trained with SGD: Information-Theoretic Bounds and Implicationscited by 31
- 2021Towards a Unified Information-Theoretic Framework for Generalizationcited by 29
- 2022A New Family of Generalization Bounds Using Samplewise Evaluated CMIcited by 28
- 2021Generalization Bounds using Lower Tail Exponents in Stochastic Optimizerscited by 27
- 2021Time-independent Generalization Bounds for SGLD in Non-convex Settingscited by 27
- 2022Generalization Bounds via Convex Analysiscited by 27
- 2023Explaining Generalization Power of a DNN Using Interactive Conceptscited by 25
38 more
- 2022Rate-Distortion Theoretic Generalization Bounds for Stochastic Learning Algorithmscited by 22
- 2023Tighter Information-Theoretic Generalization Bounds from Supersamplescited by 21
- 2022An Information-theoretical Approach to Semi-supervised Learning under Covariate-shiftcited by 18
- 2022Limitations of Information-Theoretic Generalization Bounds for Gradient Descent Methods in Stochastic Convex Optimizationcited by 18
- 2022Understanding Generalization via Leave-One-Out Conditional Mutual Informationcited by 16
- 2023Information Theoretic Lower Bounds for Information Theoretic Upper Boundscited by 16
- 2021Generalization Bounds for Noisy Iterative Algorithms Using Properties of Additive Noise Channelscited by 15
- 2022Information-Theoretic Analysis of Unsupervised Domain Adaptationcited by 15
- 2021Information Complexity and Generalization Boundscited by 14
- 2021Characterizing and Understanding the Generalization Error of Transfer Learning with Gibbs Algorithmcited by 14
- 2022Generalization Error Bounds for Deep Neural Networks Trained by SGDcited by 14
- 2022Empirical Risk Minimization with Relative Entropy Regularizationcited by 13
- 2024Information Complexity of Stochastic Convex Optimization: Applications to Generalization and Memorizationcited by 12
- 2021Information-Theoretic Characterization of the Generalization Error for Iterative Semi-Supervised Learningcited by 11
- 2022Tighter Expected Generalization Error Bounds via Convexity of Information Measurescited by 11
- 2022On the Generalization for Transfer Learning: An Information-Theoretic Analysiscited by 11
- 2022Evaluated CMI Bounds for Meta Learning: Tightness and Expressivenesscited by 11
- 2022Two Facets of SDE Under an Information-Theoretic Lens: Generalization of SGD via Training Trajectories and via Terminal Statescited by 11
- 2021Rate-Distortion Analysis of Minimum Excess Risk in Bayesian Learningcited by 10
- 2022Chained Generalisation Boundscited by 10
- 2022What You See is What You Get: Principled Deep Learning via Distributional Generalizationcited by 10
- 2023On the Validation of Gibbs Algorithms: Training Datasets, Test Datasets and their Aggregationcited by 10
- 2024Information-Theoretic Generalization Bounds for Deep Neural Networkscited by 10
- 2021Disparity Between Batches as a Signal for Early Stoppingcited by 9
- 2022Information bottleneck theory of high-dimensional regression: relevancy, efficiency and optimalitycited by 9
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Setscited by 9
- 2022Stability Based Generalization Bounds for Exponential Family Langevin Dynamicscited by 8
- 2022On Leave-One-Out Conditional Mutual Information For Generalizationcited by 8
- 2022Weight Expansion: A New Perspective on Dropout and Generalizationcited by 6
- 2022Fast Rate Generalization Error Bounds: Variations on a Themecited by 6
- 2022Learning Algorithm Generalization Error Bounds via Auxiliary Distributionscited by 6
- 2022On the Importance of Gradient Norm in PAC-Bayesian Boundscited by 6
- 2023A unified framework for information-theoretic generalization boundscited by 6
- 2021Towards Understanding Generalization via Decomposing Excess Risk Dynamicscited by 5
- 2022Information-Theoretic Analysis of Minimax Excess Riskcited by 5
- 2022Generalization Bounds for Gradient Methods via Discrete and Continuous Priorcited by 5
- 2022How Does Pseudo-Labeling Affect the Generalization Error of the Semi-Supervised Gibbs Algorithm?cited by 5
- 2023Fast Rate Information-theoretic Bounds on Generalization Errorscited by 5
50 most cited of 89 All 89 (JSON)
- 2026Drift and Dependence: Layer-wise Information-Theoretic Bounds for Replay-Based Continual Learning12 Aug 2026
- 2026Bounded-Rationality, Hedging, and Generalization14 May 2026
- 2026Information-Theoretic Generalization Bounds for Sequential Decision Making12 May 2026
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networks26 Feb 2026
- 2026Tighter Information-Theoretic Generalization Bounds via a Novel Class of Change of Measure Inequalities8 Feb 2026
- 2025An Information-Theoretic Analysis of OOD Generalization in Meta-Reinforcement Learning27 Oct 2025
- 2025Information Theory in Open-world Machine Learning Foundations, Frameworks, and Future Direction17 Oct 2025
- 2025Auditing Information Disclosure During Large-Scale Gradient-Based Training via Gradient Uniqueness13 Oct 2025
- 2025Beyond Real Data: Synthetic Data through the Lens of Regularization9 Oct 2025
- 2025Information-Theoretic Generalization Bounds of Replay-based Continual Learning16 Jul 2025
- 2025MEL: Multi-level Ensemble Learning for Resource-Constrained Environments25 Jun 2025
- 2025Algorithm- and Data-Dependent Generalization Bounds for Diffusion Models4 Jun 2025
38 more
- 2025How much do language models memorize?30 May 2025
- 2025Generalization Bounds for Quantum Learning via R'enyi Divergences16 May 2025
- 2025Generalization in Federated Learning: A Conditional Mutual Information Framework6 Mar 2025
- 2025Stability-based Generalization Bounds for Variational Inference17 Feb 2025
- 2025Revisiting Generalization Power of a DNN in Terms of Symbolic Interactions14 Feb 2025
- 2025Generalization Bounds for Markov Algorithms through Entropy Flow Computations11 Feb 2025
- 2025Generalization Error Analysis for Selective State-Space Models Through the Lens of Attention3 Feb 2025
- 2024The Method of Gaps: Exact Expressions for the Generalization Error of Supervised Learning Algorithms18 Nov 2024
- 2024Generalization Bounds via Conditional f-Information30 Oct 2024
- 2024Error Bounds of Supervised Classification from Information-Theoretic Perspective7 Jun 2024
- 2024Enhancing In-Context Learning Performance with just SVD-Based Weight Pruning: A Theoretical Perspective6 Jun 2024
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Sets26 Apr 2024
- 2024Information-Theoretic Generalization Bounds for Deep Neural Networks4 Apr 2024
- 2024An Information-Theoretic Framework for Out-of-Distribution Generalization with Applications to Stochastic Gradient Langevin Dynamics29 Mar 2024
- 2024Information Complexity of Stochastic Convex Optimization: Applications to Generalization and Memorization14 Feb 2024
- 2023Information-Theoretic Generalization Bounds for Transductive Learning and its Applications8 Nov 2023
- 2023Sample-Conditioned Hypothesis Stability Sharpens Information-Theoretic Generalization Bounds31 Oct 2023
- 2023Comparing Comparators in Generalization Bounds16 Oct 2023
- 2023Generalization error bounds for iterative learning algorithms with bounded updates10 Sep 2023
- 2023On the Validation of Gibbs Algorithms: Training Datasets, Test Datasets and their Aggregation21 Jun 2023
- 2023On the Role of Noise in the Sample Complexity of Learning Recurrent Neural Networks: Exponential Gaps for Long Sequences28 May 2023
- 2023A unified framework for information-theoretic generalization bounds18 May 2023
- 2023Exactly Tight Information-Theoretic Generalization Error Bound for the Quadratic Gaussian Problem1 May 2023
- 2023Recognizable Information Bottleneck28 Apr 2023
- 2023Learning Trajectories are Generalization Indicators25 Apr 2023
- 2023Fast Rate Information-theoretic Bounds on Generalization Errors26 Mar 2023
- 2023Generalization Error Bounds for Noisy, Iterative Algorithms via Maximal Leakage28 Feb 2023
- 2023Explaining Generalization Power of a DNN Using Interactive Concepts25 Feb 2023
- 2023Information Theoretic Lower Bounds for Information Theoretic Upper Bounds9 Feb 2023
- 2023Tighter Information-Theoretic Generalization Bounds from Supersamples5 Feb 2023
- 2022Limitations of Information-Theoretic Generalization Bounds for Gradient Descent Methods in Stochastic Convex Optimization27 Dec 2022
- 2022Two Facets of SDE Under an Information-Theoretic Lens: Generalization of SGD via Training Trajectories and via Terminal States19 Nov 2022
- 2022Empirical Risk Minimization with Relative Entropy Regularization12 Nov 2022
- 2022How Does Pseudo-Labeling Affect the Generalization Error of the Semi-Supervised Gibbs Algorithm?15 Oct 2022
- 2022Evaluated CMI Bounds for Meta Learning: Tightness and Expressiveness12 Oct 2022
- 2022A New Family of Generalization Bounds Using Samplewise Evaluated CMI12 Oct 2022
- 2022On the Importance of Gradient Norm in PAC-Bayesian Bounds12 Oct 2022
- 2022Information-Theoretic Analysis of Unsupervised Domain Adaptation3 Oct 2022
50 newest of 89 All 89 (JSON)
- 2021Information-Theoretic Generalization Bounds for Stochastic Gradient Descentcited by 84
- 2025How much do language models memorize?cited by 53
- 2021Information-theoretic generalization bounds for black-box learning algorithmscited by 49
- 2021Tighter expected generalization error bounds via Wasserstein distancecited by 43
- 2021PAC-Bayes, MAC-Bayes and Conditional Mutual Information: Fast rate bounds that handle general VC classescited by 31
- 2021On the Generalization of Models Trained with SGD: Information-Theoretic Bounds and Implicationscited by 31
- 2021Towards a Unified Information-Theoretic Framework for Generalizationcited by 29
- 2022A New Family of Generalization Bounds Using Samplewise Evaluated CMIcited by 28
- 2021Generalization Bounds using Lower Tail Exponents in Stochastic Optimizerscited by 27
- 2021Time-independent Generalization Bounds for SGLD in Non-convex Settingscited by 27
- 2022Generalization Bounds via Convex Analysiscited by 27
- 2023Explaining Generalization Power of a DNN Using Interactive Conceptscited by 25
12 most cited of 87 All 87 (JSON)
- 2026Drift and Dependence: Layer-wise Information-Theoretic Bounds for Replay-Based Continual Learning12 Aug 2026
- 2026Bounded-Rationality, Hedging, and Generalization14 May 2026
- 2026Information-Theoretic Generalization Bounds for Sequential Decision Making12 May 2026
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networks26 Feb 2026
- 2026Tighter Information-Theoretic Generalization Bounds via a Novel Class of Change of Measure Inequalities8 Feb 2026
- 2025An Information-Theoretic Analysis of OOD Generalization in Meta-Reinforcement Learning27 Oct 2025
- 2025Information Theory in Open-world Machine Learning Foundations, Frameworks, and Future Direction17 Oct 2025
- 2025Auditing Information Disclosure During Large-Scale Gradient-Based Training via Gradient Uniqueness13 Oct 2025
- 2025Beyond Real Data: Synthetic Data through the Lens of Regularization9 Oct 2025
- 2025Information-Theoretic Generalization Bounds of Replay-based Continual Learning16 Jul 2025
- 2025MEL: Multi-level Ensemble Learning for Resource-Constrained Environments25 Jun 2025
- 2025How much do language models memorize?30 May 2025
12 newest of 87 All 87 (JSON)