Résumé
We develop a convergence-rate analysis of momentum with cyclical step-sizes. We show that under some assumption on the spectral gap of Hessians in machine learning, cyclical step-sizes are provably faster than constant step-sizes. More precisely, we develop a convergence rate analysis for quadratic objectives that provides optimal parameters and shows that cyclical learning rates can improve upon traditional lower complexity bounds. We further propose a systematic approach to design optimal first order methods for quadratic minimization with a given spectral structure. Finally, we provide a local convergence rate analysis beyond quadratic minimization for the proposed methods and illustrate our findings through benchmarks on least squares and logistic regression problems.
| langue originale | Anglais |
|---|---|
| Pages (de - à) | 3028-3065 |
| Nombre de pages | 38 |
| journal | Proceedings of Machine Learning Research |
| Volume | 151 |
| état | Publié - 1 janv. 2022 |
| Evénement | 25th International Conference on Artificial Intelligence and Statistics, AISTATS 2022 - Virtual, Online, Espagne Durée: 28 mars 2022 → 30 mars 2022 |
Empreinte digitale
Examiner les sujets de recherche de « Super-Acceleration with Cyclical Step-sizes ». Ensemble, ils forment une empreinte digitale unique.Contient cette citation
- APA
- Author
- BIBTEX
- Harvard
- Standard
- RIS
- Vancouver