Two Losses Are Better Than One: Faster Optimization Using a Cheaper Proxy
Blake E. Woodworth, Konstantin Mishchenko, Francis R. Bach
Abstract
We present an algorithm for minimizing an objective with hard-to-compute gradients by using a related, easier-to-access function as a proxy. Our algorithm is based on approximate proximal point iterations on the proxy combined with relatively few stochastic gradients from the objective. When the difference between the objective and the proxy is -smooth, our algorithm guarantees convergence at a rate matching stochastic gradient descent on a -smooth objective, which can lead to substantially better sample efficiency. Our algorithm has many potential applications in machine learning, and provides a principled means of leveraging synthetic data, physics simulators, mixed public and private data, and more.
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 a4c696a0-8483-4109-846a-c5facd02cbb0Cited by top-tier papers2
- Non-convex Stochastic Composite Optimization with Polyak MomentumYuan Gao, Anton Rodomanov, Sebastian U. StichICML 2024 · 13 citations
- Accelerated Methods with Compressed Communications for Distributed Optimization Problems Under Data SimilarityDmitry Bylinkin, Aleksandr BeznosikovAAAI 2025 · 3 citations
Builds on4
- Rethinking ImageNet Pre-TrainingKaiming He, Ross B. Girshick, Piotr DollárICCV 2019 · 1,188 citations
- Mind the Gap: Assessing Temporal Generalization in Neural Language ModelsAngeliki Lazaridou, Adhiguna Kuncoro, Elena Gribovskaya, Devang Agrawal et al.NeurIPS 2021 · 315 citations
- Statistically Preconditioned Accelerated Gradient Method for Distributed OptimizationHadrien Hendrikx, Lin Xiao, Sébastien Bubeck, Francis R. Bach et al.ICML 2020 · 66 citations
- Fast Stochastic Bregman Gradient Methods: Sharp Analysis and Variance ReductionRadu-Alexandru Dragomir, Mathieu Even, Hadrien HendrikxICML 2021 · 40 citations
Related papers
- Target-based Surrogates for Stochastic OptimizationJonathan Wilder Lavington, Sharan Vaswani, Reza Babanezhad Harikandeh, Mark Schmidt et al.ICML 2023 · 6 citations
- Super-efficiency of automatic differentiation for functions defined as a minimumPierre Ablin, Gabriel Peyré, Thomas MoreauICML 2020 · 42 citations
- Stochastic Approximate Gradient Descent via the Langevin AlgorithmYixuan Qiu, Xiao WangAAAI 2020 · 5 citations
- Lower Complexity Bounds for Finite-Sum Convex-Concave Minimax Optimization ProblemsGuangzeng Xie, Luo Luo, Yijiang Lian, Zhihua ZhangICML 2020 · 21 citations
- Descent with Misaligned Gradients and Applications to Hidden ConvexityAditya Bhaskara, Ashok Cutkosky, Ravi Kumar, Manish PurohitICLR 2025
