Neural Optimizer Search with Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
We present an approach to automate the process of discovering optimization methods, with a focus on deep learning architectures. We train a Recurrent Neural Network controller to generate a string in a domain specific language that describes a mathematical update equation based on a list of primitive functions, such as the gradient, running average of the gradient, etc. The controller is trained with Reinforcement Learning to maximize the performance of a model after a few epochs. On CIFAR-10, our method discovers several update rules that are better than many commonly used optimizers, such as Adam, RMSProp, or SGD with and without Momentum on a ConvNet model. We introduce two new optimizers, named PowerSign and AddSign, which we show transfer well and improve training on a variety of different tasks and architectures, including ImageNet classification and Google's neural machine translation system.
Key facts. Cited by 82 papers in the map (data through 7 Oct 2026). First posted on arXiv 21 Sep 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 22; 2022: 23; 2023: 10; 2024: 11; 2025: 9; 2026: 7.
82 papers.
Sort:
By area:
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2022Tutorial on amortized optimizationcited by 83
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2021GLiT: Neural Architecture Search for Global and Local Image Transformercited by 53
- 2023Lion Secretly Solves Constrained Optimization: As Lyapunov Predictscited by 50
- 2021Evolving Reinforcement Learning Algorithmscited by 48
- 2022Automated Dynamic Algorithm Configurationcited by 37
- 2021AutoDropout: Learning Dropout Patterns to Regularize Deep Networkscited by 35
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizerscited by 32
- 2021How to Train Your MAML to Excel in Few-Shot Classificationcited by 22
- 2021BN-NAS: Neural Architecture Search with Batch Normalizationcited by 21
38 more
- 2022Finite Expression Method for Solving High-Dimensional Partial Differential Equationscited by 21
- 2022Open Source Vizier: Distributed Infrastructure and API for Reliable and Flexible Blackbox Optimizationcited by 17
- 2022Symbolic Learning to Optimize: Towards Interpretability and Scalabilitycited by 16
- 2023MSINet: Twins Contrastive Search of Multi-Scale Interaction for Object ReIDcited by 16
- 2021G-CoS: GNN-Accelerator Co-Search Towards Both Better Accuracy and Efficiencycited by 15
- 2023B2Opt: Learning to Optimize Black-box Optimization with Little Budgetcited by 15
- 2021Improving exploration in policy gradient search: Application to symbolic optimizationcited by 14
- 2021Training Learned Optimizers with Randomly Initialized Learned Optimizerscited by 13
- 2021Acceleration via Fractal Learning Rate Schedulescited by 13
- 2021T-AutoML: Automated Machine Learning for Lesion Segmentation using Transformers in 3D Medical Imagingcited by 11
- 2022Selecting and Composing Learning Rate Policies for Deep Neural Networkscited by 11
- 2022AutoLossGen: Automatic Loss Function Generation for Recommender Systemscited by 10
- 2022Learning to Optimize in Model Predictive Controlcited by 10
- 2024Efficient Pretraining Data Selection for Language Models via Multi-Actor Collaborationcited by 10
- 2026AI4AI-Bench: Benchmarking LLM Agents in Algorithmic Design for Recursive Self-Improvementcited by 10
- 2021MetricOpt: Learning to Optimize Black-Box Evaluation Metricscited by 9
- 2022Graph-based Neural Acceleration for Nonnegative Matrix Factorizationcited by 9
- 2022Construction of Hierarchical Neural Architecture Search Spaces based on Context-free Grammarscited by 9
- 2024Adaptive Non-uniform Timestep Sampling for Accelerating Diffusion Model Trainingcited by 8
- 2021Learning specialized activation functions with the Piecewise Linear Unitcited by 7
- 2022Meta Mirror Descent: Optimiser Learning for Fast Convergencecited by 7
- 2022Reinforcement Learning Based Query Vertex Ordering Model for Subgraph Matchingcited by 6
- 2022Efficient Non-Parametric Optimizer Search for Diverse Taskscited by 6
- 2024tnGPS: Discovering Unknown Tensor Network Structure Search Algorithms via Large Language Models (LLMs)cited by 5
- 2022Entangled Residual Mappingscited by 4
- 2023Learning To Optimize Quantum Neural Network Without Gradientscited by 4
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learningcited by 4
- 2021Single-DARTS: Towards Stable Architecture Searchcited by 3
- 2021Model-Agnostic Meta-Attack: Towards Reliable Evaluation of Adversarial Robustnesscited by 3
- 2024Narrowing the Focus: Learned Optimizers for Pretrained Modelscited by 3
- 2025Efficient End-to-End Learning for Decision-Making: A Meta-Optimization Approachcited by 3
- 2021A Novel Framework for Neural Architecture Search in the Hill Climbing Domaincited by 2
- 2022Learning to learn online with neuromodulated synaptic plasticity in spiking neural networkscited by 2
- 2022EASNet: Searching Elastic and Accurate Network Architecture for Stereo Matchingcited by 2
- 2023Federated Automatic Differentiationcited by 2
- 2023Is Scaling Learned Optimizers Worth It? Evaluating The Value of VeLO's 4000 TPU Monthscited by 2
- 2024Neural Loss Function Evolution for Large-Scale Image Classifier Convolutional Neural Networkscited by 2
- 2024Closed-form Solutions: A New Perspective on Solving Differential Equationscited by 2
50 most cited of 82 All 82 (JSON)
- 2026AI4AI-Bench: Benchmarking LLM Agents in Algorithmic Design for Recursive Self-Improvement20 Aug 2026
- 2026Searching the Space of Feed-Forward Neural-Network Weight-Update Rules with Fixed Depth Symbolic Regression23 Jul 2026
- 2026AI Training Manager: Bounded Closed-Loop Control of Adaptive Training Recipes29 Jun 2026
- 2026OPTScientist: Multi-Agent Discovery of Typed Optimizer Programs for Transformer Pretraining2 Jun 2026
- 2026PILOT: Policy-Informed Learned Optimization for Adaptive Deep Network Training23 May 2026
- 2026Learn2Splat: Extending the Horizon of Learned 3DGS Optimization15 May 2026
- 2026Learning to Discover Iterative Spectral Algorithms10 Feb 2026
- 2025Differentiable Evolutionary Reinforcement Learning15 Dec 2025
- 2025Greedy Alignment Principle for Optimizer Selection6 Dec 2025
- 2025Evolving Deep Learning Optimizers5 Dec 2025
- 2025Towards Explainable Personalized Recommendations by Learning from Users' Photos24 Oct 2025
- 2025A Trainable Optimizer3 Aug 2025
38 more
- 2025Sequential Policy Gradient for Adaptive Hyperparameter Optimization18 Jun 2025
- 2025Efficient End-to-End Learning for Decision-Making: A Meta-Optimization Approach16 May 2025
- 2025On the Duality between Gradient Transformations and Adapters19 Feb 2025
- 2025MIGT: Memory Instance Gated Transformer Framework for Financial Portfolio Management11 Feb 2025
- 2024Lightweight Design and Optimization methods for DCNNs: Progress and Futures22 Dec 2024
- 2024Adaptive Non-uniform Timestep Sampling for Accelerating Diffusion Model Training15 Nov 2024
- 2024Efficient Pretraining Data Selection for Language Models via Multi-Actor Collaboration10 Oct 2024
- 2024Narrowing the Focus: Learned Optimizers for Pretrained Models17 Aug 2024
- 2024The devil is in discretization discrepancy. Robustifying Differentiable NAS with Single-Stage Searching Protocol26 May 2024
- 2024Closed-form Solutions: A New Perspective on Solving Differential Equations23 May 2024
- 2024Neural Optimizer Equation, Decay Function, and Learning Rate Schedule Joint Evolution10 Apr 2024
- 2024Dynamic Memory Based Adaptive Optimization23 Feb 2024
- 2024Guided Evolution with Binary Discriminators for ML Program Search8 Feb 2024
- 2024tnGPS: Discovering Unknown Tensor Network Structure Search Algorithms via Large Language Models (LLMs)4 Feb 2024
- 2024Neural Loss Function Evolution for Large-Scale Image Classifier Convolutional Neural Networks30 Jan 2024
- 2023Is Scaling Learned Optimizers Worth It? Evaluating The Value of VeLO's 4000 TPU Months27 Oct 2023
- 2023Lion Secretly Solves Constrained Optimization: As Lyapunov Predicts9 Oct 2023
- 2023propnet: Propagating 2D Annotation to 3D Segmentation for Gastric Tumors on CT Scans29 May 2023
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learning13 May 2023
- 2023B2Opt: Learning to Optimize Black-box Optimization with Little Budget24 Apr 2023
- 2023Learning To Optimize Quantum Neural Network Without Gradients15 Apr 2023
- 2023MSINet: Twins Contrastive Search of Multi-Scale Interaction for Object ReID13 Mar 2023
- 2023Enhancing Machine Learning Model Performance with Hyper Parameter Optimization: A Comparative Study14 Feb 2023
- 2023Symbolic Discovery of Optimization Algorithms13 Feb 2023
- 2023Federated Automatic Differentiation18 Jan 2023
- 2022Learning to Optimize in Model Predictive Control5 Dec 2022
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Up17 Nov 2022
- 2022Construction of Hierarchical Neural Architecture Search Spaces based on Context-free Grammars3 Nov 2022
- 2022Selecting and Composing Learning Rate Policies for Deep Neural Networks24 Oct 2022
- 2022Efficient Non-Parametric Optimizer Search for Diverse Tasks27 Sep 2022
- 2022Open Source Vizier: Distributed Infrastructure and API for Reliable and Flexible Blackbox Optimization27 Jul 2022
- 2022On the benefits of non-linear weight updates25 Jul 2022
- 2022EASNet: Searching Elastic and Accurate Network Architecture for Stereo Matching20 Jul 2022
- 2022Learning to learn online with neuromodulated synaptic plasticity in spiking neural networks25 Jun 2022
- 2022Finite Expression Method for Solving High-Dimensional Partial Differential Equations21 Jun 2022
- 2022Entangled Residual Mappings2 Jun 2022
- 2022Automated Dynamic Algorithm Configuration27 May 2022
- 2022AutoLossGen: Automatic Loss Function Generation for Recommender Systems27 Apr 2022
50 newest of 82 All 82 (JSON)
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2022Tutorial on amortized optimizationcited by 83
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2021GLiT: Neural Architecture Search for Global and Local Image Transformercited by 53
- 2023Lion Secretly Solves Constrained Optimization: As Lyapunov Predictscited by 50
- 2021Evolving Reinforcement Learning Algorithmscited by 48
- 2022Automated Dynamic Algorithm Configurationcited by 37
- 2021AutoDropout: Learning Dropout Patterns to Regularize Deep Networkscited by 35
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizerscited by 32
- 2021How to Train Your MAML to Excel in Few-Shot Classificationcited by 22
- 2021BN-NAS: Neural Architecture Search with Batch Normalizationcited by 21
12 most cited of 80 All 80 (JSON)
- 2026Searching the Space of Feed-Forward Neural-Network Weight-Update Rules with Fixed Depth Symbolic Regression23 Jul 2026
- 2026AI Training Manager: Bounded Closed-Loop Control of Adaptive Training Recipes29 Jun 2026
- 2026OPTScientist: Multi-Agent Discovery of Typed Optimizer Programs for Transformer Pretraining2 Jun 2026
- 2026PILOT: Policy-Informed Learned Optimization for Adaptive Deep Network Training23 May 2026
- 2026Learn2Splat: Extending the Horizon of Learned 3DGS Optimization15 May 2026
- 2026Learning to Discover Iterative Spectral Algorithms10 Feb 2026
- 2025Differentiable Evolutionary Reinforcement Learning15 Dec 2025
- 2025Greedy Alignment Principle for Optimizer Selection6 Dec 2025
- 2025Evolving Deep Learning Optimizers5 Dec 2025
- 2025Towards Explainable Personalized Recommendations by Learning from Users' Photos24 Oct 2025
- 2025A Trainable Optimizer3 Aug 2025
- 2025Sequential Policy Gradient for Adaptive Hyperparameter Optimization18 Jun 2025
12 newest of 80 All 80 (JSON)