First-order ANIL provably learns representations despite overparametrisation
Oguz Kaan Yüksel, Etienne Boursier, Nicolas Flammarion
Abstract
Meta-learning methods leverage data from previous tasks to learn a new task in a sample-efficient manner. In particular, model-agnostic methods look for initialisation points from which gradient descent quickly adapts to any new task. Although it has been empirically suggested that such methods learns shared representations during pretraining, there is limited theoretical evidence of such behavior. In this direction, this work shows, in the limit of infinite tasks, first-order ANIL with a linear two-layer network successfully learns linear shared representations. This result even holds under overparametrisation; having a width larger than the dimension of the shared representations results in an asymptotically low-rank solution.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 79f11d7f-a949-4b32-9e0d-d007ef61ac39Cited by top-tier papers2
- Provable unlearning in topic modeling and downstream tasksStanley Wei, Sadhika Malladi, Sanjeev Arora, Amartya SanyalICLR 2025
- Long-Context Linear System IdentificationOguz Kaan Yüksel, Mathieu Even, Nicolas FlammarionICLR 2025
Builds on7
- On the Theory of Transfer Learning: The Importance of Task DiversityNilesh Tripuraneni, Michael I. Jordan, Chi JinNeurIPS 2020 · 263 citations
- Provable Meta-Learning of Linear RepresentationsNilesh Tripuraneni, Chi Jin, Michael I. JordanICML 2021 · 218 citations
- Implicit Regularization in Deep Learning May Not Be Explainable by NormsNoam Razin, Nadav CohenNeurIPS 2020 · 178 citations
- Towards Resolving the Implicit Bias of Gradient Descent for Matrix Factorization: Greedy Low-Rank LearningZhiyuan Li, Yuping Luo, Kaifeng LyuICLR 2021 · 155 citations
- MAML and ANIL Provably Learn RepresentationsLiam Collins, Aryan Mokhtari, Sewoong Oh, Sanjay ShakkottaiICML 2022 · 38 citations
Related papers
- Statistically and Computationally Efficient Linear Meta-representation LearningKiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli, Sewoong OhNeurIPS 2021 · 28 citations
- A Representation Learning Perspective on the Importance of Train-Validation Splitting in Meta-LearningNikunj Saunshi, Arushi Gupta, Wei HuICML 2021 · 19 citations
- Provable Generalization of Overparameterized Meta-learning Trained with SGDYu Huang, Yingbin Liang, Longbo HuangNeurIPS 2022 · 14 citations
- Theoretical Characterization of the Generalization Performance of Overfitted Meta-LearningPeizhong Ju, Yingbin Liang, Ness B. ShroffICLR 2023 · 3 citations
- Towards Sample-efficient Overparameterized Meta-learningYue Sun, Adhyyan Narang, Halil Ibrahim Gulluk, Samet Oymak et al.NeurIPS 2021 · 26 citations
