Super-efficiency of automatic differentiation for functions defined as a minimum
Not on the map. Why?
Abstract (arXiv)
In min-min optimization or max-min optimization, one has to compute the gradient of a function defined as a minimum. In most cases, the minimum has no closed-form, and an approximation is obtained via an iterative algorithm. There are two usual ways of estimating the gradient of the function: using either an analytic formula obtained by assuming exactness of the approximation, or automatic differentiation through the algorithm. In this paper, we study the asymptotic error made by these estimators as a function of the optimization error. We find that the error of the automatic estimator is close to the square of the error of the analytic estimator, reflecting a super-efficiency phenomenon. The convergence of the automatic estimator greatly depends on the convergence of the Jacobian of the algorithm. We analyze it for gradient descent and stochastic gradient descent and derive convergence rates for the estimators in these cases. Our analysis is backed by numerical experiments on toy problems and on Wasserstein barycenter computation. Finally, we discuss the computational complexity of these estimators and give practical guidelines to chose between them.
Key facts. Cited by 24 papers in the map (data through 7 Oct 2026). First posted on arXiv 10 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 10; 2022: 5; 2023: 4; 2024: 5.
24 papers.
Sort:
By area:
- 2021Efficient and Modular Implicit Differentiationcited by 199
- 2021Gradients are Not All You Needcited by 82
- 2021Amortized Implicit Differentiation for Stochastic Bilevel Optimizationcited by 62
- 2022Trajectory Inference via Mean-field Langevin in Path Spacecited by 29
- 2022A Unified Framework for Implicit Sinkhorn Differentiationcited by 28
- 2023One-step differentiation of iterative algorithmscited by 28
- 2022Automatic differentiation of nonsmooth iterative algorithmscited by 25
- 2021Implicit differentiation for fast hyperparameter selection in non-smooth convex learningcited by 22
- 2024Functional Bilevel Optimization for Machine Learningcited by 20
- 2021Multiset-Equivariant Set Prediction with Approximate Implicit Differentiationcited by 17
- 2023A Lower Bound and a Near-Optimal Algorithm for Bilevel Empirical Risk Minimizationcited by 16
- 2021Stable and Interpretable Unrolled Dictionary Learningcited by 15
12 more
- 2022The Curse of Unrolling: Rate of Differentiating Through Optimizationcited by 15
- 2024Let Go of Your Labels with Unsupervised Transfercited by 14
- 2021Understanding approximate and unrolled dictionary learning for pattern recoverycited by 11
- 2021Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and Morecited by 10
- 2022Learning Energy Networks with Generalized Fenchel-Young Lossescited by 8
- 2021Debiasing a First-order Heuristic for Approximate Bi-level Optimizationcited by 3
- 2021Differentiable Equilibrium Computation with Decision Diagrams for Stackelberg Models of Combinatorial Congestion Gamescited by 2
- 2023Auto-Differentiation of Relational Computations for Very Large Scale Machine Learningcited by 1
- 2023Probabilistic Unrolling: Scalable, Inverse-Free Maximum Likelihood Estimation for Latent Gaussian Modelscited by 1
- 2024Enhancing Hypergradients Estimation: A Study of Preconditioning and Reparameterizationcited by 1
- 2024Automatic Differentiation of Optimization Algorithms with Time-Varying Updatescited by 1
- 2024Derivatives of Stochastic Gradient Descent in parametric optimizationcited by 0
- 2024Automatic Differentiation of Optimization Algorithms with Time-Varying Updates21 Oct 2024
- 2024Let Go of Your Labels with Unsupervised Transfer11 Jun 2024
- 2024Derivatives of Stochastic Gradient Descent in parametric optimization24 May 2024
- 2024Functional Bilevel Optimization for Machine Learning29 Mar 2024
- 2024Enhancing Hypergradients Estimation: A Study of Preconditioning and Reparameterization26 Feb 2024
- 2023Probabilistic Unrolling: Scalable, Inverse-Free Maximum Likelihood Estimation for Latent Gaussian Models5 Jun 2023
- 2023Auto-Differentiation of Relational Computations for Very Large Scale Machine Learning31 May 2023
- 2023One-step differentiation of iterative algorithms23 May 2023
- 2023A Lower Bound and a Near-Optimal Algorithm for Bilevel Empirical Risk Minimization17 Feb 2023
- 2022The Curse of Unrolling: Rate of Differentiating Through Optimization27 Sep 2022
- 2022Automatic differentiation of nonsmooth iterative algorithms31 May 2022
- 2022Learning Energy Networks with Generalized Fenchel-Young Losses19 May 2022
12 more
- 2022Trajectory Inference via Mean-field Langevin in Path Space14 May 2022
- 2022A Unified Framework for Implicit Sinkhorn Differentiation13 May 2022
- 2021Amortized Implicit Differentiation for Stochastic Bilevel Optimization29 Nov 2021
- 2021Multiset-Equivariant Set Prediction with Approximate Implicit Differentiation23 Nov 2021
- 2021Gradients are Not All You Need10 Nov 2021
- 2021Differentiable Equilibrium Computation with Decision Diagrams for Stackelberg Models of Combinatorial Congestion Games5 Oct 2021
- 2021Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and More14 Jul 2021
- 2021Understanding approximate and unrolled dictionary learning for pattern recovery11 Jun 2021
- 2021Debiasing a First-order Heuristic for Approximate Bi-level Optimization4 Jun 2021
- 2021Stable and Interpretable Unrolled Dictionary Learning31 May 2021
- 2021Efficient and Modular Implicit Differentiation31 May 2021
- 2021Implicit differentiation for fast hyperparameter selection in non-smooth convex learning4 May 2021
- 2024Let Go of Your Labels with Unsupervised Transfercited by 14
- 2024Let Go of Your Labels with Unsupervised Transfer11 Jun 2024
- 2021Efficient and Modular Implicit Differentiationcited by 199
- 2021Gradients are Not All You Needcited by 82
- 2021Amortized Implicit Differentiation for Stochastic Bilevel Optimizationcited by 62
- 2022Trajectory Inference via Mean-field Langevin in Path Spacecited by 29
- 2022A Unified Framework for Implicit Sinkhorn Differentiationcited by 28
- 2023One-step differentiation of iterative algorithmscited by 28
- 2022Automatic differentiation of nonsmooth iterative algorithmscited by 25
- 2021Implicit differentiation for fast hyperparameter selection in non-smooth convex learningcited by 22
- 2024Functional Bilevel Optimization for Machine Learningcited by 20
- 2021Multiset-Equivariant Set Prediction with Approximate Implicit Differentiationcited by 17
- 2023A Lower Bound and a Near-Optimal Algorithm for Bilevel Empirical Risk Minimizationcited by 16
- 2021Stable and Interpretable Unrolled Dictionary Learningcited by 15
12 most cited of 23 All 23 (JSON)
- 2024Automatic Differentiation of Optimization Algorithms with Time-Varying Updates21 Oct 2024
- 2024Derivatives of Stochastic Gradient Descent in parametric optimization24 May 2024
- 2024Functional Bilevel Optimization for Machine Learning29 Mar 2024
- 2024Enhancing Hypergradients Estimation: A Study of Preconditioning and Reparameterization26 Feb 2024
- 2023Probabilistic Unrolling: Scalable, Inverse-Free Maximum Likelihood Estimation for Latent Gaussian Models5 Jun 2023
- 2023Auto-Differentiation of Relational Computations for Very Large Scale Machine Learning31 May 2023
- 2023One-step differentiation of iterative algorithms23 May 2023
- 2023A Lower Bound and a Near-Optimal Algorithm for Bilevel Empirical Risk Minimization17 Feb 2023
- 2022The Curse of Unrolling: Rate of Differentiating Through Optimization27 Sep 2022
- 2022Automatic differentiation of nonsmooth iterative algorithms31 May 2022
- 2022Learning Energy Networks with Generalized Fenchel-Young Losses19 May 2022
- 2022Trajectory Inference via Mean-field Langevin in Path Space14 May 2022
12 newest of 23 All 23 (JSON)