Super-efficiency of automatic differentiation for functions defined as a minimum
Pierre Ablin, Gabriel Peyré, Thomas Moreau
Abstract
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 487cd1dc-55d3-40d2-9ff5-7dbb203e4f66Cited by top-tier papers21
- Efficient and Modular Implicit DifferentiationMathieu Blondel, Quentin Berthet, Marco Cuturi, Roy Frostig et al.NeurIPS 2022 · 386 citations
- Amortized Implicit Differentiation for Stochastic Bilevel OptimizationMichael Arbel, Julien MairalICLR 2022 · 78 citations
- Trajectory Inference via Mean-field Langevin in Path SpaceLénaïc Chizat, Stephen Zhang, Matthieu Heitz, Geoffrey SchiebingerNeurIPS 2022 · 45 citations
- Non-Convex Bilevel Games with Critical Point Selection MapsMichael Arbel, Julien MairalNeurIPS 2022 · 40 citations
- One-step differentiation of iterative algorithmsJérôme Bolte, Edouard Pauwels, Samuel VaiterNeurIPS 2023 · 36 citations
Related papers
- Stochastic Optimization for Regularized Wasserstein EstimatorsMarin Ballu, Quentin Berthet, Francis R. BachICML 2020 · 17 citations
- Two Losses Are Better Than One: Faster Optimization Using a Cheaper ProxyBlake E. Woodworth, Konstantin Mishchenko, Francis R. BachICML 2023 · 9 citations
- Sobolev Gradient Ascent for Optimal Transport: Barycenter Optimization and Convergence AnalysisKaheon Kim, Bohan Zhou, Changbo Zhu, Xiaohui ChenICLR 2026 · 6 citations
- Statistical, Robustness, and Computational Guarantees for Sliced Wasserstein DistancesSloan Nietert, Ziv Goldfeld, Ritwik Sadhu, Kengo KatoNeurIPS 2022 · 73 citations
- Optimal Time Complexities of Parallel Stochastic Optimization Methods Under a Fixed Computation ModelAlexander Tyurin, Peter RichtárikNeurIPS 2023 · 31 citations
