Generalization Bounds for Meta-Learning via PAC-Bayes and Uniform Stability
Not on the map. Why?
Abstract (arXiv)
We are motivated by the problem of providing strong generalization guarantees in the context of meta-learning. Existing generalization bounds are either challenging to evaluate or provide vacuous guarantees in even relatively simple settings. We derive a probably approximately correct (PAC) bound for gradient-based meta-learning using two different generalization frameworks in order to deal with the qualitatively different challenges of generalization at the "base" and "meta" levels. We employ bounds for uniformly stable algorithms at the base level and bounds from the PAC-Bayes framework at the meta level. The result of this approach is a novel PAC bound that is tighter when the base learner adapts quickly, which is precisely the goal of meta-learning. We show that our bound provides a tighter guarantee than other bounds on a toy non-convex problem on the unit sphere and a text-based classification example. We also present a practical regularization scheme motivated by the bound in settings where the bound is loose and demonstrate improved performance over baseline techniques.
Key facts. Cited by 29 papers in the map (data through 7 Oct 2026). Builds on 38 of its 80 references that are in the map. First posted on arXiv 12 Feb 2021.
Builds on
38 of 80 references.
References · 38
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2014Explaining and Harnessing Adversarial Examplescited by 7,287
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2016Matching Networks for One Shot Learningcited by 2,246
- 2018On First-Order Meta-Learning Algorithmscited by 1,003
- 2020Meta-Learning in Neural Networks: A Surveycited by 944
- 2019Introduction to Online Convex Optimizationcited by 793
- 2017Spectrally-normalized margin bounds for neural networkscited by 681
- 2016CVXPY: A Python-Embedded Modeling Language for Convex Optimizationcited by 633
- 2017Exploring Generalization in Deep Learningcited by 595
- 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 547
- 2019A Theoretical Analysis of Contrastive Unsupervised Representation Learningcited by 530
26 more
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 527
- 2019Fantastic Generalization Measures and Where to Find Themcited by 507
- 2017Certifying Some Distributional Robustness with Principled Adversarial Trainingcited by 333
- 2017A PAC-Bayesian Approach to Spectrally-Normalized Margin Bounds for Neural Networkscited by 309
- 2018Stronger generalization bounds for deep nets via a compression approachcited by 306
- 2017A Rewriting System for Convex Optimization Problemscited by 178
- 2018Non-Vacuous Generalization Bounds at the ImageNet Scale: A PAC-Bayesian Compression Approachcited by 125
- 2020Tighter risk certificates for neural networkscited by 112
- 2019Meta-Learning without Memorizationcited by 92
- 2019Sharper bounds for uniformly stable algorithmscited by 91
- 2020PACOH: Bayes-Optimal Meta-Learning with PAC-Guaranteescited by 85
- 2017Meta-Learning by Adjusting Priors Based on Extended PAC-Bayes Theorycited by 76
- 2013A PAC-Bayesian bound for Lifelong Learningcited by 69
- 2020learn2learn: A Library for Meta-Learning Researchcited by 69
- 2019Provable Guarantees for Gradient-Based Meta-Learningcited by 67
- 2020PAC-Bayes Analysis Beyond the Usual Boundscited by 63
- 2016A Strongly Quasiconvex PAC-Bayesian Boundcited by 54
- 2013A PAC-Bayesian Tutorial with A Dropout Boundcited by 43
- 2020PAC-Bayes unleashed: generalisation bounds with unbounded lossescited by 43
- 2019PAC-Bayes with Backpropcited by 36
- 2017A PAC-Bayesian Analysis of Randomized Learning with Application to Stochastic Gradient Descentcited by 29
- 2018PAC-Bayes bounds for stable algorithms with instance-dependent priorscited by 21
- 2016Regret Bounds for Lifelong Learningcited by 19
- 2016Stability revisited: new generalisation bounds for the Leave-one-Outcited by 10
- 2019An Exponential Efron-Stein Inequality for Lq Stable Learning Rulescited by 6
- 2013On Learnability, Complexity and Stabilitycited by 4
Built on by
Citing papers per year: 2021: 1; 2022: 9; 2023: 7; 2024: 6; 2025: 5; 2026: 1.
29 papers.
Sort:
- 2022Sharp-MAML: Sharpness-Aware Model-Agnostic Meta Learningcited by 57
- 2023Exponential Smoothing for Off-Policy Learningcited by 15
- 2022Meta-Learning Priors for Safe Bayesian Optimizationcited by 14
- 2022Is Bayesian Model-Agnostic Meta Learning Better than Model-Agnostic Meta Learning, Provably?cited by 13
- 2022Understanding Benign Overfitting in Gradient-Based Meta Learningcited by 11
- 2022Evaluated CMI Bounds for Meta Learning: Tightness and Expressivenesscited by 11
- 2022Scalable PAC-Bayesian Meta-Learning via the PAC-Optimal Hyper-Posterior: From Theory to Practicecited by 11
- 2023Learning via Wasserstein-Based High Probability Generalisation Boundscited by 11
- 2024More Flexible PAC-Bayesian Meta-Learning by Learning Learning Algorithmscited by 10
- 2021Learning an Explicit Hyperparameter Prediction Function Conditioned on Taskscited by 9
- 2023Bayes meets Bernstein at the Meta Level: an Analysis of Fast Rates in Meta-Learning with PAC-Bayescited by 8
- 2022On the Importance of Gradient Norm in PAC-Bayesian Boundscited by 6
17 more
- 2024Tight PAC-Bayesian Risk Certificates for Contrastive Learningcited by 6
- 2024Meta-Learning with Generalized Ridge Regression: High-dimensional Asymptotics, Optimality and Hyper-covariance Estimationcited by 5
- 2024Data-Driven Performance Guarantees for Classical and Learned Optimizerscited by 5
- 2025From Low Intrinsic Dimensionality to Non-Vacuous Generalization Bounds in Deep Multi-Task Learningcited by 5
- 2025Is Meta-Learning Out? Rethinking Unsupervised Few-Shot Classification with Limited Entropycited by 4
- 2023PAC-Bayesian Soft Actor-Critic Learningcited by 3
- 2023Wasserstein PAC-Bayes Learning: Exploiting Optimisation Guarantees to Explain Generalisationcited by 3
- 2022A Statistical Model for Predicting Generalization in Few-Shot Classificationcited by 2
- 2023Federated Learning with Nonvacuous Generalisation Boundscited by 2
- 2024Tighter Generalisation Bounds via Interpolationcited by 2
- 2024Meta-learning for Positive-unlabeled Classificationcited by 2
- 2025Fast Rate Bounds for Multi-Task and Meta-Learning with Different Sample Sizescited by 2
- 2022Hypernetwork approach to Bayesian MAMLcited by 0
- 2023Towards Understanding the Generalizability of Delayed Stochastic Gradient Descentcited by 0
- 2025Domain-Generalization to Improve Learning in Meta-Learning Algorithmscited by 0
- 2025An Information-Theoretic Analysis of OOD Generalization in Meta-Reinforcement Learningcited by 0
- 2026On the Stability and Generalization of First-order Bilevel Minimax Optimizationcited by 0
- 2026On the Stability and Generalization of First-order Bilevel Minimax Optimization22 Apr 2026
- 2025An Information-Theoretic Analysis of OOD Generalization in Meta-Reinforcement Learning27 Oct 2025
- 2025Is Meta-Learning Out? Rethinking Unsupervised Few-Shot Classification with Limited Entropy16 Sep 2025
- 2025Domain-Generalization to Improve Learning in Meta-Learning Algorithms13 Aug 2025
- 2025Fast Rate Bounds for Multi-Task and Meta-Learning with Different Sample Sizes21 May 2025
- 2025From Low Intrinsic Dimensionality to Non-Vacuous Generalization Bounds in Deep Multi-Task Learning31 Jan 2025
- 2024Tight PAC-Bayesian Risk Certificates for Contrastive Learning4 Dec 2024
- 2024Meta-learning for Positive-unlabeled Classification6 Jun 2024
- 2024Data-Driven Performance Guarantees for Classical and Learned Optimizers22 Apr 2024
- 2024Meta-Learning with Generalized Ridge Regression: High-dimensional Asymptotics, Optimality and Hyper-covariance Estimation27 Mar 2024
- 2024Tighter Generalisation Bounds via Interpolation7 Feb 2024
- 2024More Flexible PAC-Bayesian Meta-Learning by Learning Learning Algorithms6 Feb 2024
17 more
- 2023Federated Learning with Nonvacuous Generalisation Bounds17 Oct 2023
- 2023Towards Understanding the Generalizability of Delayed Stochastic Gradient Descent18 Aug 2023
- 2023Learning via Wasserstein-Based High Probability Generalisation Bounds7 Jun 2023
- 2023Exponential Smoothing for Off-Policy Learning25 May 2023
- 2023Wasserstein PAC-Bayes Learning: Exploiting Optimisation Guarantees to Explain Generalisation14 Apr 2023
- 2023Bayes meets Bernstein at the Meta Level: an Analysis of Fast Rates in Meta-Learning with PAC-Bayes23 Feb 2023
- 2023PAC-Bayesian Soft Actor-Critic Learning30 Jan 2023
- 2022A Statistical Model for Predicting Generalization in Few-Shot Classification13 Dec 2022
- 2022Scalable PAC-Bayesian Meta-Learning via the PAC-Optimal Hyper-Posterior: From Theory to Practice14 Nov 2022
- 2022Evaluated CMI Bounds for Meta Learning: Tightness and Expressiveness12 Oct 2022
- 2022On the Importance of Gradient Norm in PAC-Bayesian Bounds12 Oct 2022
- 2022Hypernetwork approach to Bayesian MAML6 Oct 2022
- 2022Meta-Learning Priors for Safe Bayesian Optimization3 Oct 2022
- 2022Understanding Benign Overfitting in Gradient-Based Meta Learning27 Jun 2022
- 2022Sharp-MAML: Sharpness-Aware Model-Agnostic Meta Learning8 Jun 2022
- 2022Is Bayesian Model-Agnostic Meta Learning Better than Model-Agnostic Meta Learning, Provably?6 Mar 2022
- 2021Learning an Explicit Hyperparameter Prediction Function Conditioned on Tasks6 Jul 2021
- 2022Sharp-MAML: Sharpness-Aware Model-Agnostic Meta Learningcited by 57
- 2023Exponential Smoothing for Off-Policy Learningcited by 15
- 2022Meta-Learning Priors for Safe Bayesian Optimizationcited by 14
- 2022Is Bayesian Model-Agnostic Meta Learning Better than Model-Agnostic Meta Learning, Provably?cited by 13
- 2022Understanding Benign Overfitting in Gradient-Based Meta Learningcited by 11
- 2022Evaluated CMI Bounds for Meta Learning: Tightness and Expressivenesscited by 11
- 2022Scalable PAC-Bayesian Meta-Learning via the PAC-Optimal Hyper-Posterior: From Theory to Practicecited by 11
- 2023Learning via Wasserstein-Based High Probability Generalisation Boundscited by 11
- 2024More Flexible PAC-Bayesian Meta-Learning by Learning Learning Algorithmscited by 10
- 2021Learning an Explicit Hyperparameter Prediction Function Conditioned on Taskscited by 9
- 2023Bayes meets Bernstein at the Meta Level: an Analysis of Fast Rates in Meta-Learning with PAC-Bayescited by 8
- 2022On the Importance of Gradient Norm in PAC-Bayesian Boundscited by 6
12 most cited of 29 All 29 (JSON)
- 2026On the Stability and Generalization of First-order Bilevel Minimax Optimization22 Apr 2026
- 2025An Information-Theoretic Analysis of OOD Generalization in Meta-Reinforcement Learning27 Oct 2025
- 2025Is Meta-Learning Out? Rethinking Unsupervised Few-Shot Classification with Limited Entropy16 Sep 2025
- 2025Domain-Generalization to Improve Learning in Meta-Learning Algorithms13 Aug 2025
- 2025Fast Rate Bounds for Multi-Task and Meta-Learning with Different Sample Sizes21 May 2025
- 2025From Low Intrinsic Dimensionality to Non-Vacuous Generalization Bounds in Deep Multi-Task Learning31 Jan 2025
- 2024Tight PAC-Bayesian Risk Certificates for Contrastive Learning4 Dec 2024
- 2024Meta-learning for Positive-unlabeled Classification6 Jun 2024
- 2024Data-Driven Performance Guarantees for Classical and Learned Optimizers22 Apr 2024
- 2024Meta-Learning with Generalized Ridge Regression: High-dimensional Asymptotics, Optimality and Hyper-covariance Estimation27 Mar 2024
- 2024Tighter Generalisation Bounds via Interpolation7 Feb 2024
- 2024More Flexible PAC-Bayesian Meta-Learning by Learning Learning Algorithms6 Feb 2024
12 newest of 29 All 29 (JSON)