Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Training Deep Learning Models with Norm-Constrained LMOs
 
conference paper

Training Deep Learning Models with Norm-Constrained LMOs

Pethick, Thomas  
•
Xie, Wanyun  
•
Antonakopoulos, Kimon  
Show more
July 2025
Proceedings of the 42 nd International Conference on Machine Learning
Forty-Second International Conference on Machine Learning

In this work, we study optimization methods that leverage the linear minimization oracle (lmo) over a norm-ball. We propose a new stochastic family of algorithms that uses the lmo to adapt to the geometry of the problem and, perhaps surprisingly, show that they can be applied to unconstrained problems. The resulting update rule unifies several existing optimization methods under a single framework. Furthermore, we propose an explicit choice of norm for deep architectures, which, as a side benefit, leads to the transferability of hyperparameters across model sizes. Experimentally, we demonstrate significant speedups on nanoGPT training using our algorithm, Scion, without any reliance on Adam. The proposed method is memory-efficient, requiring only one set of model weights and one set of gradients, which can be stored in halfprecision. The code is available at https: //github.com/LIONS-EPFL/scion.

  • Files
  • Details
  • Metrics
Type
conference paper
Author(s)
Pethick, Thomas  

EPFL

Xie, Wanyun  

EPFL

Antonakopoulos, Kimon  

EPFL

Zhu, Zhenyu  

EPFL

Silveti-Falls, Antonio
Cevher, Volkan  orcid-logo

EPFL

Date Issued

2025-07

Published in
Proceedings of the 42 nd International Conference on Machine Learning
Series title/Series vol.

Proceedings of Machine Learning Research; 267

ISSN (of the series)

2640-3498

Subjects

ML-AI

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
LIONS  
Event nameEvent acronymEvent placeEvent date
Forty-Second International Conference on Machine Learning

ICML 2025

Vancouver, Canada

2025-07-13 - 2025-07-19

Available on Infoscience
August 8, 2025
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/252844
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés