mlsanityData to 7 Oct 2026

Super-efficiency of automatic differentiation for functions defined as a minimum

arXiv 2002.03722 ↗First posted 10 Feb 2020stat.ML cs.LGcited by 24

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

Cited by year

Citing papers per year: 2021: 10; 2022: 5; 2023: 4; 2024: 5.

24 papers.

Sort:

By area:

  1. 2021Efficient and Modular Implicit Differentiationcited by 199
  2. 2021Gradients are Not All You Needcited by 82
  3. 2021Amortized Implicit Differentiation for Stochastic Bilevel Optimizationcited by 62
  4. 2022Trajectory Inference via Mean-field Langevin in Path Spacecited by 29
  5. 2022A Unified Framework for Implicit Sinkhorn Differentiationcited by 28
  6. 2023One-step differentiation of iterative algorithmscited by 28
  7. 2022Automatic differentiation of nonsmooth iterative algorithmscited by 25
  8. 2021Implicit differentiation for fast hyperparameter selection in non-smooth convex learningcited by 22
  9. 2024Functional Bilevel Optimization for Machine Learningcited by 20
  10. 2021Multiset-Equivariant Set Prediction with Approximate Implicit Differentiationcited by 17
  11. 2023A Lower Bound and a Near-Optimal Algorithm for Bilevel Empirical Risk Minimizationcited by 16
  12. 2021Stable and Interpretable Unrolled Dictionary Learningcited by 15
12 more
  1. 2022The Curse of Unrolling: Rate of Differentiating Through Optimizationcited by 15
  2. 2024Let Go of Your Labels with Unsupervised Transfercited by 14
  3. 2021Understanding approximate and unrolled dictionary learning for pattern recoverycited by 11
  4. 2021Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and Morecited by 10
  5. 2022Learning Energy Networks with Generalized Fenchel-Young Lossescited by 8
  6. 2021Debiasing a First-order Heuristic for Approximate Bi-level Optimizationcited by 3
  7. 2021Differentiable Equilibrium Computation with Decision Diagrams for Stackelberg Models of Combinatorial Congestion Gamescited by 2
  8. 2023Auto-Differentiation of Relational Computations for Very Large Scale Machine Learningcited by 1
  9. 2023Probabilistic Unrolling: Scalable, Inverse-Free Maximum Likelihood Estimation for Latent Gaussian Modelscited by 1
  10. 2024Enhancing Hypergradients Estimation: A Study of Preconditioning and Reparameterizationcited by 1
  11. 2024Automatic Differentiation of Optimization Algorithms with Time-Varying Updatescited by 1
  12. 2024Derivatives of Stochastic Gradient Descent in parametric optimizationcited by 0