Learning-to-learn non-convex piecewise-Lipschitz functions
Maria-Florina Balcan, Mikhail Khodak, Dravyansh Sharma, Ameet Talwalkar
Abstract
We analyze the meta-learning of the initialization and step-size of learning algorithms for piecewise-Lipschitz functions, a non-convex setting with applications to both machine learning and algorithms. Starting from recent regret bounds for the exponential forecaster on losses with dispersed discontinuities, we generalize them to be initialization-dependent and then use this result to propose a practical meta-learning procedure that learns both the initialization and the step-size of the algorithm from multiple online learning tasks. Asymptotically, we guarantee that the average regret across tasks scales with a natural notion of task-similarity that measures the amount of overlap between near-optimal regions of different tasks. Finally, we instantiate the method and its guarantee in two important settings: robust meta-learning and multi-task data-driven algorithm design.
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 180abd5f-153a-4e54-8e0a-68b0d9d39738Cited by top-tier papers8
- Learning Predictions for Algorithms with PredictionsMisha Khodak, Maria-Florina Balcan, Ameet Talwalkar, Sergei VassilvitskiiNeurIPS 2022 · 40 citations
- Meta-Learning Adversarial Bandit AlgorithmsMisha Khodak, Ilya Osadchiy, Keegan Harris, Maria-Florina Balcan et al.NeurIPS 2023 · 13 citations
- Learning-augmented private algorithms for multiple quantile releaseMikhail Khodak, Kareem Amin, Travis Dick, Sergei VassilvitskiiICML 2023 · 7 citations
- Learning in Structured Stackelberg GamesNina Balcan, Kiriaki Fragkia, Keegan HarrisICML 2026 · 4 citations
- Conservative classifiers do consistently well with improving agents: characterizing statistical and online learningDravyansh Sharma, Alec SunNeurIPS 2025 · 3 citations
Builds on8
- Rapid Learning or Feature Reuse? Towards Understanding the Effectiveness of MAMLAniruddh Raghu, Maithra Raghu, Samy Bengio, Oriol VinyalsICLR 2020 · 736 citations
- Provable Meta-Learning of Linear RepresentationsNilesh Tripuraneni, Chi Jin, Michael I. JordanICML 2021 · 218 citations
- Differentially Private Meta-LearningJeffrey Li, Mikhail Khodak, Sebastian Caldas, Ameet TalwalkarICLR 2020 · 125 citations
- Few-Shot Learning via Learning the Representation, ProvablySimon Shaolei Du, Wei Hu, Sham M. Kakade, Jason D. Lee et al.ICLR 2021 · 56 citations
- Robust Meta-learning for Mixed Linear Regression with Small BatchesWeihao Kong, Raghav Somani, Sham M. Kakade, Sewoong OhNeurIPS 2020 · 38 citations
Related papers
- Improved Regret Bounds for Non-Convex Online-Within-Online Meta LearningJiechao Guan, Hui XiongICLR 2024
- On the Stability and Generalization of Meta-LearningYunjuan Wang, Raman AroraNeurIPS 2024 · 12 citations
- Gradient-Variation Online Learning under Generalized SmoothnessYan-Feng Xie, Peng Zhao, Zhi-Hua ZhouNeurIPS 2024 · 14 citations
- Online Learning with Unknown ConstraintsKarthik Sridharan, Seung Won Wilson YooICML 2025
- Structured Prediction for Conditional Meta-LearningRuohan Wang, Yiannis Demiris, Carlo CilibertoNeurIPS 2020 · 19 citations
