Generalization Bounds For Meta-Learning: An Information-Theoretic Analysis
Qi Chen, Changjian Shui, Mario Marchand
Abstract
We derive a novel information-theoretic analysis of the generalization property of meta-learning algorithms. Concretely, our analysis proposes a generic understanding of both the conventional learning-to-learn framework [1] and the modern model-agnostic meta learning (MAML) algorithms [2] . Moreover, we provide a data-dependent generalization bound for a stochastic variant of MAML, which is non-vacuous for deep few-shot learning. As compared to previous bounds that depend on the square norm of gradients, empirical validations on both simulated data and a well-known few-shot benchmark show that the proposed bound is orders of magnitude tighter in most situations.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers21
- On Learning Fairness and Accuracy on Multiple SubgroupsChangjian Shui, Gezheng Xu, Qi Chen, Jiaqi Li et al.NeurIPS 2022 · 58 citations
- Out-of-Domain Robustness via Targeted AugmentationsIrena Gao, Shiori Sagawa, Pang Wei Koh, Tatsunori Hashimoto et al.ICML 2023 · 33 citations
- On the Stability-Plasticity Dilemma in Continual Meta-Learning: Theory and AlgorithmQi Chen, Changjian Shui, Ligong Han, Mario MarchandNeurIPS 2023 · 32 citations
- Understanding Benign Overfitting in Gradient-Based Meta LearningLisha Chen, Songtao Lu, Tianyi ChenNeurIPS 2022 · 20 citations
- A Simple Yet Effective Strategy to Robustify the Meta Learning ParadigmQi Wang, Yiqin Lv, Yang-He Feng, Zheng Xie et al.NeurIPS 2023 · 17 citations
Builds on4
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- On Generalization Error Bounds of Noisy Gradient Methods for Non-Convex LearningJian Li, Xuanyuan Luo, Mingda QiaoICLR 2020 · 95 citations
- How Important is the Train-Validation Split in Meta-Learning?Yu Bai, Minshuo Chen, Pan Zhou, Tuo Zhao et al.ICML 2021 · 60 citations
- A Representation Learning Perspective on the Importance of Train-Validation Splitting in Meta-LearningNikunj Saunshi, Arushi Gupta, Wei HuICML 2021 · 19 citations
Related papers
- Bridging the Gap Between Practice and PAC-Bayes Theory in Few-Shot Meta-LearningNan Ding, Xi Chen, Tomer Levinboim, Sebastian Goodman et al.NeurIPS 2021 · 34 citations
- Theoretical bounds on estimation error for meta-learningJames Lucas, Mengye Ren, Irene Raissa Kameni, Toniann Pitassi et al.ICLR 2021 · 12 citations
- A Closer Look at the Training Strategy for Modern Meta-LearningJiaxin Chen, Xiao-Ming Wu, Yanke Li, Qimai Li et al.NeurIPS 2020 · 48 citations
- Provable Generalization of Overparameterized Meta-learning Trained with SGDYu Huang, Yingbin Liang, Longbo HuangNeurIPS 2022 · 14 citations
- Sharp-MAML: Sharpness-Aware Model-Agnostic Meta LearningMomin Abbas, Quan Xiao, Lisha Chen, Pin-Yu Chen et al.ICML 2022 · 105 citations
