Reasoning About Generalization via Conditional Mutual Information
Not on the map. Why?
Abstract (arXiv)
We provide an information-theoretic framework for studying the generalization properties of machine learning algorithms. Our framework ties together existing approaches, including uniform convergence bounds and recent methods for adaptive data analysis. Specifically, we use Conditional Mutual Information (CMI) to quantify how well the input (i.e., the training data) can be recognized given the output (i.e., the trained model) of the learning algorithm. We show that bounds on CMI can be obtained from VC dimension, compression schemes, differential privacy, and other methods. We then show that bounded CMI implies various forms of generalization.
Key facts. Cited by 145 papers in the map (data through 7 Oct 2026). First posted on arXiv 24 Jan 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 25; 2022: 33; 2023: 26; 2024: 15; 2025: 25; 2026: 21.
145 papers.
Sort:
By area:
- 2021User-friendly introduction to PAC-Bayes boundscited by 234
- 2023Privacy Auditing with One (1) Training Runcited by 119
- 2023How Does Information Bottleneck Help Deep Learning?cited by 92
- 2021Invariant Information Bottleneck for Domain Generalizationcited by 86
- 2021Information-Theoretic Generalization Bounds for Stochastic Gradient Descentcited by 84
- 2023To Compress or Not to Compress- Self-Supervised Learning and Information Theory: A Reviewcited by 72
- 2025How much do language models memorize?cited by 53
- 2021Information-theoretic generalization bounds for black-box learning algorithmscited by 49
- 2021Generalization Bounds For Meta-Learning: An Information-Theoretic Analysiscited by 45
- 2021Tighter expected generalization error bounds via Wasserstein distancecited by 43
- 2023Stability is Stable: Connections between Replicability, Privacy, and Adaptive Generalizationcited by 42
- 2021PAC-Bayes, MAC-Bayes and Conditional Mutual Information: Fast rate bounds that handle general VC classescited by 31
38 more
- 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
- 2022Generalization Bounds via Convex Analysiscited by 27
- 2022Single Trajectory Nonparametric Learning of Nonlinear Dynamicscited by 26
- 2023An Information-Theoretic Perspective on Variance-Invariance-Covariance Regularizationcited by 25
- 2024Knowledge Overshadowing Causes Amalgamated Hallucination in Large Language Modelscited by 24
- 2022Rate-Distortion Theoretic Generalization Bounds for Stochastic Learning Algorithmscited by 22
- 2024Mutual Information Estimation via Normalizing Flowscited by 22
- 2023Tighter Information-Theoretic Generalization Bounds from Supersamplescited by 21
- 2023Information Bottleneck Analysis of Deep Neural Networks via Lossy Compressioncited by 19
- 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
- 2023More PAC-Bayes bounds: From bounded losses, to losses with general tail behaviors, to anytime validitycited by 17
- 2025Does Generation Require Memorization? Creative Diffusion Models using Ambient Diffusioncited by 17
- 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
- 2022What Do We Maximize in Self-Supervised Learning?cited by 15
- 2022Information-Theoretic Analysis of Unsupervised Domain Adaptationcited by 15
- 2023Generalization Bounds with Data-dependent Fractal Dimensionscited by 15
- 2021Information Complexity and Generalization Boundscited by 14
- 2021Characterizing and Understanding the Generalization Error of Transfer Learning with Gibbs Algorithmcited by 14
- 2022Rate-Distortion Theoretic Bounds on Generalization Error for Distributed Learningcited by 14
- 2022Generalization Error Bounds for Deep Neural Networks Trained by SGDcited by 14
- 2023Online-to-PAC Conversions: Generalization Bounds via Regret Analysiscited by 14
- 2025A Causal Adjustment Module for Debiasing Scene Graph Generationcited by 14
- 2021Information-Theoretic Bounds on the Moments of the Generalization Error of Learning Algorithmscited by 12
- 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
- 2024Minimum Description Length and Generalization Guarantees for Representation Learningcited by 11
- 2021An Information-Theoretic Analysis of the Impact of Task Similarity on Meta-Learningcited by 10
- 2021Rate-Distortion Analysis of Minimum Excess Risk in Bayesian Learningcited by 10
- 2021Information-Theoretic Analysis of Epistemic Uncertainty in Bayesian Meta-learningcited by 10
50 most cited of 145 All 145 (JSON)
- 2026Information-Theoretic Analysis of Next-Token Prediction under Markovian Data28 Sep 2026
- 2026Drift and Dependence: Layer-wise Information-Theoretic Bounds for Replay-Based Continual Learning12 Aug 2026
- 2026Local Redundancy: An Information-Theoretic Measure of Plasticity from Synthetic Memorization15 Jul 2026
- 2026Is Randomness Necessary for Adaptive Data Analysis?8 Jul 2026
- 2026Predictability as a Fine-Grained Measure for Privacy18 Jun 2026
- 2026What Fits (Into Few Tokens) Doesn't Overfit: Compression and Generalization in ML Research Agents9 Jun 2026
- 2026Towards the Connection between Activation Sparsity and Flat Minima25 May 2026
- 2026Bounded-Rationality, Hedging, and Generalization14 May 2026
- 2026Information-Theoretic Generalization Bounds for Sequential Decision Making12 May 2026
- 2026A Hierarchical Sampling Framework for bounding the Generalization Error of Federated Learning5 May 2026
- 2026When Do Early-Exit Networks Generalize? A PAC-Bayesian Theory of Adaptive Depth17 Apr 2026
- 2026A Bayesian Information-Theoretic Approach to Data Attribution4 Apr 2026
38 more
- 2026Bound to Disagree: Generalization Bounds via Certifiable Surrogates26 Feb 2026
- 2026The geometry of invariant learning: an information-theoretic analysis of data augmentation and generalization16 Feb 2026
- 2026Block-Sample MAC-Bayes Generalization Bounds13 Feb 2026
- 2026TRACE: Theoretical Risk Attribution under Covariate-shift Effects11 Feb 2026
- 2026Tighter Information-Theoretic Generalization Bounds via a Novel Class of Change of Measure Inequalities8 Feb 2026
- 2026Performative Learning Theory4 Feb 2026
- 2026Hallucination is a Consequence of Space-Optimality: A Rate-Distortion Theorem for Membership Testing31 Jan 2026
- 2026Excess Description Length of Learning Generalizable Predictors8 Jan 2026
- 2026Leveraging Flatness to Improve Information-Theoretic Generalization Bounds for SGD4 Jan 2026
- 2025Unbiased Visual Reasoning with Controlled Visual Inputs19 Dec 2025
- 2025In Good GRACEs: Principled Teacher Selection for Knowledge Distillation4 Nov 2025
- 2025An Information-Theoretic Analysis of OOD Generalization in Meta-Reinforcement Learning27 Oct 2025
- 2025AI Agents as Universal Task Solvers14 Oct 2025
- 2025A Context-Aware Dual-Metric Framework for Confidence Estimation in Large Language Models1 Aug 2025
- 2025Tight Bounds for Answering Adaptively Chosen Concentrated Queries18 Jul 2025
- 2025Information-Theoretic Generalization Bounds of Replay-based Continual Learning16 Jul 2025
- 2025Stability, Complexity and Data-Dependent Worst-Case Generalization Bounds9 Jul 2025
- 2025MEL: Multi-level Ensemble Learning for Resource-Constrained Environments25 Jun 2025
- 2025Fairness Overfitting in Machine Learning: An Information-Theoretic Perspective9 Jun 2025
- 2025Trade-offs in Data Memorization via Strong Data Processing Inequalities2 Jun 2025
- 2025Generalization in VAE and Diffusion Models: A Unified Information-Theoretic Analysis1 Jun 2025
- 2025How much do language models memorize?30 May 2025
- 2025FSL-SAGE: Accelerating Federated Split Learning via Smashed Activation Gradient Estimation29 May 2025
- 2025Information-theoretic Generalization Analysis for VQ-VAEs: A Role of Latent Variables26 May 2025
- 2025Laplace Sample Information: Data Informativeness Through a Bayesian Lens21 May 2025
- 2025Multiview Representation Learning via Distributed Joint Latent Space Structuring25 Apr 2025
- 2025A Causal Adjustment Module for Debiasing Scene Graph Generation22 Mar 2025
- 2025Generalization in Federated Learning: A Conditional Mutual Information Framework6 Mar 2025
- 2025Heterogeneity Matters even More in Distributed Learning: Study from Generalization Perspective3 Mar 2025
- 2025Does Generation Require Memorization? Creative Diffusion Models using Ambient Diffusion28 Feb 2025
- 2025Generalization Guarantees for Representation Learning via Data-Dependent Gaussian Mixture Priors21 Feb 2025
- 2025Compositional Generalization from Learned Skills via CoT Training: A Theoretical and Structural Analysis for Reasoning7 Feb 2025
- 2025Generalization Error Analysis for Selective State-Space Models Through the Lens of Attention3 Feb 2025
- 2025InfoBridge: Mutual Information estimation via Bridge Matching3 Feb 2025
- 2024Understanding Generalization of Federated Learning: the Trade-off between Model Stability and Optimization25 Nov 2024
- 2024Generalization Bounds via Conditional f-Information30 Oct 2024
- 2024Theoretical Insights into Fine-Tuning Attention Mechanism: Generalization and Optimization3 Oct 2024
- 2024Generalization and Robustness of the Tilted Empirical Risk28 Sep 2024
50 newest of 145 All 145 (JSON)
- 2021User-friendly introduction to PAC-Bayes boundscited by 234
- 2023Privacy Auditing with One (1) Training Runcited by 119
- 2023How Does Information Bottleneck Help Deep Learning?cited by 92
- 2021Invariant Information Bottleneck for Domain Generalizationcited by 86
- 2021Information-Theoretic Generalization Bounds for Stochastic Gradient Descentcited by 84
- 2023To Compress or Not to Compress- Self-Supervised Learning and Information Theory: A Reviewcited by 72
- 2025How much do language models memorize?cited by 53
- 2021Information-theoretic generalization bounds for black-box learning algorithmscited by 49
- 2021Generalization Bounds For Meta-Learning: An Information-Theoretic Analysiscited by 45
- 2021Tighter expected generalization error bounds via Wasserstein distancecited by 43
- 2023Stability is Stable: Connections between Replicability, Privacy, and Adaptive Generalizationcited by 42
- 2021PAC-Bayes, MAC-Bayes and Conditional Mutual Information: Fast rate bounds that handle general VC classescited by 31
12 most cited of 138 All 138 (JSON)
- 2026Information-Theoretic Analysis of Next-Token Prediction under Markovian Data28 Sep 2026
- 2026Drift and Dependence: Layer-wise Information-Theoretic Bounds for Replay-Based Continual Learning12 Aug 2026
- 2026Local Redundancy: An Information-Theoretic Measure of Plasticity from Synthetic Memorization15 Jul 2026
- 2026Is Randomness Necessary for Adaptive Data Analysis?8 Jul 2026
- 2026Predictability as a Fine-Grained Measure for Privacy18 Jun 2026
- 2026Towards the Connection between Activation Sparsity and Flat Minima25 May 2026
- 2026Bounded-Rationality, Hedging, and Generalization14 May 2026
- 2026Information-Theoretic Generalization Bounds for Sequential Decision Making12 May 2026
- 2026A Hierarchical Sampling Framework for bounding the Generalization Error of Federated Learning5 May 2026
- 2026A Bayesian Information-Theoretic Approach to Data Attribution4 Apr 2026
- 2026Bound to Disagree: Generalization Bounds via Certifiable Surrogates26 Feb 2026
- 2026The geometry of invariant learning: an information-theoretic analysis of data augmentation and generalization16 Feb 2026
12 newest of 138 All 138 (JSON)