OLion: Approaching the Hadamard Ideal by Intersecting Spectral and L inf Implicit Biases
Abstract
Lay Summary
Large AI models are expensive to train, and the choice of training method can strongly affect both their speed and the quality of the final model. Existing methods often make a tradeoff: some control the overall shape of model updates well, while others control individual numbers in the model well, but they do not naturally do both. We propose OLion, a new training method that combines these two kinds of control in one simple update. It first reshapes each update so that it is well balanced overall, then keeps only the direction of each individual value, which helps prevent a few unusually large values from dominating training. This design uses less extra memory than widely used adaptive methods such as AdamW, which is important when training very large models. We tested OLion on language model pretraining, image model pretraining, and fine-tuning large language models. Across these settings, OLion matches or improves over strong existing methods while keeping memory costs low. This matters because better training methods can reduce the cost of building large AI systems and make large-model training and fine-tuning more practical.