Double Variance Reduction: A Smoothing Trick for Composite Optimization Problems without First-Order Gradient
Hao Di, Haishan Ye, Yueling Zhang, Xiangyu Chang, Guang Dai, Ivor W. Tsang
Abstract
Variance reduction techniques are designed to decrease the sampling variance, thereby accelerating convergence rates of first-order (FO) and zeroth-order (ZO) optimization methods. However, in composite optimization problems, ZO methods encounter an additional variance called the coordinate-wise variance, which stems from the random gradient estimation. To reduce this variance, prior works require estimating all partial derivatives, essentially approximating FO information. This approach demands O(d) function evaluations (d is the dimension size), which incurs substantial computational costs and is prohibitive in high-dimensional scenarios. This paper proposes the Zeroth-order Proximal Double Variance Reduction (ZPDVR) method, which utilizes the averaging trick to reduce both sampling and coordinate-wise variances. Compared to prior methods, ZPDVR relies solely on random gradient estimates, calls the stochastic zeroth-order oracle (SZO) in expectation times per iteration, and achieves the optimal SZO query complexity in the strongly convex and smooth setting, where represents the condition number and is the desired accuracy. Empirical results validate ZPDVR's linear convergence and demonstrate its superior performance over other related methods.
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 59003553-76e7-4d5e-b119-dd281993a327Builds on4
- Fine-Tuning Language Models with Just Forward PassesSadhika Malladi, Tianyu Gao, Eshaan Nichani, Alex Damian et al.NeurIPS 2023 · 495 citations
- Adaptive Proximal Gradient Methods for Structured Neural NetworksJihun Yun, Aurélie C. Lozano, Eunho YangNeurIPS 2021 · 34 citations
- Faster Gradient-Free Algorithms for Nonsmooth Nonconvex Stochastic OptimizationLesi Chen, Jing Xu, Luo LuoICML 2023 · 26 citations
- Decentralized Gradient-Free Methods for Stochastic Non-smooth Non-convex OptimizationZhenwei Lin, Jingfan Xia, Qi Deng, Luo LuoAAAI 2024 · 11 citations
Related papers
- Optimal Algorithms for Stochastic Multi-Level Compositional OptimizationWei Jiang, Bokun Wang, Yibo Wang, Lijun Zhang et al.ICML 2022 · 25 citations
- Accelerated Cyclic Coordinate Dual Averaging with Extrapolation for Composite Convex OptimizationCheuk Yin Lin, Chaobing Song, Jelena DiakonikolasICML 2023 · 9 citations
- Private Zeroth-Order Nonsmooth Nonconvex OptimizationQinzi Zhang, Hoang Tran, Ashok CutkoskyICLR 2024 · 9 citations
- RandProx: Primal-Dual Optimization Algorithms with Randomized Proximal UpdatesLaurent Condat, Peter RichtárikICLR 2023 · 1 citation
- Efficient Continual Finite-Sum MinimizationIoannis Mavrothalassitis, Stratis Skoulakis, Leello Tadesse Dadi, Volkan CevherICLR 2024
