A PAC-Bayesian Analysis of Randomized Learning with Application to Stochastic Gradient Descent
Not on the map. Why?
Abstract (arXiv)
We study the generalization error of randomized learning algorithms -- focusing on stochastic gradient descent (SGD) -- using a novel combination of PAC-Bayes and algorithmic stability. Importantly, our generalization bounds hold for all posterior distributions on an algorithm's random hyperparameters, including distributions that depend on the training data. This inspires an adaptive sampling algorithm for SGD that optimizes the posterior at runtime. We analyze this algorithm in the context of our generalization bounds and evaluate it on a benchmark dataset. Our experiments demonstrate that adaptive sampling can reduce empirical risk faster than uniform sampling while also improving out-of-sample accuracy.
Key facts. Cited by 29 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Sep 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 8; 2022: 7; 2023: 6; 2024: 5; 2025: 2; 2026: 1.
29 papers.
Sort:
- 2021User-friendly introduction to PAC-Bayes boundscited by 234
- 2021Heavy Tails in SGD and Compressibility of Overparametrized Neural Networkscited by 44
- 2021Stability and Generalization of Stochastic Gradient Methods for Minimax Problemscited by 35
- 2021On the Generalization of Models Trained with SGD: Information-Theoretic Bounds and Implicationscited by 31
- 2021Generalization Bounds for Meta-Learning via PAC-Bayes and Uniform Stabilitycited by 29
- 2021Variance Reduced Training with Stratified Sampling for Forecasting Modelscited by 13
- 2021PAC-Bayes Bounds for Meta-learning with Data-Dependent Priorcited by 11
- 2022Generalization Bounds for Stochastic Gradient Descent via Localized varepsilon-Coverscited by 11
- 2023Improved Stability and Generalization Guarantees of the Decentralized SGD Algorithmcited by 10
- 2021Optimal Rates for Random Order Online Optimizationcited by 9
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Setscited by 9
- 2022Uniform Stability for First-Order Empirical Risk Minimizationcited by 8
17 more
- 2022On the Importance of Gradient Norm in PAC-Bayesian Boundscited by 6
- 2022Generalization Bounds for Gradient Methods via Discrete and Continuous Priorcited by 5
- 2023High Probability Analysis for Non-Convex Stochastic Optimization with Clippingcited by 5
- 2023Sample-Conditioned Hypothesis Stability Sharpens Information-Theoretic Generalization Boundscited by 5
- 2022PAC-Bayesian Learning of Optimization Algorithmscited by 4
- 2024Generalization Error Matters in Decentralized Learning Under Byzantine Attackscited by 4
- 2023Enhance Diffusion to Improve Robust Generalizationcited by 3
- 2024Leveraging Gradients for Unsupervised Accuracy Estimation under Distribution Shiftcited by 3
- 2023Minimax Excess Risk of First-Order Methods for Statistical Learning with Data-Dependent Oraclescited by 2
- 2023Information-Theoretic Generalization Bounds for Transductive Learning and its Applicationscited by 2
- 2022Demystify Optimization and Generalization of Over-parameterized PAC-Bayesian Learningcited by 1
- 2022Super-model ecosystem: A domain-adaptation perspectivecited by 1
- 2024Convex SGD: Generalization Without Early Stoppingcited by 1
- 2024A Generalization Result for Convergence in Learning-to-Optimizecited by 1
- 2025Randomized Pairwise Learning with Adaptive Sampling: A PAC-Bayes Analysiscited by 1
- 2025Scaling Law Analysis in Federated Learning: How to Select the Optimal Model Size?cited by 0
- 2026Overfitting and Generalizing with (PAC) Bayesian Prediction in Noisy Binary Classificationcited by 0
- 2026Overfitting and Generalizing with (PAC) Bayesian Prediction in Noisy Binary Classification23 Mar 2026
- 2025Scaling Law Analysis in Federated Learning: How to Select the Optimal Model Size?15 Nov 2025
- 2025Randomized Pairwise Learning with Adaptive Sampling: A PAC-Bayes Analysis3 Apr 2025
- 2024A Generalization Result for Convergence in Learning-to-Optimize10 Oct 2024
- 2024Generalization Error Matters in Decentralized Learning Under Byzantine Attacks11 Jul 2024
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Sets26 Apr 2024
- 2024Leveraging Gradients for Unsupervised Accuracy Estimation under Distribution Shift17 Jan 2024
- 2024Convex SGD: Generalization Without Early Stopping8 Jan 2024
- 2023Information-Theoretic Generalization Bounds for Transductive Learning and its Applications8 Nov 2023
- 2023Sample-Conditioned Hypothesis Stability Sharpens Information-Theoretic Generalization Bounds31 Oct 2023
- 2023High Probability Analysis for Non-Convex Stochastic Optimization with Clipping25 Jul 2023
- 2023Minimax Excess Risk of First-Order Methods for Statistical Learning with Data-Dependent Oracles10 Jul 2023
17 more
- 2023Improved Stability and Generalization Guarantees of the Decentralized SGD Algorithm5 Jun 2023
- 2023Enhance Diffusion to Improve Robust Generalization5 Jun 2023
- 2022PAC-Bayesian Learning of Optimization Algorithms20 Oct 2022
- 2022On the Importance of Gradient Norm in PAC-Bayesian Bounds12 Oct 2022
- 2022Generalization Bounds for Stochastic Gradient Descent via Localized varepsilon-Covers19 Sep 2022
- 2022Super-model ecosystem: A domain-adaptation perspective30 Aug 2022
- 2022Uniform Stability for First-Order Empirical Risk Minimization17 Jul 2022
- 2022Generalization Bounds for Gradient Methods via Discrete and Continuous Prior27 May 2022
- 2022Demystify Optimization and Generalization of Over-parameterized PAC-Bayesian Learning4 Feb 2022
- 2021User-friendly introduction to PAC-Bayes bounds21 Oct 2021
- 2021On the Generalization of Models Trained with SGD: Information-Theoretic Bounds and Implications7 Oct 2021
- 2021Optimal Rates for Random Order Online Optimization29 Jun 2021
- 2021Heavy Tails in SGD and Compressibility of Overparametrized Neural Networks7 Jun 2021
- 2021Stability and Generalization of Stochastic Gradient Methods for Minimax Problems8 May 2021
- 2021Variance Reduced Training with Stratified Sampling for Forecasting Models2 Mar 2021
- 2021Generalization Bounds for Meta-Learning via PAC-Bayes and Uniform Stability12 Feb 2021
- 2021PAC-Bayes Bounds for Meta-learning with Data-Dependent Prior7 Feb 2021
- 2021User-friendly introduction to PAC-Bayes boundscited by 234
- 2021Heavy Tails in SGD and Compressibility of Overparametrized Neural Networkscited by 44
- 2021Stability and Generalization of Stochastic Gradient Methods for Minimax Problemscited by 35
- 2021On the Generalization of Models Trained with SGD: Information-Theoretic Bounds and Implicationscited by 31
- 2021Generalization Bounds for Meta-Learning via PAC-Bayes and Uniform Stabilitycited by 29
- 2021Variance Reduced Training with Stratified Sampling for Forecasting Modelscited by 13
- 2021PAC-Bayes Bounds for Meta-learning with Data-Dependent Priorcited by 11
- 2022Generalization Bounds for Stochastic Gradient Descent via Localized varepsilon-Coverscited by 11
- 2023Improved Stability and Generalization Guarantees of the Decentralized SGD Algorithmcited by 10
- 2021Optimal Rates for Random Order Online Optimizationcited by 9
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Setscited by 9
- 2022Uniform Stability for First-Order Empirical Risk Minimizationcited by 8
12 most cited of 29 All 29 (JSON)
- 2026Overfitting and Generalizing with (PAC) Bayesian Prediction in Noisy Binary Classification23 Mar 2026
- 2025Scaling Law Analysis in Federated Learning: How to Select the Optimal Model Size?15 Nov 2025
- 2025Randomized Pairwise Learning with Adaptive Sampling: A PAC-Bayes Analysis3 Apr 2025
- 2024A Generalization Result for Convergence in Learning-to-Optimize10 Oct 2024
- 2024Generalization Error Matters in Decentralized Learning Under Byzantine Attacks11 Jul 2024
- 2024Uniform Generalization Bounds on Data-Dependent Hypothesis Sets via PAC-Bayesian Theory on Random Sets26 Apr 2024
- 2024Leveraging Gradients for Unsupervised Accuracy Estimation under Distribution Shift17 Jan 2024
- 2024Convex SGD: Generalization Without Early Stopping8 Jan 2024
- 2023Information-Theoretic Generalization Bounds for Transductive Learning and its Applications8 Nov 2023
- 2023Sample-Conditioned Hypothesis Stability Sharpens Information-Theoretic Generalization Bounds31 Oct 2023
- 2023High Probability Analysis for Non-Convex Stochastic Optimization with Clipping25 Jul 2023
- 2023Minimax Excess Risk of First-Order Methods for Statistical Learning with Data-Dependent Oracles10 Jul 2023
12 newest of 29 All 29 (JSON)