Skip to main navigation Skip to search Skip to main content

Super-Acceleration with Cyclical Step-sizes

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

Research output: Contribution to journalConference articlepeer-review

18 Citations (Scopus)

Abstract

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.

Original languageEnglish
Pages (from-to)3028-3065
Number of pages38
JournalProceedings of Machine Learning Research
Volume151
Publication statusPublished - 1 Jan 2022
Event25th International Conference on Artificial Intelligence and Statistics, AISTATS 2022 - Virtual, Online, Spain
Duration: 28 Mar 202230 Mar 2022

Fingerprint

Dive into the research topics of 'Super-Acceleration with Cyclical Step-sizes'. Together they form a unique fingerprint.

Cite this