kth.sePublikationer KTH
Ändra sökning
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf
Convergence of a stochastic gradient method with momentum for non-smooth non-convex optimization
KTH, Skolan för elektroteknik och datavetenskap (EECS), Intelligenta system, Reglerteknik.
KTH, Skolan för elektroteknik och datavetenskap (EECS), Intelligenta system, Reglerteknik.ORCID-id: 0000-0002-2237-2580
2020 (Engelska)Ingår i: 37th International Conference on Machine Learning, ICML 2020, International Machine Learning Society (IMLS) , 2020, s. 6576-6585Konferensbidrag, Publicerat paper (Refereegranskat)
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 unconstrained case can be analyzed under weaker assumptions than the state-of-The-Art. Numerical results confirm our theoretical developments.

Ort, förlag, år, upplaga, sidor
International Machine Learning Society (IMLS) , 2020. s. 6576-6585
Nyckelord [en]
Constrained optimization, Convex optimization, Gradient methods, Lyapunov functions, Machine learning, Momentum, Complexity bounds, Constrained optimi-zation problems, Convergence rates, Nonconvex optimization, Numerical results, Stochastic gradient methods, Sub-gradient methods, Theoretical development, Stochastic systems
Nationell ämneskategori
Reglerteknik
Identifikatorer
URN: urn:nbn:se:kth:diva-302903Scopus ID: 2-s2.0-85099888979OAI: oai:DiVA.org:kth-302903DiVA, id: diva2:1599870
Konferens
37th International Conference on Machine Learning, ICML 2020, July 13-18, 2020
Anmärkning

Part of ISBN 9781713821120

QC 20260715

Tillgänglig från: 2021-10-02 Skapad: 2021-10-02 Senast uppdaterad: 2026-07-15Bibliografiskt granskad

Open Access i DiVA

Fulltext saknas i DiVA

Scopus

Person

Mai, Vien V.Johansson, Mikael

Sök vidare i DiVA

Av författaren/redaktören
Mai, Vien V.Johansson, Mikael
Av organisationen
Reglerteknik
Reglerteknik

Sök vidare utanför DiVA

GoogleGoogle Scholar

urn-nbn

Altmetricpoäng

urn-nbn
Totalt: 32 träffar
RefereraExporteraLänk till posten
Permanent länk

Direktlänk
Referera
Referensformat
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Annat format
Fler format
Språk
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Annat språk
Fler språk
Utmatningsformat
  • html
  • text
  • asciidoc
  • rtf