Digitala Vetenskapliga Arkivet

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
Convergence of a Stochastic Gradient Method with Momentum for Non-Smooth Non-Convex Optimization
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). Division of Decision and Control Systems, EECS, KTH Royal Institute of Technology, Stockholm, Sweden.ORCID iD: 0000-0002-2237-2580
2020 (English)In: Proceedings of Machine Learning Research - International Conference on Machine Learning, ICML 2020, ML Research Press , 2020, Vol. 119Conference paper, Published paper (Refereed)
Abstract [en]

Stochastic gradient methods with momentum are widely used in applications and at the core of optimization subroutines in many popular machine learning libraries. However, their sample complexities have not been obtained for problems beyond those that are convex or smooth. This paper establishes the convergence rate of a stochastic subgradient method with a momentum term of Polyak type for a broad class of non-smooth, non-convex, and constrained optimization problems. Our key innovation is the construction of a special Lyapunov function for which the proven complexity can be achieved without any tuning of the momentum parameter. For smooth problems, we extend the known complexity bound to the constrained case and demonstrate how the un- constrained case can be analyzed under weaker assumptions than the state-of-the-art. Numerical results confirm our theoretical developments.

Place, publisher, year, edition, pages
ML Research Press , 2020. Vol. 119
National Category
Control Engineering Computational Mathematics
Identifiers
URN: urn:nbn:se:kth:diva-385508Scopus ID: 2-s2.0-105022302041OAI: oai:DiVA.org:kth-385508DiVA, id: diva2:2086577
Conference
37th International Conference on Machine Learning, ICML 2020, Virtual, Online, July 13-18, 2020
Note

Not duplicate with diva 1599870

QC 20260715

Available from: 2026-07-15 Created: 2026-07-15 Last updated: 2026-07-15Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Scopusfulltext

Search in DiVA

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

Search outside of DiVA

GoogleGoogle Scholar

urn-nbn

Altmetric score

urn-nbn
Total: 4 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