Learning to Optimize
Not on the map. Why?
Abstract (arXiv)
Algorithm design is a laborious process and often requires many iterations of ideation and validation. In this paper, we explore automating algorithm design and present a method to learn an optimization algorithm, which we believe to be the first method that can automatically discover a better algorithm. We approach this problem from a reinforcement learning perspective and represent any particular optimization algorithm as a policy. We learn an optimization algorithm using guided policy search and demonstrate that the resulting algorithm outperforms existing hand-engineered algorithms in terms of convergence speed and/or the final objective value.
Key facts. Cited by 195 papers in the map (data through 7 Oct 2026). First posted on arXiv 6 Jun 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 48; 2022: 40; 2023: 44; 2024: 31; 2025: 21; 2026: 11.
195 papers.
Sort:
By area:
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2021True Few-Shot Learning with Language Modelscited by 343
- 2023Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selectioncited by 291
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2021Trustworthy AI: From Principles to Practicescited by 206
- 2021Investigating Bi-Level Optimization for Learning and Vision from a Unified Perspective: A Survey and Beyondcited by 151
- 2021Hyperparameter Optimization: Foundations, Algorithms, Best Practices and Open Challengescited by 143
- 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
- 2022Tutorial on amortized optimizationcited by 83
- 2021MetaAvatar: Learning Animatable Clothed Human Models from Few Depth Imagescited by 78
38 more
- 2023Permutation Equivariant Neural Functionalscited by 69
- 2023Advances and Challenges in Meta-Learning: A Technical Reviewcited by 65
- 2021Towards Gradient-based Bilevel Optimization with Non-convex Followers and Beyondcited by 62
- 2022Sharp-MAML: Sharpness-Aware Model-Agnostic Meta Learningcited by 57
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategiescited by 52
- 2023Neural Functional Transformerscited by 50
- 2021Multi-Objective Meta Learningcited by 39
- 2022Transformers are Meta-Reinforcement Learnerscited by 39
- 2022Hypernetworks in Meta-Reinforcement Learningcited by 39
- 2022Automated Dynamic Algorithm Configurationcited by 37
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraintscited by 33
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizerscited by 32
- 2022Learning Iterative Reasoning through Energy Minimizationcited by 32
- 2023Learning to Warm-Start Fixed-Point Optimization Algorithmscited by 32
- 2021A Metalearning Approach for Physics-Informed Neural Networks (PINNs): Application to Parameterized PDEscited by 31
- 2021Accelerating Quadratic Optimization with Reinforcement Learningcited by 30
- 2022A Closer Look at Learned Optimization: Stability, Robustness, and Inductive Biasescited by 30
- 2024G3R: Gradient Guided Generalizable Reconstructioncited by 29
- 2023A Systematic Review of Few-Shot Learning in Medical Imagingcited by 28
- 2021A Bit More Bayesian: Domain-Invariant Learning with Uncertaintycited by 27
- 2021How to Train Your MAML to Excel in Few-Shot Classificationcited by 22
- 2022Global Convergence of MAML and Theory-Inspired Neural Architecture Search for Few-Shot Learningcited by 22
- 2021Meta-HAR: Federated Representation Learning for Human Activity Recognitioncited by 21
- 2024Position: Leverage Foundational Models for Black-Box Optimizationcited by 21
- 2022End-to-End Stochastic Optimization with Energy-Based Modelcited by 19
- 2024Offline Model-Based Optimization via Policy-Guided Gradient Searchcited by 19
- 2021Out-of-Domain Generalization from a Single Source: An Uncertainty Quantification Approachcited by 18
- 2024Verbalized Machine Learning: Revisiting Machine Learning with Language Modelscited by 17
- 2022Symbolic Learning to Optimize: Towards Interpretability and Scalabilitycited by 16
- 2024In-context Exploration-Exploitation for Reinforcement Learningcited by 16
- 2023B2Opt: Learning to Optimize Black-box Optimization with Little Budgetcited by 15
- 2024Can Learned Optimization Make Reinforcement Learning Less Difficult?cited by 14
- 2021Graph neural network initialisation of quantum approximate optimisationcited by 13
- 2021Automated Deep Learning: Neural Architecture Search Is Not the Endcited by 13
- 2022Transformer-Based Learned Optimizationcited by 13
- 2023Towards Constituting Mathematical Structures for Learning to Optimizecited by 13
- 2022A Deep Conjugate Direction Method for Iteratively Solving Linear Systemscited by 12
50 most cited of 195 All 195 (JSON)
- 2026SOLAR: A State-Driven Online Learning Rate Scheduler for LLM Pretraining28 Sep 2026
- 2026Reinforcement learning to choose optimizers1 Sep 2026
- 2026Few-Shot Open-Set Audio Classification Using Attention Information-Fused Prototypes1 Jul 2026
- 2026AI Training Manager: Bounded Closed-Loop Control of Adaptive Training Recipes29 Jun 2026
- 2026Scalable Deep Unfolding of Conic Optimizers11 Jun 2026
- 2026Reinforcement Learning for Neural Model Editing11 Jun 2026
- 2026Regret Equals Covariance: A Closed-Form Characterization for Stochastic Optimization13 May 2026
- 2026Learning to Optimize Radiotherapy Plans via Fluence Maps Diffusion Model Generation and LSTM-based Optimization13 May 2026
- 2026Learning to Test: Physics-Informed Representation for Dynamical Instability Detection13 Apr 2026
- 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimization10 Mar 2026
- 2026Majorization-Minimization Networks for Inverse Problems: An Application to EEG Imaging23 Jan 2026
- 2025A Model-Guided Neural Network Method for the Inverse Scattering Problem10 Dec 2025
38 more
- 2025Evolving Deep Learning Optimizers5 Dec 2025
- 2025Gradient Descent with Provably Tuned Learning-rate Schedules4 Dec 2025
- 2025SemanticOpt: Towards LLM-Based Semantic Black-Box Optimization29 Oct 2025
- 2025Beyond-Diagonal RIS Under Non-Idealities: Learning-Based Architecture Discovery and Optimization17 Oct 2025
- 2025ZeroShotOpt: Towards Zero-Shot Pretrained Models for Efficient Black-Box Optimization3 Oct 2025
- 2025Learning a Zeroth-Order Optimizer for Fine-Tuning LLMs1 Oct 2025
- 2025Bundle Network: a Machine Learning-Based Bundle Method29 Sep 2025
- 2025Reinforcement learning for adaptive interior point methods in convex quadratic programming9 Sep 2025
- 2025L-SR1: Learned Symmetric-Rank-One Preconditioning17 Aug 2025
- 2025A Trainable Optimizer3 Aug 2025
- 2025Learning to optimize with guarantees: a complete characterization of linearly convergent algorithms1 Aug 2025
- 2025MACS: Multi-Agent Reinforcement Learning for Optimization of Crystal Structures4 Jun 2025
- 2025Deciphering Trajectory-Aided LLM Reasoning: An Optimization Perspective26 May 2025
- 2025Efficient End-to-End Learning for Decision-Making: A Meta-Optimization Approach16 May 2025
- 2025Improving Learning to Optimize Using Parameter Symmetries21 Apr 2025
- 2025Make Optimization Once and for All with Fine-grained Guidance14 Mar 2025
- 2025Energy-Latency Attacks: A New Adversarial Threat to Deep Learning6 Mar 2025
- 2025On the Duality between Gradient Transformations and Adapters19 Feb 2025
- 2025Diffusion-based learning framework for Constrained Nonconvex Optimization with Weighted Bootstrapped Refinement14 Feb 2025
- 2025Energy Backdoor Attack to Deep Neural Networks14 Jan 2025
- 2024Learning to Generate Gradients for Test-Time Adaptation via Test-Time Training Layers22 Dec 2024
- 2024Reinforcement learning to learn quantum states for Heisenberg scaling accuracy3 Dec 2024
- 2024Adaptive Non-uniform Timestep Sampling for Accelerating Diffusion Model Training15 Nov 2024
- 2024Convergence Rate Analysis of LION12 Nov 2024
- 2024Gradient Methods with Online Scaling4 Nov 2024
- 2024Transferable Sequential Recommendation via Vector Quantized Meta Learning4 Nov 2024
- 2024A neural network approach for solving the Monge-Amp`ere equation with transport boundary condition25 Oct 2024
- 2024On the Learn-to-Optimize Capabilities of Transformers in In-Context Sparse Recovery17 Oct 2024
- 2024Learning a Neural Solver for Parametric PDE to Enhance Physics-Informed Methods9 Oct 2024
- 2024G3R: Gradient Guided Generalizable Reconstruction28 Sep 2024
- 2024Self-Supervised Learning of Iterative Solvers for Constrained Optimization12 Sep 2024
- 2024A New First-Order Meta-Learning Algorithm with Convergence Guarantees5 Sep 2024
- 2024Narrowing the Focus: Learned Optimizers for Pretrained Models17 Aug 2024
- 2024Non-Asymptotic Uncertainty Quantification in High-Dimensional Learning18 Jul 2024
- 2024Can Learned Optimization Make Reinforcement Learning Less Difficult?9 Jul 2024
- 2024MetaFollower: Adaptable Personalized Autonomous Car Following23 Jun 2024
- 2024Primitive Agentic First-Order Optimization7 Jun 2024
- 2024Verbalized Machine Learning: Revisiting Machine Learning with Language Models6 Jun 2024
50 newest of 195 All 195 (JSON)
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraintscited by 33
- 2024Adaptive Non-uniform Timestep Sampling for Accelerating Diffusion Model Trainingcited by 8
- 2025Diffusion-based learning framework for Constrained Nonconvex Optimization with Weighted Bootstrapped Refinementcited by 1
- 2026Learning to Optimize Radiotherapy Plans via Fluence Maps Diffusion Model Generation and LSTM-based Optimizationcited by 0
- 2026Learning to Optimize Radiotherapy Plans via Fluence Maps Diffusion Model Generation and LSTM-based Optimization13 May 2026
- 2025Diffusion-based learning framework for Constrained Nonconvex Optimization with Weighted Bootstrapped Refinement14 Feb 2025
- 2024Adaptive Non-uniform Timestep Sampling for Accelerating Diffusion Model Training15 Nov 2024
- 2024Diffusion Models as Constrained Samplers for Optimization with Unknown Constraints28 Feb 2024
- 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimizationcited by 9
- 2024Primitive Agentic First-Order Optimizationcited by 0
- 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimization10 Mar 2026
- 2024Primitive Agentic First-Order Optimization7 Jun 2024
- 2026Reinforcement Learning for Neural Model Editingcited by 0
- 2026Reinforcement Learning for Neural Model Editing11 Jun 2026
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2021True Few-Shot Learning with Language Modelscited by 343
- 2023Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selectioncited by 291
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2021Trustworthy AI: From Principles to Practicescited by 206
- 2021Investigating Bi-Level Optimization for Learning and Vision from a Unified Perspective: A Survey and Beyondcited by 151
- 2021Hyperparameter Optimization: Foundations, Algorithms, Best Practices and Open Challengescited by 143
- 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
- 2022Tutorial on amortized optimizationcited by 83
- 2021MetaAvatar: Learning Animatable Clothed Human Models from Few Depth Imagescited by 78
12 most cited of 185 All 185 (JSON)
- 2026SOLAR: A State-Driven Online Learning Rate Scheduler for LLM Pretraining28 Sep 2026
- 2026Reinforcement learning to choose optimizers1 Sep 2026
- 2026Few-Shot Open-Set Audio Classification Using Attention Information-Fused Prototypes1 Jul 2026
- 2026AI Training Manager: Bounded Closed-Loop Control of Adaptive Training Recipes29 Jun 2026
- 2026Scalable Deep Unfolding of Conic Optimizers11 Jun 2026
- 2026Regret Equals Covariance: A Closed-Form Characterization for Stochastic Optimization13 May 2026
- 2026Learning to Test: Physics-Informed Representation for Dynamical Instability Detection13 Apr 2026
- 2026Majorization-Minimization Networks for Inverse Problems: An Application to EEG Imaging23 Jan 2026
- 2025A Model-Guided Neural Network Method for the Inverse Scattering Problem10 Dec 2025
- 2025Evolving Deep Learning Optimizers5 Dec 2025
- 2025Gradient Descent with Provably Tuned Learning-rate Schedules4 Dec 2025
- 2025SemanticOpt: Towards LLM-Based Semantic Black-Box Optimization29 Oct 2025
12 newest of 185 All 185 (JSON)