Learning to Optimize Neural Nets
Not on the map. Why?
Abstract (arXiv)
Learning to Optimize is a recently proposed framework for learning optimization algorithms using reinforcement learning. In this paper, we explore learning an optimization algorithm for training shallow neural nets. Such high-dimensional stochastic optimization problems present interesting challenges for existing reinforcement learning algorithms. We develop an extension that is suited to learning optimization algorithms in this setting and demonstrate that the learned optimization algorithm consistently outperforms other known optimization algorithms even on unseen tasks and is robust to changes in stochasticity of gradients and the neural net architecture. More specifically, we show that an optimization algorithm trained with the proposed method on the problem of training a neural net on MNIST generalizes to the problems of training neural nets on the Toronto Faces Dataset, CIFAR-10 and CIFAR-100.
Key facts. Cited by 41 papers in the map (data through 7 Oct 2026). First posted on arXiv 1 Mar 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 1; 2020: 0; 2021: 5; 2022: 10; 2023: 8; 2024: 6; 2025: 9; 2026: 1.
41 papers.
Sort:
By area:
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
- 2022Tutorial on amortized optimizationcited by 83
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2021MetaAlign: Coordinating Domain Alignment and Classification for Unsupervised Domain Adaptationcited by 54
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategiescited by 52
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variablescited by 50
- 2018Learning a Prior over Intent via Meta-Inverse Reinforcement Learningcited by 36
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizerscited by 32
- 2022A Closer Look at Learned Optimization: Stability, Robustness, and Inductive Biasescited by 30
- 2022Symbolic Learning to Optimize: Towards Interpretability and Scalabilitycited by 16
- 2024Artificial Intelligence for Operations Research: Revolutionizing the Operations Research Processcited by 11
29 more
- 2025QuickSplat: Fast 3D Surface Reconstruction via Learned Gaussian Initializationcited by 11
- 2022Learning to Optimize in Model Predictive Controlcited by 10
- 2022Amortized Proximal Optimizationcited by 9
- 2024Large Language Models for Human-Machine Collaborative Particle Accelerator Tuning through Natural Languagecited by 9
- 2023Low-Variance Gradient Estimation in Unrolled Computation Graphs with ES-Singlecited by 7
- 2023Deep Model Predictive Optimizationcited by 7
- 2024Learning to optimize with convergence guarantees using nonlinear system theorycited by 7
- 2023A Nonstochastic Control Approach to Optimizationcited by 4
- 2023Learning To Optimize Quantum Neural Network Without Gradientscited by 4
- 2022Learning to Optimize Quasi-Newton Methodscited by 3
- 2024Learning to Generate Gradients for Test-Time Adaptation via Test-Time Training Layerscited by 3
- 2025Learning a Zeroth-Order Optimizer for Fine-Tuning LLMscited by 3
- 2023Federated Automatic Differentiationcited by 2
- 2023Learning to Do or Learning While Doing: Reinforcement Learning and Bayesian Optimisation for Online Continuous Tuningcited by 2
- 2025Improving Learning to Optimize Using Parameter Symmetriescited by 2
- 2025Deciphering Trajectory-Aided LLM Reasoning: An Optimization Perspectivecited by 2
- 2025A Semi-amortized Lifted Learning-to-Optimize Masked (SALLO-M) Transformer Model for Scalable and Generalizable Beamformingcited by 2
- 2021Efficient Meta Subspace Optimizationcited by 1
- 2021Reinforcement Learning based Sequential Batch-sampling for Bayesian Optimal Experimental Designcited by 1
- 2022Reinforcement Teachingcited by 1
- 2023Subspace Adaptation Prior for Few-Shot Learningcited by 1
- 2025MatRL: Provably Generalizable Iterative Algorithm Discovery via Monte-Carlo Tree Searchcited by 1
- 2023HUB: Guiding Learned Optimizers with Continuous Prompt Tuningcited by 0
- 2024Dynamic Memory Based Adaptive Optimizationcited by 0
- 2024A Memetic Algorithm To Find a Hamiltonian Cycle in a Hamiltonian Graphcited by 0
- 2025Meta-Sparsity: Learning Optimal Sparse Structures in Multi-task Networks through Meta-learningcited by 0
- 2025Optimization Problem Solving Can Transition to Evolutionary Agentic Workflowscited by 0
- 2025Thinking Out of the Box: Hybrid SAT Solving by Unconstrained Continuous Optimizationcited by 0
- 2026Greedy dynamical meta-learningcited by 0
- 2026Greedy dynamical meta-learning27 Jul 2026
- 2025A Semi-amortized Lifted Learning-to-Optimize Masked (SALLO-M) Transformer Model for Scalable and Generalizable Beamforming15 Oct 2025
- 2025Learning a Zeroth-Order Optimizer for Fine-Tuning LLMs1 Oct 2025
- 2025MatRL: Provably Generalizable Iterative Algorithm Discovery via Monte-Carlo Tree Search4 Jul 2025
- 2025Thinking Out of the Box: Hybrid SAT Solving by Unconstrained Continuous Optimization31 May 2025
- 2025Deciphering Trajectory-Aided LLM Reasoning: An Optimization Perspective26 May 2025
- 2025QuickSplat: Fast 3D Surface Reconstruction via Learned Gaussian Initialization8 May 2025
- 2025Optimization Problem Solving Can Transition to Evolutionary Agentic Workflows7 May 2025
- 2025Improving Learning to Optimize Using Parameter Symmetries21 Apr 2025
- 2025Meta-Sparsity: Learning Optimal Sparse Structures in Multi-task Networks through Meta-learning21 Jan 2025
- 2024Learning to Generate Gradients for Test-Time Adaptation via Test-Time Training Layers22 Dec 2024
- 2024Large Language Models for Human-Machine Collaborative Particle Accelerator Tuning through Natural Language14 May 2024
29 more
- 2024Learning to optimize with convergence guarantees using nonlinear system theory14 Mar 2024
- 2024Dynamic Memory Based Adaptive Optimization23 Feb 2024
- 2024A Memetic Algorithm To Find a Hamiltonian Cycle in a Hamiltonian Graph1 Feb 2024
- 2024Artificial Intelligence for Operations Research: Revolutionizing the Operations Research Process6 Jan 2024
- 2023Subspace Adaptation Prior for Few-Shot Learning13 Oct 2023
- 2023Deep Model Predictive Optimization6 Oct 2023
- 2023Learning to Do or Learning While Doing: Reinforcement Learning and Bayesian Optimisation for Online Continuous Tuning6 Jun 2023
- 2023HUB: Guiding Learned Optimizers with Continuous Prompt Tuning26 May 2023
- 2023Low-Variance Gradient Estimation in Unrolled Computation Graphs with ES-Single21 Apr 2023
- 2023Learning To Optimize Quantum Neural Network Without Gradients15 Apr 2023
- 2023A Nonstochastic Control Approach to Optimization19 Jan 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
- 2022Learning to Optimize Quasi-Newton Methods11 Oct 2022
- 2022Learning to Learn with Generative Models of Neural Network Checkpoints26 Sep 2022
- 2022A Closer Look at Learned Optimization: Stability, Robustness, and Inductive Biases22 Sep 2022
- 2022Reinforcement Teaching25 Apr 2022
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizers22 Mar 2022
- 2022Symbolic Learning to Optimize: Towards Interpretability and Scalability13 Mar 2022
- 2022Amortized Proximal Optimization28 Feb 2022
- 2022Tutorial on amortized optimization1 Feb 2022
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategies27 Dec 2021
- 2021Reinforcement Learning based Sequential Batch-sampling for Bayesian Optimal Experimental Design21 Dec 2021
- 2021Efficient Meta Subspace Optimization28 Oct 2021
- 2021MetaAlign: Coordinating Domain Alignment and Classification for Unsupervised Domain Adaptation25 Mar 2021
- 2021Learning to Optimize: A Primer and A Benchmark23 Mar 2021
- 2019Meta-Inverse Reinforcement Learning with Probabilistic Context Variables20 Sep 2019
- 2018Learning a Prior over Intent via Meta-Inverse Reinforcement Learning31 May 2018
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
- 2022Tutorial on amortized optimizationcited by 83
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2021MetaAlign: Coordinating Domain Alignment and Classification for Unsupervised Domain Adaptationcited by 54
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategiescited by 52
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizerscited by 32
- 2022A Closer Look at Learned Optimization: Stability, Robustness, and Inductive Biasescited by 30
- 2022Symbolic Learning to Optimize: Towards Interpretability and Scalabilitycited by 16
- 2024Artificial Intelligence for Operations Research: Revolutionizing the Operations Research Processcited by 11
- 2025QuickSplat: Fast 3D Surface Reconstruction via Learned Gaussian Initializationcited by 11
- 2022Learning to Optimize in Model Predictive Controlcited by 10
12 most cited of 37 All 37 (JSON)
- 2026Greedy dynamical meta-learning27 Jul 2026
- 2025A Semi-amortized Lifted Learning-to-Optimize Masked (SALLO-M) Transformer Model for Scalable and Generalizable Beamforming15 Oct 2025
- 2025Learning a Zeroth-Order Optimizer for Fine-Tuning LLMs1 Oct 2025
- 2025MatRL: Provably Generalizable Iterative Algorithm Discovery via Monte-Carlo Tree Search4 Jul 2025
- 2025Thinking Out of the Box: Hybrid SAT Solving by Unconstrained Continuous Optimization31 May 2025
- 2025QuickSplat: Fast 3D Surface Reconstruction via Learned Gaussian Initialization8 May 2025
- 2025Improving Learning to Optimize Using Parameter Symmetries21 Apr 2025
- 2025Meta-Sparsity: Learning Optimal Sparse Structures in Multi-task Networks through Meta-learning21 Jan 2025
- 2024Learning to Generate Gradients for Test-Time Adaptation via Test-Time Training Layers22 Dec 2024
- 2024Large Language Models for Human-Machine Collaborative Particle Accelerator Tuning through Natural Language14 May 2024
- 2024Learning to optimize with convergence guarantees using nonlinear system theory14 Mar 2024
- 2024Dynamic Memory Based Adaptive Optimization23 Feb 2024
12 newest of 37 All 37 (JSON)