kth.sePublications KTH
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
Stability and Convergence of Stochastic Gradient Clipping: Beyond Lipschitz Continuity and Smoothness
KTH, School of Electrical Engineering and Computer Science (EECS), Intelligent systems, Decision and Control Systems (Automatic Control).
KTH, School of Electrical Engineering and Computer Science (EECS), Intelligent systems, Decision and Control Systems (Automatic Control).ORCID iD: 0000-0002-2237-2580
2021 (English)In: Proceedings of the 38th International Conference on Machine Learning, ICML 2021, ML Research Press , 2021, p. 7325-7335Conference paper, Published paper (Refereed)
Abstract [en]

Stochastic gradient algorithms are often unstable when applied to functions that do not have Lipschitz-continuous and/or bounded gradients. Gradient clipping is a simple and effective technique to stabilize the training process for problems that are prone to the exploding gradient problem. Despite its widespread popularity, the convergence properties of the gradient clipping heuristic are poorly understood, especially for stochastic problems. This paper establishes both qualitative and quantitative convergence results of the clipped stochastic (sub)gradient method (SGD) for non-smooth convex functions with rapidly growing subgradients. Our analyses show that clipping enhances the stability of SGD and that the clipped SGD algorithm enjoys finite convergence rates in many cases. We also study the convergence of a clipped method with momentum, which includes clipped SGD as a special case, for weakly convex problems under standard assumptions. With a novel Lyapunov analysis, we show that the proposed method achieves the best-known rate for the considered class of problems, demonstrating the effectiveness of clipped methods also in this regime. Numerical results confirm our theoretical developments.

Place, publisher, year, edition, pages
ML Research Press , 2021. p. 7325-7335
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:kth:diva-333347ISI: 000768182703043Scopus ID: 2-s2.0-85156201826OAI: oai:DiVA.org:kth-333347DiVA, id: diva2:1784978
Conference
38th International Conference on Machine Learning, ICML 2021, Virtual, Online, Jul 18 2021 - Jul 24 2021
Note

Part of ISBN 9781713845065

QC 20250225

Available from: 2023-08-01 Created: 2023-08-01 Last updated: 2026-06-30Bibliographically approved

Open Access in DiVA

No full text in DiVA

Scopus

Authority records

Mai, Vien V.Johansson, Mikael

Search in DiVA

By author/editor
Mai, Vien V.Johansson, Mikael
By organisation
Decision and Control Systems (Automatic Control)
Computer Sciences

Search outside of DiVA

GoogleGoogle Scholar

urn-nbn

Altmetric score

urn-nbn
Total: 87 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf