Gradient-based Hyperparameter Optimization through Reversible Learning
Not on the map. Why?
Abstract (arXiv)
Tuning hyperparameters of learning algorithms is hard because gradients are usually unavailable. We compute exact gradients of cross-validation performance with respect to all hyperparameters by chaining derivatives backwards through the entire training procedure. These gradients allow us to optimize thousands of hyperparameters, including step-size and momentum schedules, weight initialization distributions, richly parameterized regularization schemes, and neural network architectures. We compute hyperparameter gradients by exactly reversing the dynamics of stochastic gradient descent with momentum.
Key facts. Cited by 357 papers in the map (data through 7 Oct 2026). First posted on arXiv 11 Feb 2015.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 69; 2022: 87; 2023: 69; 2024: 52; 2025: 43; 2026: 37.
357 papers.
Sort:
By area:
- 2022Dataset Distillation by Matching Training Trajectoriescited by 437
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
- 2021Graph Contrastive Learning Automatedcited by 287
- 2022Dataset Condensation via Efficient Synthetic-Data Parameterizationcited by 187
- 2022Dataset Distillation using Neural Feature Regressioncited by 177
- 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
- 2022Minimizing the Accumulated Trajectory Error to Improve Dataset Distillationcited by 142
- 2023Dataset Distillation: A Comprehensive Reviewcited by 126
- 2023A Comprehensive Survey of Dataset Distillationcited by 114
- 2021HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPOcited by 111
- 2022Efficient Dataset Distillation Using Random Feature Approximationcited by 111
38 more
- 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
- 2022A framework for bilevel optimization that enables stochastic and global variance reduction algorithmscited by 99
- 2022Wild Patterns Reloaded: A Survey of Machine Learning Security against Training Data Poisoningcited by 93
- 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
- 2023Data Distillation: A Surveycited by 86
- 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
- 2021HyperMorph: Amortized Hyperparameter Learning for Image Registrationcited by 84
- 2022Tutorial on amortized optimizationcited by 83
- 2021Gradients are Not All You Needcited by 82
- 2021On Training Implicit Modelscited by 73
- 2022Auto-Lambda: Disentangling Dynamic Task Relationshipscited by 73
- 2021Learning Normal Dynamics in Videos with Meta Prototype Networkcited by 71
- 2023Using Large Language Models for Hyperparameter Optimizationcited by 71
- 2022Theseus: A Library for Differentiable Nonlinear Optimizationcited by 64
- 2021Towards Gradient-based Bilevel Optimization with Non-convex Followers and Beyondcited by 62
- 2022Accelerating Dataset Distillation via Model Augmentationcited by 62
- 2022FedNest: Federated Bilevel, Minimax, and Compositional Optimizationcited by 60
- 2021An End-to-End Framework for Molecular Conformation Generation via Bilevel Programmingcited by 59
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2023Revisiting Weighted Aggregation in Federated Learning with Neural Networkscited by 57
- 2021A Fully Single Loop Algorithm for Bilevel Optimization without Hessian Inversecited by 53
- 2025End-to-End Test-Time Training for Long Contextcited by 53
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategiescited by 52
- 2024Erasing Undesirable Influence in Diffusion Modelscited by 52
- 2021Soft-Label Anonymous Gastric X-ray Image Distillationcited by 51
- 2021iDARTS: Differentiable Architecture Search with Stochastic Implicit Gradientscited by 51
- 2022Model-Based Deep Learning: On the Intersection of Deep Learning and Optimizationcited by 50
- 2023Class-Incremental Exemplar Compression for Class-Incremental Learningcited by 50
- 2022AutoBalance: Optimized Loss Functions for Imbalanced Datacited by 49
- 2022Self-Guided Noise-Free Data Generation for Efficient Zero-Shot Learningcited by 49
- 2021Lower Bounds and Accelerated Algorithms for Bilevel Optimizationcited by 46
- 2023Dataset Distillation with Convexified Implicit Gradientscited by 46
- 2021Bootstrapped Meta-Learningcited by 45
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learningcited by 43
- 2024Principled Penalty-based Methods for Bilevel Reinforcement Learning and RLHFcited by 41
- 2023First-order penalty methods for bilevel optimizationcited by 40
- 2023An Introduction to Bi-level Optimization: Foundations and Applications in Signal Processing and Machine Learningcited by 39
- 2021HYDRA: Hypergradient Data Relevance Analysis for Interpreting Deep Neural Networkscited by 38
50 most cited of 357 All 357 (JSON)
- 2026Trust the Direction, Search the Step: Zero-and-First-Order Methods for LLM Fine-Tuning1 Oct 2026
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026ABO-Med: Accelerated Bilevel Optimization for Few-Shot Medical Image Classification27 Sep 2026
- 2026Intrinsic-Extrinsic Coupling in Learning Dynamics24 Sep 2026
- 2026Recursive self-improvement of AI research agents22 Sep 2026
- 2026TBR: Transport-Based Rendering with Deposition Strokes for Inverse Graphics8 Sep 2026
- 2026Disciplined Bilevel Programming1 Sep 2026
- 2026Stored in Optimizer State, Valued by Later Training: A Causal Account of Subliminal Trait Transfer20 Aug 2026
- 2026Sharper Analysis of Single-Loop Methods for Bilevel Optimization11 Jul 2026
- 2026Epistemic Goggles: A Pretrained Module that Induces an Epistemic Frame via Gradient Editing2 Jul 2026
- 2026Revocable Learned State via Process Sidecars29 Jun 2026
- 2026Bilevel Optimization for Neural Architecture Search28 Jun 2026
38 more
- 2026JKO-RAG: Distributional Retrieval as Wasserstein Free-Energy Gradient Flow15 Jun 2026
- 2026FastMix: Fast Data Mixture Optimization via Gradient Descent12 Jun 2026
- 2026Efficient Hyperparameter Optimization for LLM Reinforcement Learning2 Jun 2026
- 2026On the Difficulty of Learning a Meta-network for Training Data Selection30 May 2026
- 2026Detectability in Diversity: Improved Canary Crafting for Privacy Auditing in One Run26 May 2026
- 2026Semiparametric Efficient Bilevel Gradient Estimation20 May 2026
- 2026The Thermodynamic Costs of Simple Linear Regression18 May 2026
- 2026When Losses Align: Gradient-Based Composite Loss Weighting for Efficient Pretraining8 May 2026
- 2026NoiseRater: Meta-Learned Noise Valuation for Diffusion Model Training2 May 2026
- 2026Generalization Guarantees on Data-Driven Tuning of Gradient Descent with Langevin Updates13 Apr 2026
- 2026What do near-optimal learning rate schedules look like?11 Mar 2026
- 2026BiEvLight: Bi-level Learning of Task-Aware Event Refinement for Low-Light Image Enhancement5 Mar 2026
- 2026Dynamic Symmetric Point Tracking: Tackling Non-ideal Reference in Analog In-memory Training24 Feb 2026
- 2026Oracle-Robust Online Alignment for Large Language Models24 Feb 2026
- 2026Exploring the Performance of ML/DL Architectures on the MNIST-1D Dataset12 Feb 2026
- 2026Fully First-Order Algorithms for Online Bilevel Optimization12 Feb 2026
- 2026SkillRater: Untangling Capabilities in Multimodal Data12 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
- 2026Stealthy Poisoning Attacks Bypass Defenses in Regression Settings29 Jan 2026
- 2026Learning What to Predict: Downstream-Guided Task Design for Continued Pretraining29 Jan 2026
- 2026TBDFiltering: Sample-Efficient Tree-Based Data Filtering29 Jan 2026
- 2026Beyond Objective-Based Improvement: Stationarity-Aware Expected Improvement for Bayesian Optimization29 Jan 2026
- 2026Teaching Models to Teach Themselves: Reasoning at the Edge of Learnability26 Jan 2026
- 2026Optimal Transport under Group Fairness Constraints12 Jan 2026
- 2025End-to-End Test-Time Training for Long Context29 Dec 2025
- 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Duration26 Dec 2025
- 2025Generative Bayesian Hyperparameter Tuning23 Dec 2025
- 2025Greedy Alignment Principle for Optimizer Selection6 Dec 2025
- 2025Neural Coherence : Find higher performance to out-of-distribution tasks from few samples5 Dec 2025
- 2025Gradient Descent with Provably Tuned Learning-rate Schedules4 Dec 2025
- 2025A Fully First-Order Layer for Differentiable Optimization2 Dec 2025
- 2025Lower Complexity Bounds for Nonconvex-Strongly-Convex Bilevel Optimization with First-Order Oracles24 Nov 2025
- 2025Solving bilevel optimization via sequential minimax optimization10 Nov 2025
- 2025Bilevel Learning via Inexact Stochastic Gradient Descent10 Nov 2025
- 2025Bilevel Models for Adversarial Learning and A Case Study29 Oct 2025
- 2025A Single-Loop First-Order Algorithm for Linearly Constrained Bilevel Optimization28 Oct 2025
- 2025Scalable Principal-Agent Contract Design via Gradient-Based Optimization24 Oct 2025
50 newest of 357 All 357 (JSON)
- 2024Erasing Undesirable Influence in Diffusion Modelscited by 52
- 2024Generative Dataset Distillation Based on Diffusion Modelcited by 13
- 2025Towards Resilient Safety-driven Unlearning for Diffusion Models against Downstream Fine-tuningcited by 8
- 2024One Category One Prompt: Dataset Distillation using Diffusion Modelscited by 5
- 2025A First-order Generative Bilevel Optimization Framework for Diffusion Modelscited by 4
- 2026NoiseRater: Meta-Learned Noise Valuation for Diffusion Model Trainingcited by 1
- 2026NoiseRater: Meta-Learned Noise Valuation for Diffusion Model Training2 May 2026
- 2025Towards Resilient Safety-driven Unlearning for Diffusion Models against Downstream Fine-tuning22 Jul 2025
- 2025A First-order Generative Bilevel Optimization Framework for Diffusion Models12 Feb 2025
- 2024Generative Dataset Distillation Based on Diffusion Model16 Aug 2024
- 2024One Category One Prompt: Dataset Distillation using Diffusion Models11 Mar 2024
- 2024Erasing Undesirable Influence in Diffusion Models11 Jan 2024
- 2022Wild Patterns Reloaded: A Survey of Machine Learning Security against Training Data Poisoningcited by 93
- 2025BLUR: A Bi-Level Optimization Approach for LLM Unlearningcited by 23
- 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
- 2026Detectability in Diversity: Improved Canary Crafting for Privacy Auditing in One Runcited by 0
- 2026Detectability in Diversity: Improved Canary Crafting for Privacy Auditing in One Run26 May 2026
- 2025Optimizing Canaries for Privacy Auditing with Metagradient Descent21 Jul 2025
- 2025BLUR: A Bi-Level Optimization Approach for LLM Unlearning9 Jun 2025
- 2023Hyperparameter Learning under Data Poisoning: Analysis of the Influence of Regularization via Multiobjective Bilevel Optimization2 Jun 2023
- 2022Wild Patterns Reloaded: A Survey of Machine Learning Security against Training Data Poisoning4 May 2022
- 2026Recursive self-improvement of AI research agentscited by 3
- 2026Recursive self-improvement of AI research agents22 Sep 2026
- 2022Dataset Distillation by Matching Training Trajectoriescited by 437
- 2021Graph Contrastive Learning Automatedcited by 287
- 2022Dataset Condensation via Efficient Synthetic-Data Parameterizationcited by 187
- 2022Dataset Distillation using Neural Feature Regressioncited by 177
- 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
- 2022Minimizing the Accumulated Trajectory Error to Improve Dataset Distillationcited by 142
- 2023Dataset Distillation: A Comprehensive Reviewcited by 126
- 2023A Comprehensive Survey of Dataset Distillationcited by 114
- 2021HPOBench: A Collection of Reproducible Multi-Fidelity Benchmark Problems for HPOcited by 111
- 2022Efficient Dataset Distillation Using Random Feature Approximationcited by 111
- 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
12 most cited of 339 All 339 (JSON)
- 2026Trust the Direction, Search the Step: Zero-and-First-Order Methods for LLM Fine-Tuning1 Oct 2026
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026ABO-Med: Accelerated Bilevel Optimization for Few-Shot Medical Image Classification27 Sep 2026
- 2026Intrinsic-Extrinsic Coupling in Learning Dynamics24 Sep 2026
- 2026TBR: Transport-Based Rendering with Deposition Strokes for Inverse Graphics8 Sep 2026
- 2026Disciplined Bilevel Programming1 Sep 2026
- 2026Stored in Optimizer State, Valued by Later Training: A Causal Account of Subliminal Trait Transfer20 Aug 2026
- 2026Sharper Analysis of Single-Loop Methods for Bilevel Optimization11 Jul 2026
- 2026Epistemic Goggles: A Pretrained Module that Induces an Epistemic Frame via Gradient Editing2 Jul 2026
- 2026Revocable Learned State via Process Sidecars29 Jun 2026
- 2026Bilevel Optimization for Neural Architecture Search28 Jun 2026
- 2026FastMix: Fast Data Mixture Optimization via Gradient Descent12 Jun 2026
12 newest of 339 All 339 (JSON)