Forward and Reverse Gradient-Based Hyperparameter Optimization
Not on the map. Why?
Abstract (arXiv)
We study two procedures (reverse-mode and forward-mode) for computing the gradient of the validation error with respect to the hyperparameters of any iterative learning algorithm such as stochastic gradient descent. These procedures mirror two methods of computing gradients for recurrent neural networks and have different trade-offs in terms of running time and space requirements. Our formulation of the reverse-mode procedure is linked to previous work by Maclaurin et al. [2015] but does not require reversible dynamics. The forward-mode procedure is suitable for real-time hyperparameter updates, which may significantly speed up hyperparameter optimization on large datasets. We present experiments on data cleaning and on learning task interactions. We also present one large-scale experiment where the use of previous gradient-based methods would be prohibitive.
Key facts. Cited by 234 papers in the map (data through 14 Oct 2026). First posted on arXiv 6 Mar 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 40; 2022: 48; 2023: 51; 2024: 44; 2025: 30; 2026: 21.
234 papers.
Sort:
By area:
- 2021Graph Contrastive Learning Automatedcited by 287
- 2021Efficient and Modular Implicit Differentiationcited by 203
- 2021Discrete Graph Structure Learning for Forecasting Multiple Time Seriescited by 171
- 2021Investigating Bi-Level Optimization for Learning and Vision from a Unified Perspective: A Survey and Beyondcited by 153
- 2021Hyperparameter Optimization: Foundations, Algorithms, Best Practices and Open Challengescited by 145
- 2023Dataset Distillation: A Comprehensive Reviewcited by 127
- 2021A Near-Optimal Algorithm for Stochastic Bilevel Optimization via Double-Momentumcited by 106
- 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
- 2022BOME! Bilevel Optimization Made Easy: A Simple First-Order Approachcited by 105
- 2022Remember the Past: Distilling Datasets into Addressable Memories for Neural Networkscited by 104
- 2022A framework for bilevel optimization that enables stochastic and global variance reduction algorithmscited by 99
- 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
38 more
- 2022Tutorial on amortized optimizationcited by 83
- 2021On Training Implicit Modelscited by 75
- 2023Using Large Language Models for Hyperparameter Optimizationcited by 71
- 2024SEAL: Safety-enhanced Aligned LLM Fine-tuning via Bilevel Data Selectioncited by 64
- 2021Towards Gradient-based Bilevel Optimization with Non-convex Followers and Beyondcited by 63
- 2021Amortized Implicit Differentiation for Stochastic Bilevel Optimizationcited by 63
- 2021A Value-Function-based Interior-point Method for Non-convex Bi-level Optimizationcited by 60
- 2022FedNest: Federated Bilevel, Minimax, and Compositional Optimizationcited by 60
- 2021An End-to-End Framework for Molecular Conformation Generation via Bilevel Programmingcited by 59
- 2023Revisiting Weighted Aggregation in Federated Learning with Neural Networkscited by 58
- 2022Advancing Model Pruning via Bi-level Optimizationcited by 56
- 2021A Fully Single Loop Algorithm for Bilevel Optimization without Hessian Inversecited by 53
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategiescited by 52
- 2021iDARTS: Differentiable Architecture Search with Stochastic Implicit Gradientscited by 51
- 2022Self-Guided Noise-Free Data Generation for Efficient Zero-Shot Learningcited by 50
- 2021Lower Bounds and Accelerated Algorithms for Bilevel Optimizationcited by 46
- 2023Optimal Linear Decay Learning Rate Schedules and Further Refinementscited by 43
- 2023First-order penalty methods for bilevel optimizationcited by 41
- 2023An Introduction to Bi-level Optimization: Foundations and Applications in Signal Processing and Machine Learningcited by 41
- 2024Principled Penalty-based Methods for Bilevel Reinforcement Learning and RLHFcited by 41
- 2023On Finding Small Hyper-Gradients in Bilevel Optimization: Hardness Results and Improved Analysiscited by 39
- 2021HYDRA: Hypergradient Data Relevance Analysis for Interpreting Deep Neural Networkscited by 38
- 2024ScaleBiO: Scalable Bilevel Optimization for LLM Data Reweightingcited by 37
- 2022Will Bilevel Optimizers Benefit from Loopscited by 36
- 2022Online Knowledge Distillation via Mutual Contrastive Learning for Visual Recognitioncited by 36
- 2021On the Convergence Theory for Hessian-Free Bilevel Algorithmscited by 35
- 2022Learning the Effect of Registration Hyperparameters with HyperMorphcited by 35
- 2023AutoML in the Age of Large Language Models: Current Challenges, Future Opportunities and Riskscited by 35
- 2022Dataset Condensation with Latent Space Knowledge Factorization and Sharingcited by 34
- 2021Stability and Generalization of Bilevel Programming in Hyperparameter Optimizationcited by 31
- 2023Optimal Algorithms for Stochastic Bilevel Optimization under Relaxed Smoothness Conditionscited by 31
- 2021A General Descent Aggregation Framework for Gradient-based Bi-level Optimizationcited by 28
- 2022Betty: An Automatic Differentiation Library for Multilevel Optimizationcited by 27
- 2023Probabilistic Bilevel Coreset Selectioncited by 27
- 2022Gradient-based Bi-level Optimization for Deep Learning: A Surveycited by 26
- 2023Averaged Method of Multipliers for Bi-Level Optimization without Lower-Level Strong Convexitycited by 26
- 2021Value-Function-based Sequential Minimization for Bi-level Optimizationcited by 24
- 2024Super Consistency of Neural Network Landscapes and Learning Rate Transfercited by 23
50 most cited of 234 All 234 (JSON)
- 2026Probabilistic Multi-Source Data Selection via Learnable Categorical Priors: Variance Reduction and Guaranteed Derandomization10 Oct 2026
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026ABO-Med: Accelerated Bilevel Optimization for Few-Shot Medical Image Classification27 Sep 2026
- 2026Efficient Hessian-Free Methods for Multi-Objective Bilevel Optimization with Nonconvex Lower Level13 Aug 2026
- 2026Sharper Analysis of Single-Loop Methods for Bilevel Optimization11 Jul 2026
- 2026Escaping the Variance Trap: Jacobian-Free Dynamics for Root-Finding Bilevel Optimization21 Jun 2026
- 2026On the Stability and Generalization of First-order Bilevel Minimax Optimization22 Apr 2026
- 2026Meta Additive Model: Interpretable Sparse Learning With Auto Weighting22 Apr 2026
- 2026Fine-grained Analysis of Stability and Generalization for Stochastic Bilevel Optimization5 Apr 2026
- 2026Efficient Bilevel Optimization with KFAC-Based Hypergradients31 Mar 2026
- 2026Robust Unscented Kalman Filtering via Recurrent Meta-Adaptation of Sigma-Point Weights4 Mar 2026
- 2026On the Convergence of Single-Loop Stochastic Bilevel Optimization with Approximate Implicit Differentiation27 Feb 2026
38 more
- 2026Dynamic Symmetric Point Tracking: Tackling Non-ideal Reference in Analog In-memory Training24 Feb 2026
- 2026Fully First-Order Algorithms for Online Bilevel Optimization12 Feb 2026
- 2026Natural Hypergradient Descent: Algorithm Design, Convergence Analysis, and Parallel Implementation11 Feb 2026
- 2026Achieving Better Local Regret Bound for Online Non-Convex Bilevel Optimization6 Feb 2026
- 2026Bilevel Optimization with Lower-Level Uniform Convexity: Theory and Algorithm4 Feb 2026
- 2026FlowSymm: Physics Aware, Symmetry Preserving Graph Attention for Network Flow Completion29 Jan 2026
- 2026Stealthy Poisoning Attacks Bypass Defenses in Regression Settings29 Jan 2026
- 2026Non-Stationary Functional Bilevel Optimization21 Jan 2026
- 2026Optimal Transport under Group Fairness Constraints12 Jan 2026
- 2025Greedy Alignment Principle for Optimizer Selection6 Dec 2025
- 2025Bilevel Data Curation for LLM Fine-tuning: Offline Selection and Online Self-Refining Generation26 Nov 2025
- 2025Lower Complexity Bounds for Nonconvex-Strongly-Convex Bilevel Optimization with First-Order Oracles24 Nov 2025
- 2025Bridging Constraints and Stochasticity: A Fully First-Order Method for Stochastic Bilevel Optimization with Linear Constraints13 Nov 2025
- 2025Solving bilevel optimization via sequential minimax optimization10 Nov 2025
- 2025Cost-Sensitive Freeze-thaw Bayesian Optimization for Efficient Hyperparameter Tuning24 Oct 2025
- 2025Computational Budget Should Be Considered in Data Selection19 Oct 2025
- 2025From Data to Rewards: a Bilevel Optimization Perspective on Maximum Likelihood Estimation8 Oct 2025
- 2025Adaptive Algorithms with Sharp Convergence Rates for Stochastic Hierarchical Optimization18 Sep 2025
- 2025Privileged Contrastive Pretraining for Multimodal Affect Modelling30 Jul 2025
- 2025Optimizing Canaries for Privacy Auditing with Metagradient Descent21 Jul 2025
- 2025You Only Train Once4 Jun 2025
- 2025Rethinking Data Mixture for Large Language Models: A Comprehensive Survey and New Perspectives27 May 2025
- 2025Sequential QCQP for Bilevel Optimization with Line Search20 May 2025
- 2025POCAII: Parameter Optimization with Conscious Allocation using Iterative Intelligence16 May 2025
- 2025Argus: Federated Non-convex Bilevel Learning over 6G Space-Air-Ground Integrated Network14 May 2025
- 2025A stochastic gradient method for trilevel optimization11 May 2025
- 2025Efficient Curvature-Aware Hypergradient Approximation for Bilevel Optimization4 May 2025
- 2025Scalable Meta-Learning via Mixed-Mode Differentiation1 May 2025
- 2025MAGIC: Near-Optimal Data Attribution for Deep Learning23 Apr 2025
- 2025Frozen Layers: Memory-efficient Many-fidelity Hyperparameter Optimization14 Apr 2025
- 2025PRDP: Progressively Refined Differentiable Physics26 Feb 2025
- 2025Sparse Hyperparametric Itakura-Saito Nonnegative Matrix Factorization via Bi-Level Optimization24 Feb 2025
- 2025A First-order Generative Bilevel Optimization Framework for Diffusion Models12 Feb 2025
- 2025LDC-MTL: Balancing Multi-Task Learning through Scalable Loss Discrepancy Control12 Feb 2025
- 2025Riemannian Manifold Learning for Stackelberg Games with Neural Flow Representations8 Feb 2025
- 2025Bilevel ZOFO: Efficient LLM Fine-Tuning and Meta-Training5 Feb 2025
- 2025qNBO: quasi-Newton Meets Bilevel Optimization3 Feb 2025
- 2025Adaptive Width Neural Networks27 Jan 2025
50 newest of 234 All 234 (JSON)
- 2024Bilevel reinforcement learning via the development of hyper-gradient without lower-level convexitycited by 10
- 2026Fine-grained Analysis of Stability and Generalization for Stochastic Bilevel Optimizationcited by 3
- 2024Meta-Reinforcement Learning with Universal Policy Adaptation: Provable Near-Optimality under All-task Optimum Comparatorcited by 2
- 2025From Data to Rewards: a Bilevel Optimization Perspective on Maximum Likelihood Estimationcited by 0
- 2026On the Stability and Generalization of First-order Bilevel Minimax Optimizationcited by 0
- 2026Escaping the Variance Trap: Jacobian-Free Dynamics for Root-Finding Bilevel Optimizationcited by 0
- 2026Sharper Analysis of Single-Loop Methods for Bilevel Optimizationcited by 0
- 2026Sharper Analysis of Single-Loop Methods for Bilevel Optimization11 Jul 2026
- 2026Escaping the Variance Trap: Jacobian-Free Dynamics for Root-Finding Bilevel Optimization21 Jun 2026
- 2026On the Stability and Generalization of First-order Bilevel Minimax Optimization22 Apr 2026
- 2026Fine-grained Analysis of Stability and Generalization for Stochastic Bilevel Optimization5 Apr 2026
- 2025From Data to Rewards: a Bilevel Optimization Perspective on Maximum Likelihood Estimation8 Oct 2025
- 2024Meta-Reinforcement Learning with Universal Policy Adaptation: Provable Near-Optimality under All-task Optimum Comparator13 Oct 2024
- 2024Bilevel reinforcement learning via the development of hyper-gradient without lower-level convexity30 May 2024
- 2023Sharpness-Aware Data Poisoning Attackcited by 10
- 2024Machine Unlearning for Traditional Models and Large Language Models: A Short Surveycited by 10
- 2025Optimizing Canaries for Privacy Auditing with Metagradient Descentcited by 6
- 2023Hyperparameter Learning under Data Poisoning: Analysis of the Influence of Regularization via Multiobjective Bilevel Optimizationcited by 3
- 2025Optimizing Canaries for Privacy Auditing with Metagradient Descent21 Jul 2025
- 2024Machine Unlearning for Traditional Models and Large Language Models: A Short Survey1 Apr 2024
- 2023Hyperparameter Learning under Data Poisoning: Analysis of the Influence of Regularization via Multiobjective Bilevel Optimization2 Jun 2023
- 2023Sharpness-Aware Data Poisoning Attack24 May 2023
- 2021Graph Contrastive Learning Automatedcited by 287
- 2021Efficient and Modular Implicit Differentiationcited by 203
- 2021Discrete Graph Structure Learning for Forecasting Multiple Time Seriescited by 171
- 2021Investigating Bi-Level Optimization for Learning and Vision from a Unified Perspective: A Survey and Beyondcited by 153
- 2021Hyperparameter Optimization: Foundations, Algorithms, Best Practices and Open Challengescited by 145
- 2023Dataset Distillation: A Comprehensive Reviewcited by 127
- 2021A Near-Optimal Algorithm for Stochastic Bilevel Optimization via Double-Momentumcited by 106
- 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
- 2022BOME! Bilevel Optimization Made Easy: A Simple First-Order Approachcited by 105
- 2022Remember the Past: Distilling Datasets into Addressable Memories for Neural Networkscited by 104
- 2022A framework for bilevel optimization that enables stochastic and global variance reduction algorithmscited by 99
- 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
12 most cited of 216 All 216 (JSON)
- 2026Probabilistic Multi-Source Data Selection via Learnable Categorical Priors: Variance Reduction and Guaranteed Derandomization10 Oct 2026
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026ABO-Med: Accelerated Bilevel Optimization for Few-Shot Medical Image Classification27 Sep 2026
- 2026Efficient Hessian-Free Methods for Multi-Objective Bilevel Optimization with Nonconvex Lower Level13 Aug 2026
- 2026Meta Additive Model: Interpretable Sparse Learning With Auto Weighting22 Apr 2026
- 2026Efficient Bilevel Optimization with KFAC-Based Hypergradients31 Mar 2026
- 2026Robust Unscented Kalman Filtering via Recurrent Meta-Adaptation of Sigma-Point Weights4 Mar 2026
- 2026On the Convergence of Single-Loop Stochastic Bilevel Optimization with Approximate Implicit Differentiation27 Feb 2026
- 2026Dynamic Symmetric Point Tracking: Tackling Non-ideal Reference in Analog In-memory Training24 Feb 2026
- 2026Fully First-Order Algorithms for Online Bilevel Optimization12 Feb 2026
- 2026Natural Hypergradient Descent: Algorithm Design, Convergence Analysis, and Parallel Implementation11 Feb 2026
- 2026Achieving Better Local Regret Bound for Online Non-Convex Bilevel Optimization6 Feb 2026
12 newest of 216 All 216 (JSON)