Optimizing Millions of Hyperparameters by Implicit Differentiation
Not on the map. Why?
Abstract (arXiv)
We propose an algorithm for inexpensive gradient-based hyperparameter optimization that combines the implicit function theorem (IFT) with efficient inverse Hessian approximations. We present results about the relationship between the IFT and differentiating through optimization, motivating our algorithm. We use the proposed approach to train modern network architectures with millions of weights and millions of hyper-parameters. For example, we learn a data-augmentation network - where every weight is a hyperparameter tuned for validation performance - outputting augmented training examples. Jointly tuning weights and hyperparameters with our approach is only a few times more costly in memory and compute than standard training.
Key facts. Cited by 304 papers in the map (data through 14 Oct 2026). First posted on arXiv 6 Nov 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 53; 2022: 70; 2023: 53; 2024: 60; 2025: 38; 2026: 30.
304 papers.
Sort:
By area:
- 2021Efficient and Modular Implicit Differentiationcited by 203
- 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 153
- 2021Hyperparameter Optimization: Foundations, Algorithms, Best Practices and Open Challengescited by 145
- 2023Dataset Distillation: A Comprehensive Reviewcited by 127
- 2022Efficient Dataset Distillation Using Random Feature Approximationcited by 111
- 2022BOME! Bilevel Optimization Made Easy: A Simple First-Order Approachcited by 105
- 2024EraseAnything: Enabling Concept Erasure in Rectified Flow Transformerscited by 95
- 2022Wild Patterns Reloaded: A Survey of Machine Learning Security against Training Data Poisoningcited by 94
- 2023DREAM: Efficient Dataset Distillation by Representative Matchingcited by 94
- 2021Proximal Optimal Transport Modeling of Population Dynamicscited by 93
- 2023Data Distillation: A Surveycited by 87
38 more
- 2021JFB: Jacobian-Free Backpropagation for Implicit Networkscited by 86
- 2024Better than classical? The subtle art of benchmarking quantum machine learning modelscited by 85
- 2022Tutorial on amortized optimizationcited by 83
- 2021Is Attention Better Than Matrix Decomposition?cited by 77
- 2021On Training Implicit Modelscited by 75
- 2021Label-Imbalanced and Group-Sensitive Classification under Overparameterizationcited by 74
- 2023Using Large Language Models for Hyperparameter Optimizationcited by 71
- 2022Theseus: A Library for Differentiable Nonlinear Optimizationcited by 65
- 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
- 2022Improving Object-centric Learning with Query Optimizationcited by 60
- 2021Unbiased Gradient Estimation in Unrolled Computation Graphs with Persistent Evolution Strategiescited by 52
- 2021iDARTS: Differentiable Architecture Search with Stochastic Implicit Gradientscited by 51
- 2021Meta-learning PINN loss functionscited by 51
- 2022Model-Based Deep Learning: On the Intersection of Deep Learning and Optimizationcited by 50
- 2022Self-Guided Noise-Free Data Generation for Efficient Zero-Shot Learningcited by 50
- 2022AutoBalance: Optimized Loss Functions for Imbalanced Datacited by 49
- 2022Exploring Example Influence in Continual Learningcited by 44
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learningcited by 44
- 2022Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximationscited by 42
- 2022Relaxing Equivariance Constraints with Non-stationary Continuous Filterscited by 41
- 2021Deep Proxy Causal Learning and its Application to Confounded Bandit Policy Evaluationcited by 40
- 2021HYDRA: Hypergradient Data Relevance Analysis for Interpreting Deep Neural Networkscited by 38
- 2024ScaleBiO: Scalable Bilevel Optimization for LLM Data Reweightingcited by 37
- 2021Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiationcited by 35
- 2021On the Convergence Theory for Hessian-Free Bilevel Algorithmscited by 35
- 2023AutoML in the Age of Large Language Models: Current Challenges, Future Opportunities and Riskscited by 35
- 2021Stability and Generalization of Bilevel Programming in Hyperparameter Optimizationcited by 31
- 2022On Implicit Bias in Overparameterized Bilevel Optimizationcited by 30
- 2021Meta-Calibration: Learning of Model Calibration Using Differentiable Expected Calibration Errorcited by 29
- 2022Learning Model Predictive Controllers with Real-Time Attention for Real-World Navigationcited by 29
- 2025Think Smarter not Harder: Adaptive Reasoning with Inference Aware Optimizationcited by 29
- 2021A General Descent Aggregation Framework for Gradient-based Bi-level Optimizationcited by 28
- 2023One-step differentiation of iterative algorithmscited by 28
- 2021Meta-Learning to Improve Pre-Trainingcited by 27
- 2022Betty: An Automatic Differentiation Library for Multilevel Optimizationcited by 27
50 most cited of 304 All 304 (JSON)
- 2026Probabilistic Multi-Source Data Selection via Learnable Categorical Priors: Variance Reduction and Guaranteed Derandomization10 Oct 2026
- 2026BRIDGE: Bilevel Retrieval-Credit-Aware Agentic Reinforcement Learning29 Sep 2026
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026A Light Bilevel Refinement Aligns Self-Supervised Representations for Stronger Task-Specific Learning27 Sep 2026
- 2026ABO-Med: Accelerated Bilevel Optimization for Few-Shot Medical Image Classification27 Sep 2026
- 2026Bilevel Optimization of Topology and Hyperparameters (BOTH)18 Sep 2026
- 2026Disciplined Bilevel Programming1 Sep 2026
- 2026Understanding Differentiable Embeddings Through Differential and Integral Geometry7 Aug 2026
- 2026Hypergradient-based Bilevel Reinforcement Learning with Improved Sample Complexity30 Jul 2026
- 2026Revocable Learned State via Process Sidecars29 Jun 2026
- 2026Non-Linear Strategic Classification Made Practical26 Jun 2026
- 2026Escaping the Variance Trap: Jacobian-Free Dynamics for Root-Finding Bilevel Optimization21 Jun 2026
38 more
- 2026Diff2SP: Diffusion Models for Correlated Scenario Generation in Stochastic Programming4 Jun 2026
- 2026Two to Tango: Coupled Task-Reference Selection for Safe LLM Fine-tuning1 Jun 2026
- 2026On the Difficulty of Learning a Meta-network for Training Data Selection30 May 2026
- 2026Density-aware Sample-specific Attack27 May 2026
- 2026Semiparametric Efficient Bilevel Gradient Estimation20 May 2026
- 2026CHAL: Council of Hierarchical Agentic Language12 May 2026
- 2026When Losses Align: Gradient-Based Composite Loss Weighting for Efficient Pretraining8 May 2026
- 2026Safe Bilevel Delegation (SBD): A Formal Framework for Runtime Delegation Safety in Multi-Agent Systems30 Apr 2026
- 2026Diff3R: Feed-forward 3D Gaussian Splatting with Uncertainty-aware Differentiable Optimization1 Apr 2026
- 2026Efficient Bilevel Optimization with KFAC-Based Hypergradients31 Mar 2026
- 2026Why AI systems don't learn and what to do about it: Lessons on autonomous learning from cognitive science16 Mar 2026
- 2026Feed m Birds with One Scone: Accelerating Multi-task Gradient Balancing via Bi-level Optimization8 Mar 2026
- 2026SkillRater: Untangling Capabilities in Multimodal Data12 Feb 2026
- 2026Natural Hypergradient Descent: Algorithm Design, Convergence Analysis, and Parallel Implementation11 Feb 2026
- 2026Scaling GraphLLM with Bilevel-Optimized Sparse Querying30 Jan 2026
- 2026TBDFiltering: Sample-Efficient Tree-Based Data Filtering29 Jan 2026
- 2026Non-Stationary Functional Bilevel Optimization21 Jan 2026
- 2026Motion Attribution for Video Generation13 Jan 2026
- 2025Can Small Training Runs Reliably Guide Data Curation? Rethinking Proxy-Model Practice30 Dec 2025
- 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Duration26 Dec 2025
- 2025Gradient Descent with Provably Tuned Learning-rate Schedules4 Dec 2025
- 2025A Fully First-Order Layer for Differentiable Optimization2 Dec 2025
- 2025Modality Alignment across Trees on Heterogeneous Hyperbolic Manifolds31 Oct 2025
- 2025Scalable Principal-Agent Contract Design via Gradient-Based Optimization24 Oct 2025
- 2025Computational Budget Should Be Considered in Data Selection19 Oct 2025
- 2025Bilevel optimization for learning hyperparameters: Application to solving PDEs and inverse problems with Gaussian processes7 Oct 2025
- 2025Taught Well Learned Ill: Towards Distillation-conditional Backdoor Attack28 Sep 2025
- 2025Information-Theoretic Bayesian Optimization for Bilevel Optimization Problems26 Sep 2025
- 2025Reversible Deep Equilibrium Models16 Sep 2025
- 2025Curvature Learning for Generalization of Hyperbolic Neural Networks24 Aug 2025
- 2025Fusion Sampling Validation in Data Partitioning for Machine Learning2 Aug 2025
- 2025Optimizing Canaries for Privacy Auditing with Metagradient Descent21 Jul 2025
- 2025Better Training Data Attribution via Better Inverse Hessian-Vector Products19 Jul 2025
- 2025Robust Molecular Property Prediction via Densifying Scarce Labeled Data13 Jun 2025
- 2025Rethinking Data Mixture for Large Language Models: A Comprehensive Survey and New Perspectives27 May 2025
- 2025DataRater: Meta-Learned Dataset Curation23 May 2025
- 2025POCAII: Parameter Optimization with Conscious Allocation using Iterative Intelligence16 May 2025
- 2025Bi-Level Policy Optimization with Nystr"om Hypergradients16 May 2025
50 newest of 304 All 304 (JSON)
- 2024EraseAnything: Enabling Concept Erasure in Rectified Flow Transformerscited by 95
- 2024One Category One Prompt: Dataset Distillation using Diffusion Modelscited by 6
- 2025Score Distillation Sampling for Audio: Source Separation, Synthesis, and Beyondcited by 6
- 2025You Only Look One Step: Accelerating Backpropagation in Diffusion Sampling with Gradient Shortcutscited by 0
- 2026Diff2SP: Diffusion Models for Correlated Scenario Generation in Stochastic Programmingcited by 0
- 2026Diff2SP: Diffusion Models for Correlated Scenario Generation in Stochastic Programming4 Jun 2026
- 2025You Only Look One Step: Accelerating Backpropagation in Diffusion Sampling with Gradient Shortcuts12 May 2025
- 2025Score Distillation Sampling for Audio: Source Separation, Synthesis, and Beyond7 May 2025
- 2024EraseAnything: Enabling Concept Erasure in Rectified Flow Transformers29 Dec 2024
- 2024One Category One Prompt: Dataset Distillation using Diffusion Models11 Mar 2024
- 2025Think Smarter not Harder: Adaptive Reasoning with Inference Aware Optimizationcited by 29
- 2025Can Small Training Runs Reliably Guide Data Curation? Rethinking Proxy-Model Practicecited by 3
- 2025Rethinking Data Mixture for Large Language Models: A Comprehensive Survey and New Perspectivescited by 2
- 2026CHAL: Council of Hierarchical Agentic Languagecited by 0
- 2026CHAL: Council of Hierarchical Agentic Language12 May 2026
- 2025Can Small Training Runs Reliably Guide Data Curation? Rethinking Proxy-Model Practice30 Dec 2025
- 2025Rethinking Data Mixture for Large Language Models: A Comprehensive Survey and New Perspectives27 May 2025
- 2025Think Smarter not Harder: Adaptive Reasoning with Inference Aware Optimization29 Jan 2025
- 2026Density-aware Sample-specific Attack27 May 2026
- 2025Optimizing Canaries for Privacy Auditing with Metagradient Descent21 Jul 2025
- 2025Data Poisoning in Deep Learning: A Survey27 Mar 2025
- 2022Wild Patterns Reloaded: A Survey of Machine Learning Security against Training Data Poisoning4 May 2022
- 2025Modality Alignment across Trees on Heterogeneous Hyperbolic Manifolds31 Oct 2025
- 2024Let Go of Your Labels with Unsupervised Transfer11 Jun 2024
- 2021Efficient and Modular Implicit Differentiationcited by 203
- 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 153
- 2021Hyperparameter Optimization: Foundations, Algorithms, Best Practices and Open Challengescited by 145
- 2023Dataset Distillation: A Comprehensive Reviewcited by 127
- 2022Efficient Dataset Distillation Using Random Feature Approximationcited by 111
- 2022BOME! Bilevel Optimization Made Easy: A Simple First-Order Approachcited by 105
- 2023DREAM: Efficient Dataset Distillation by Representative Matchingcited by 94
- 2021Proximal Optimal Transport Modeling of Population Dynamicscited by 93
- 2023Data Distillation: A Surveycited by 87
- 2021JFB: Jacobian-Free Backpropagation for Implicit Networkscited by 86
- 2024Better than classical? The subtle art of benchmarking quantum machine learning modelscited by 85
12 most cited of 282 All 282 (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
- 2026A Light Bilevel Refinement Aligns Self-Supervised Representations for Stronger Task-Specific Learning27 Sep 2026
- 2026ABO-Med: Accelerated Bilevel Optimization for Few-Shot Medical Image Classification27 Sep 2026
- 2026Bilevel Optimization of Topology and Hyperparameters (BOTH)18 Sep 2026
- 2026Disciplined Bilevel Programming1 Sep 2026
- 2026Understanding Differentiable Embeddings Through Differential and Integral Geometry7 Aug 2026
- 2026Revocable Learned State via Process Sidecars29 Jun 2026
- 2026Non-Linear Strategic Classification Made Practical26 Jun 2026
- 2026On the Difficulty of Learning a Meta-network for Training Data Selection30 May 2026
- 2026Semiparametric Efficient Bilevel Gradient Estimation20 May 2026
- 2026When Losses Align: Gradient-Based Composite Loss Weighting for Efficient Pretraining8 May 2026
12 newest of 282 All 282 (JSON)