PreprintCited by 303Evolutionary Algorithms and ApplicationsPolynomial and algebraic computationMetaheuristic Optimization Algorithms Research

The DOI matches its DataCite record; 100 of the 111 references it cites were found in a registry or an index; 6 of the 27 sources its claims cite were checked for support.

Symbolic Discovery of Optimization Algorithms

Open access

Abstract

We present a method to formulate algorithm discovery as program search, and apply it to discover optimization algorithms for deep neural network training. We leverage efficient search techniques to explore an infinite and sparse program space. To bridge the large generalization gap between proxy and target tasks, we also introduce program selection and simplification strategies. Our method discovers a simple and effective optimization algorithm, $\textbf{Lion}$ ($\textit{Evo$\textbf{L}$ved S$\textbf{i}$gn M$\textbf{o}$me$\textbf{n}$tum}$). It is more memory-efficient than Adam as it only keeps track of the momentum. Different from adaptive optimizers, its update has the same magnitude for each parameter calculated through the sign operation. We compare Lion with widely used optimizers, such as Adam and Adafactor, for training a variety of models on different tasks. On image classification, Lion boosts the accuracy of ViT by up to 2% on ImageNet and saves up to 5x the pre-training compute on JFT. On vision-language contrastive learning, we achieve 88.3% $\textit{zero-shot}$ and 91.1% $\textit{fine-tuning}$ accuracy on ImageNet, surpassing the previous best results by 2% and 0.1%, respectively. On diffusion models, Lion outperforms Adam by achieving a better FID score and reducing the training compute by up to 2.3x. For autoregressive, masked language modeling, and fine-tuning, Lion exhibits a similar or better performance compared to Adam. Our analysis of Lion reveals that its performance gain grows with the training batch size. It also requires a smaller learning rate than Adam due to the larger norm of the update produced by the sign function. Additionally, we examine the limitations of Lion and identify scenarios where its improvements are small or not statistically significant. Lion is also successfully deployed in production systems such as Google search ads CTR model.

Overview

AI snapshot

Written from the paper’s full text.

TL;DR

The paper formulates optimizer discovery as program search and discovers Lion, a momentum-only optimizer that uses the sign operation for updates. Lion is evaluated across vision, vision-language, diffusion, and language tasks, often matching or outperforming AdamW/Adafactor while sometimes saving compute.

Key findings

  • Lion is more memory-efficient than Adam because it only tracks momentum, and its update has uniform magnitude per parameter via the sign operation.
  • On image classification, Lion boosts ViT accuracy by up to 2% on ImageNet and saves up to 5x pre-training compute on JFT; on vision-language contrastive learning, BASIC-L with Lion reaches 88.3% zero-shot and 91.1% fine-tuning ImageNet accuracy, surpassing previous best results by 2% and 0.1%.
  • On diffusion models, Lion achieves a better FID score and reduces training compute by up to 2.3x; for 256×256 generation, final FID is 4.1 (Lion) vs. 4.7 (AdamW).
  • On language modeling, Lion shows similar or better performance than Adam, with up to 2x compute savings; on GLUE fine-tuning, Lion beats AdamW on average across T5 Base, Large, and 11B, with 10, 12, and 10 wins out of 12 scores, respectively.

Methods

  • Algorithm discovery is cast as program search over imperative programs operating on n-dimensional arrays, using regularized evolution with warm-start and restart, abstract execution, funnel selection, and program simplification.
  • Search uses low-cost proxy tasks: for vision, a 3-layer ViT on 10% ImageNet for 30k steps; for language, a 2-layer Transformer on LM1B for 20K steps; evaluation can complete on one TPU V2 chip within 20 min.
  • Each search experiment uses 100 TPU V2 chips for ~72h, generating 200–300K programs with ~20–30K actually evaluated; total cost is ~3K TPU V2 days. Evaluation compares Lion with AdamW/Adafactor across image classification, vision-language contrastive learning, diffusion, language modeling, and fine-tuning, with tuned peak learning rate and decoupled weight decay.

Limitations

  • The paper states search limitations: the search space is inspired by first-order optimizers, lacks functions for advanced second-order algorithms, has large search cost, requires manual simplification, and uses a simplistic program structure without effective conditional/loop/function constructs.
  • The paper states Lion limitations: gains are small on ResNets, decrease with strong augmentations, and are similar to AdamW on the Imagen text-to-image base model, large-scale internal-dataset autoregressive perplexity, and C4 masked language modeling; Lion likely performs no better than AdamW with batch size <64, and bfloat16 momentum tracking can still be expensive for giant models.

Model-written summary. Check it against the paper before relying on it.

Integrity signals

Registry

DataCite · 10.48550/arxiv.2302.06675

Cites

100 of 100 references verified0 citation errors · 0 not checkable

Statistics

Nothing to recompute.

Verification

Claims the paper makes

71 Own result · 31 Own interpretation · 17 No basis given · 7 Not checked yet · 4 Sources not found · 3 Supported · 3 Partly supported · accuracy not yet measured

Verify all 6 references — free account

Show all 69 claims

  • Probably not a substantial claim. 25
  • It may state two separate claims as one. 50
  • The quote only partly states this claim. 7
67 of 67
  • The authors state this as an assumption.

    Simpler programs are more likely to generalize.

    “Simpler programs are easier to understand and our intuition is that they are more likely to generalize, so we simplify the programs with the following steps.”
  • The paper relies on this without support.

    The meta-validation tasks used for selection are representative enough of the target tasks to predict generalization.

    “We use performance on metavalidation tasks that are larger than the proxy tasks by increasing the model size and training steps, to select the programs that generalize beyond proxy tasks then further simplify them.”
    “The discovered algorithms face a significant challenge due to the substantial gap between the proxy tasks during search and the target tasks.”
  • The paper relies on this without support.

    The program search space contains high-quality optimization algorithms that can be discovered.

    “We present a method to formulate algorithm discovery as program search, and apply it to discover optimization algorithms for deep neural network training.”
    “Our method discovers a simple and effective optimization algorithm, Lion ( Evo L ved S i gn M o me n tum ).”
  • The paper relies on this without support.

    AdamW is the de facto standard optimizer on a majority of learning tasks.

    “as it is exceedingly popular and the de facto standard optimizer on a majority of learning tasks”
4 of 4

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

Each contradiction between two passages listed here was confirmed by an independent check.

2 of 2

Citations

Cites

  • Deep Residual Learning for Image Recognition

    He, K., Zhang, X., Ren, S., & Sun, J. (2016) · 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr.2016.90 · Cited by 235,474 · cited 1×

    “Train from scratch on ImageNet Following previous works (Dosovitskiy et al., 2021; He et al., 2016), we train ResNet-50 for 90 epochs with a batch size of 1,024, and other models for 300 epochs with a batch size of 4,096.”
    Method
  • Attention Is All You Need

    Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A. N., Kaiser, L., & Polosukhin, I. (2017) · arXiv · doi:10.48550/arxiv.1706.03762 · Cited by 120,404 · cited 1×

    “…weight decay [60], also referred to as AdamW, and Adafactor with factorized second moments [86], are still the de facto standard optimizers for training most deep neural networks, especially the recent state-of-the-art language [12, 23, 90], vision [21, 26, 102] and multimodal [75, 84, 100] models.”
    Background
  • Adam: A Method for Stochastic Optimization

    Kingma, D. P., & Ba, J. (2014) · arXiv · doi:10.48550/arxiv.1412.6980 · Cited by 98,946 · cited 2×

    “However, Adam (Kingma and Ba, 2014) with decoupled weight decay (Loshchilov and Hutter, 2019), also referred to as AdamW, and Adafactor with factorized second moments (Shazeer and Stern, 2018), are still the de facto standard optimizers for training most deep neural networks, especially the recent state-of-the-art lan…”
    Background
  • BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding

    Devlin, J., Chang, M.-W., Lee, K., & Toutanova, K. (2019) · Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers) · doi:10.18653/v1/n19-1423 · Cited by 54,325

  • An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale

    Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., Gelly, S., et al. (2020) · arXiv · doi:10.48550/arxiv.2010.11929 · Cited by 47,472 · cited 3×

    “Train from scratch on ImageNet Following previous works [26, 37], we train ResNet-50 for 90 epochs with a batch size of 1,024, and other models for 300 epochs with a batch size of 4,096.”
    BackgroundMethod
  • Microsoft COCO: Common Objects in Context

    Lin, T.-Y., Maire, M., Belongie, S., Hays, J., Perona, P., Ramanan, D., Dollár, P., & Zitnick, C. L. (2014) · Lecture Notes in Computer Science · doi:10.1007/978-3-319-10602-1_48 · Cited by 45,460 · cited 3×

    “The zero-shot image-text retrieval results on MSCOCO (Lin et al., 2014) and Flickr30K (Plummer et al.”
    Method
  • Array programming with NumPy

    Harris, C. R., Millman, K. J., van der Walt, S. J., Gommers, R., Virtanen, P., Cournapeau, D., Wieser, E., Taylor, J., Berg, S., Smith, N. J., et al. (2020) · Nature · doi:10.1038/s41586-020-2649-2 · Cited by 23,826 · cited 1×

    “They are similar to Python code using NumPy / JAX [10, 36] as well as pseudo code of optimization algorithms.”
    Method
  • Language Models are Few-Shot Learners

    Brown, T. B., Mann, B., Ryder, N., Subbiah, M., Kaplan, J., Dhariwal, P., Neelakantan, A., Shyam, P., Sastry, G., Askell, A., et al. (2020) · arXiv · doi:10.48550/arxiv.2005.14165 · Cited by 23,372 · cited 2×

    “Following GPT-3 [12], we train three models, from 1.1B to 7.5B parameters, on 300B tokens with a batch size of 3M tokens and a context length of 1K.”
    BackgroundMethod
  • RoBERTa: A Robustly Optimized BERT Pretraining Approach

    Liu, Y., Ott, M., Goyal, N., Du, J., Joshi, M., Chen, D., Levy, O., Lewis, M., Zettlemoyer, L., & Stoyanov, V. (2019) · arXiv · doi:10.48550/arxiv.1907.11692 · Cited by 16,515

    Self-citation
  • Learning Transferable Visual Models From Natural Language Supervision

    Radford, A., Kim, J. W., Hallacy, C., Ramesh, A., Goh, G., Agarwal, S., Sastry, G., Askell, A., Mishkin, P., Clark, J., et al. (2021) · arXiv · doi:10.48550/arxiv.2103.00020 · Cited by 15,293 · cited 2×

    “This section focuses on the vision-language contrastive training [75].”
    Background

Show all 111 references

Research works printed but not linked: 10

  1. Andrei Barbu, David Mayo, Julian Alverio, William Luo, Christopher Wang, Dan Gutfreund, Josh Tenenbaum, and Boris Katz. Objectnet: A large-scale bias-controlled dataset for pushing the limits of object recognition models. In H. Wallach, H. Larochelle, A. Beygelzimer, F. d'Alché-Buc, E. Fox, and R. Garnett, editors, Advances in Neural Information Processing Systems , volume 32. Curran Associates, Inc., 2019.
  2. Jonathan Berant, Andrew Chou, Roy Frostig, and Percy Liang. Semantic parsing on Freebase from questionanswer pairs. In Proceedings of the 2013 Conference on Empirical Methods in Natural Language Processing , pages 1533-1544, Seattle, Washington, USA, October 2013. Association for Computational Linguistics.
  3. Timothy Dozat. Incorporating Nesterov Momentum into Adam. In Proceedings of the 4th International Conference on Learning Representations , pages 1-4, 2016.
  4. Andrew Gordon, Zornitsa Kozareva, and Melissa Roemmele. SemEval-2012 task 7: Choice of plausible alternatives: An evaluation of commonsense causal reasoning. In *SEM 2012: The First Joint Conference on Lexical and Computational Semantics - Volume 1: Proceedings of the main conference and the shared task, and Volume 2: Proceedings of the Sixth International Workshop on Semantic Evaluation (SemEval 2012) , pages 394-398, Montréal, Canada, 7-8 June 2012. Association for Computational Linguistics.
  5. Mandy Guo, Zihang Dai, Denny Vrandečić, and Rami Al-Rfou. Wiki-40B: Multilingual language model dataset. In Proceedings of the Twelfth Language Resources and Evaluation Conference , pages 2440-2452, Marseille, France, May 2020. European Language Resources Association. ISBN 979-10-95546-34-4.
  6. John H Holland. Adaptation in natural and artificial systems: an introductory analysis with applications to biology, control, and artificial intelligence . MIT press, 1992.
  7. Hector J. Levesque, Ernest Davis, and Leora Morgenstern. The winograd schema challenge. In Proceedings of the Thirteenth International Conference on Principles of Knowledge Representation and Reasoning , KR'12, page 552-561. AAAI Press, 2012. ISBN 9781577355601.
  8. Ilya Loshchilov and Frank Hutter. Decoupled weight decay regularization. In International Conference on Learning Representations , 2019.
  9. Sachin Ravi and Hugo Larochelle. Optimization as a model for few-shot learning. In International Conference on Learning Representations , 2017.
  10. Yuanhao Xiong, Li-Cheng Lan, Xiangning Chen, Ruochen Wang, and Cho-Jui Hsieh. Learning to schedule learning rate with graph neural networks. In International Conference on Learning Representations , 2022.

Cited by

Sorted by citations

Metadata

Authors
Chen, X., Liang, C., Huang, D., Real, E., Wang, K., Liu, Y., Pham, H., Dong, X., Luong, T., Hsieh, C.-J., Lu, Y., & Le, Q. V.
Venue
arXiv
Publisher
arXiv
Published
2023
Pages
49205-49233
ISSN
1049-5258 (ISSN-L)
OpenAlex
W7133224204
Cited by
303
References
111
Citations by year
2026: 55 · 2025: 81 · 2024: 116 · 2023: 51
FWCI
53.56
Type
conference paper
Language
en
Open access
Locations
  • neural information processing systems (publishedVersion)
  • arXiv (Cornell University) (submittedVersion)
  • arXiv (Cornell University)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Evolutionary Algorithms and Applications
  • Physical Sciences › Computer Science › Computational Theory and Mathematics › Polynomial and algebraic computation
  • Physical Sciences › Computer Science › Artificial Intelligence › Metaheuristic Optimization Algorithms Research
Keywords
symbolic computation, optimization algorithms, algorithm discovery

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

A language model drew the claims from the paper on 2026-10-10, under version 3 of the claim definition (model deepseek-v4.1-flash, prompt extract-v3).

A second model labelled each claim on 2026-10-10 (model jev-1.13-20260917, definition revision gold-guideline-v0.2).

Cited sources are checked with a language model’s help (gpt-6-sol).

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review