Passer à la navigation principale Passer à la recherche Passer au contenu principal

Super-Acceleration with Cyclical Step-sizes

  • Samsung SAIL Montreal
  • PSL research University & IPSL
  • Google Inc.

Résultats de recherche: Contribution à un journalArticle de conférenceRevue par des pairs

18 Citations (Scopus)

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 originaleAnglais
Pages (de - à)3028-3065
Nombre de pages38
journalProceedings of Machine Learning Research
Volume151
étatPublié - 1 janv. 2022
Evénement25th International Conference on Artificial Intelligence and Statistics, AISTATS 2022 - Virtual, Online, Espagne
Durée: 28 mars 202230 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