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
Loading...
Thumbnail Image
Name

12949_Training_Deep_Learning_M.pdf

Type

Main Document

Version

Accepted version

Access type

openaccess

License Condition

N/A

Size

906.02 KB

Format

Adobe PDF

Checksum (MD5)

21a588d6321cc4898d7777d8c718f9fe

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