Information Complexity of Stochastic Convex Optimization: Applications to Generalization and Memorization
Not on the map. Why?
Abstract (arXiv)
In this work, we investigate the interplay between memorization and learning in the context of stochastic convex optimization (SCO). We define memorization via the information a learning algorithm reveals about its training data points. We then quantify this information using the framework of conditional mutual information (CMI) proposed by Steinke and Zakynthinou (2020). Our main result is a precise characterization of the tradeoff between the accuracy of a learning algorithm and its CMI, answering an open question posed by Livni (2023). We show that, in the L2 Lipschitz--bounded setting and under strong convexity, every learner with an excess error varepsilon has CMI bounded below by Omega(1/varepsilon2) and Omega(1/varepsilon), respectively. We further demonstrate the essential role of memorization in learning problems in SCO by designing an adversary capable of accurately identifying a significant fraction of the training samples in specific SCO problems. Finally, we enumerate several implications of our results, such as a limitation of generalization bounds based on CMI and the incompressibility of samples in SCO problems.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Builds on 52 of its 68 references that are in the map. First posted on arXiv 14 Feb 2024.
Builds on
52 of 68 references.
References · 52
- 2016Membership Inference Attacks against Machine Learning Modelscited by 2,221
- 2020Extracting Training Data from Large Language Modelscited by 2,144
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
- 2018The Secret Sharer: Evaluating and Testing Unintended Memorization in Neural Networkscited by 847
- 2021Membership Inference Attacks From First Principlescited by 839
- 2022Quantifying Memorization Across Neural Language Modelscited by 755
- 2020What Neural Networks Memorize and Why: Discovering the Long Tail via Influence Estimationcited by 492
- 2019Does Learning Require Memorization? A Short Tale about a Long Tailcited by 452
- 2017Information-theoretic analysis of generalization capability of learning algorithmscited by 299
- 2021User-friendly introduction to PAC-Bayes boundscited by 234
- 2020Stability of Stochastic Gradient Descent on Nonsmooth Convex Lossescited by 172
- 2020Reasoning About Generalization via Conditional Mutual Informationcited by 145
38 more
- 2022Reconstructing Training Data from Trained Neural Networkscited by 129
- 2019Information-Theoretic Generalization Bounds for SGLD via Data-Dependent Estimatescited by 118
- 2019Tightening Mutual Information Based Bounds on Generalization Errorcited by 117
- 2019A Short Note on Concentration Inequalities for Random Vectors with SubGaussian Normcited by 105
- 2018Privately Learning High-Dimensional Distributionscited by 103
- 2020When is Memorization of Irrelevant Training Data Necessary for High-Accuracy Learning?cited by 102
- 2019High probability generalization bounds for uniformly stable algorithms with nearly optimal ratecited by 97
- 2015How much does your data exploration overfit? Controlling bias via information usagecited by 95
- 2020Sharpened Generalization Bounds based on Conditional Mutual Information and an Application to Noisy, Iterative Algorithmscited by 89
- 2019VC Classes are Adversarially Robustly Learnable, but Only Improperlycited by 83
- 2013Fingerprinting Codes and the Price of Approximate Differential Privacycited by 81
- 2022A Characterization of Multiclass Learnabilitycited by 66
- 2023Generalization Bounds: Perspectives from Information Theory and PAC-Bayescited by 60
- 2020Proper Learning, Helly Number, and an Optimal SVM Boundcited by 52
- 2021Information-theoretic generalization bounds for black-box learning algorithmscited by 49
- 2015Sample compression schemes for VC classescited by 46
- 2021SGD Generalizes Better Than GD (And Regularization Doesn't Help)cited by 38
- 2016Generalization of ERM in Stochastic Convex Optimization: The Dimension Strikes Backcited by 37
- 2021PAC-Bayes, MAC-Bayes and Conditional Mutual Information: Fast rate bounds that handle general VC classescited by 31
- 2020Individually Conditional Individual Mutual Information Bound on Generalization Errorcited by 30
- 2021Towards a Unified Information-Theoretic Framework for Generalizationcited by 29
- 2020On Random Subset Generalization Error Bounds and the Stochastic Gradient Langevin Dynamics Algorithmcited by 28
- 2020Reducing Adversarially Robust Learning to Non-Robust PAC Learningcited by 28
- 2020A Limitation of the PAC-Bayes Frameworkcited by 24
- 2021Adversarially Robust Learning with Unknown Perturbation Setscited by 24
- 2023Optimal Learners for Realizable Regression: PAC Learning and Online Learningcited by 23
- 2022Benign Underfitting of Stochastic Gradient Descentcited by 18
- 2022Limitations of Information-Theoretic Generalization Bounds for Gradient Descent Methods in Stochastic Convex Optimizationcited by 18
- 2022Adversarially Robust Learning: A Generic Minimax Optimal Learner and Characterizationcited by 17
- 2020Can Implicit Bias Explain Generalization? Stochastic Convex Optimization as a Case Studycited by 16
- 2022Understanding Generalization via Leave-One-Out Conditional Mutual Informationcited by 16
- 2023Information Theoretic Lower Bounds for Information Theoretic Upper Boundscited by 16
- 2021Never Go Full Batch (in Stochastic Convex Optimization)cited by 14
- 2014A Compression Technique for Analyzing Disagreement-Based Active Learningcited by 11
- 2022Evaluated CMI Bounds for Meta Learning: Tightness and Expressivenesscited by 11
- 2022Strong Memory Lower Bounds for Learning Natural Modelscited by 10
- 2017Nearest-Neighbor Sample Compression: Efficiency, Consistency, Infinite Dimensionscited by 7
- 2018Agnostic Sample Compression Schemes for Regressioncited by 7
50 most cited of 52 All 52 (JSON)
Built on by
Citing papers per year: 2024: 6; 2025: 3; 2026: 3.
12 papers.
- 2024Are we making progress in unlearning? Findings from the first NeurIPS unlearning competitioncited by 48
- 2025Does Generation Require Memorization? Creative Diffusion Models using Ambient Diffusioncited by 17
- 2024The Pitfalls of Memorization: When Memorization Hurts Generalizationcited by 16
- 2024Improved Localized Machine Unlearning Through the Lens of Memorizationcited by 6
- 2024The Sample Complexity of Gradient Descent in Stochastic Convex Optimizationcited by 5
- 2024Not Every Image is Worth a Thousand Words: Quantifying Originality in Stable Diffusioncited by 4
- 2024Provable Privacy Attacks on Trained Shallow Neural Networkscited by 4
- 2026Is your algorithm unlearning or untraining?cited by 3
- 2026Cram Less to Fit More: Training Data Pruning Improves Memorization of Factscited by 2
- 2025A Lightweight Method to Disrupt Memorized Sequences in LLMcited by 1
- 2025Lower Bounds for Public-Private Learning under Distribution Shiftcited by 0
- 2026All ERMs Can Fail in Stochastic Convex Optimization Lower Bounds in Linear Dimensioncited by 0
- 2026Cram Less to Fit More: Training Data Pruning Improves Memorization of Facts9 Apr 2026
- 2026Is your algorithm unlearning or untraining?9 Apr 2026
- 2026All ERMs Can Fail in Stochastic Convex Optimization Lower Bounds in Linear Dimension9 Feb 2026
- 2025Lower Bounds for Public-Private Learning under Distribution Shift23 Jul 2025
- 2025Does Generation Require Memorization? Creative Diffusion Models using Ambient Diffusion28 Feb 2025
- 2025A Lightweight Method to Disrupt Memorized Sequences in LLM7 Feb 2025
- 2024The Pitfalls of Memorization: When Memorization Hurts Generalization10 Dec 2024
- 2024Improved Localized Machine Unlearning Through the Lens of Memorization3 Dec 2024
- 2024Provable Privacy Attacks on Trained Shallow Neural Networks10 Oct 2024
- 2024Not Every Image is Worth a Thousand Words: Quantifying Originality in Stable Diffusion15 Aug 2024
- 2024Are we making progress in unlearning? Findings from the first NeurIPS unlearning competition13 Jun 2024
- 2024The Sample Complexity of Gradient Descent in Stochastic Convex Optimization7 Apr 2024