Super-efficiency of automatic differentiation for functions defined as a minimum
Pierre Ablin, Gabriel Peyré, Thomas Moreau
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper21
- Efficient and Modular Implicit DifferentiationMathieu Blondel, Quentin Berthet, Marco Cuturi, Roy Frostig 等NeurIPS 2022 · 被引用 386 次
- Amortized Implicit Differentiation for Stochastic Bilevel OptimizationMichael Arbel, Julien MairalICLR 2022 · 被引用 78 次
- Trajectory Inference via Mean-field Langevin in Path SpaceLénaïc Chizat, Stephen Zhang, Matthieu Heitz, Geoffrey SchiebingerNeurIPS 2022 · 被引用 45 次
- Non-Convex Bilevel Games with Critical Point Selection MapsMichael Arbel, Julien MairalNeurIPS 2022 · 被引用 40 次
- One-step differentiation of iterative algorithmsJérôme Bolte, Edouard Pauwels, Samuel VaiterNeurIPS 2023 · 被引用 36 次
相关 Paper
- Stochastic Optimization for Regularized Wasserstein EstimatorsMarin Ballu, Quentin Berthet, Francis R. BachICML 2020 · 被引用 17 次
- Two Losses Are Better Than One: Faster Optimization Using a Cheaper ProxyBlake E. Woodworth, Konstantin Mishchenko, Francis R. BachICML 2023 · 被引用 9 次
- Sobolev Gradient Ascent for Optimal Transport: Barycenter Optimization and Convergence AnalysisKaheon Kim, Bohan Zhou, Changbo Zhu, Xiaohui ChenICLR 2026 · 被引用 6 次
- Statistical, Robustness, and Computational Guarantees for Sliced Wasserstein DistancesSloan Nietert, Ziv Goldfeld, Ritwik Sadhu, Kengo KatoNeurIPS 2022 · 被引用 73 次
- Optimal Time Complexities of Parallel Stochastic Optimization Methods Under a Fixed Computation ModelAlexander Tyurin, Peter RichtárikNeurIPS 2023 · 被引用 31 次
