arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2202.04640math.OCcs.DScs.LG
收藏

原始-对偶外梯度方法下可分离极大极小与有限和Optimization的更快速度

Sharper Rates for Separable Minimax and Finite Sum Optimization via Primal-Dual Extragradient Methods

Stanford University(斯坦福大学)

Yujia Jin, Aaron Sidford, Kevin Tian

摘要:

我们设计了几个基本类的优化问题的加速算法,提高了速度。我们的算法都建立在与通过[CST 21]最近提出的相对Lipschitzness分析原始-对偶超梯度方法相关的技术基础上。(1)可分离极小极大优化本文研究了可分极小极大优化问题$\min_x \max_y f(x) - g(y) + h(x, y)$,其中$f$和$g$具有光滑性和强凸性参数$(L^x, μ^x)$,$(L^y, μ^y)$,$h$是凸凹的,且具有$(Λ^{xx}, Λ^{xy}, Λ^{yy})$ -分块算子范数有界Hessian.我们提供了一个算法梯度查询复杂度$\tilde{O}\left(\sqrt{\frac{L^{x}}{μ^{x}}} + \sqrt{\frac{L^{y}}{μ^{y}}} + \frac{Λ^{xx}}{μ^{x}} + \frac{Λ^{xy}}{\sqrt{μ^{x}μ^{y}}} + \frac{Λ^{yy}}{μ^{y}}\right)$。值得注意的是,对于具有双线性耦合的凸-凹极大极小问题(例如,二次型),其中$Λ^{xx} = Λ^{yy} = 0$,我们的速率匹配[ZHZ 19]的下界。(2)有限和优化。我们研究有限和优化问题$\min_x \frac{1}{n}\sum_{i\in[n]} f_i(x)$,其中每个$f_i$是$L_i$ -光滑的,整个问题是$μ$ -强凸的。我们提供了一个算法梯度查询复杂度$\tilde{O}\left(n + \sum_{i\in[n]} \sqrt{\frac{L_i}{nμ}} \right)$。值得注意的是,当平滑边界$\{L_i\}_{i\in[n]}$不均匀时,我们的速率在加速SVRG [LMH 15,FGKS 15]和Katyusha [All 17]上提高了$\sqrt{n}$因子。(3)Minimax finite sums.我们推广我们的算法,极大极小和有限和优化,以解决自然家庭的极大极小有限和优化问题的加速速度,封装上述结果的对数因子。

英文摘要:

We design accelerated algorithms with improved rates for several fundamental classes of optimization problems. Our algorithms all build upon techniques related to the analysis of primal-dual extragradient methods via relative Lipschitzness proposed recently by [CST21]. (1) Separable minimax optimization. We study separable minimax optimization problems $\min_x \max_y f(x) - g(y) + h(x, y)$, where $f$ and $g$ have smoothness and strong convexity parameters $(L^x, μ^x)$, $(L^y, μ^y)$, and $h$ is convex-concave with a $(Λ^{xx}, Λ^{xy}, Λ^{yy})$-blockwise operator norm bounded Hessian. We provide an algorithm with gradient query complexity $\tilde{O}\left(\sqrt{\frac{L^{x}}{μ^{x}}} + \sqrt{\frac{L^{y}}{μ^{y}}} + \frac{Λ^{xx}}{μ^{x}} + \frac{Λ^{xy}}{\sqrt{μ^{x}μ^{y}}} + \frac{Λ^{yy}}{μ^{y}}\right)$. Notably, for convex-concave minimax problems with bilinear coupling (e.g.\ quadratics), where $Λ^{xx} = Λ^{yy} = 0$, our rate matches a lower bound of [ZHZ19]. (2) Finite sum optimization. We study finite sum optimization problems $\min_x \frac{1}{n}\sum_{i\in[n]} f_i(x)$, where each $f_i$ is $L_i$-smooth and the overall problem is $μ$-strongly convex. We provide an algorithm with gradient query complexity $\tilde{O}\left(n + \sum_{i\in[n]} \sqrt{\frac{L_i}{nμ}} \right)$. Notably, when the smoothness bounds $\{L_i\}_{i\in[n]}$ are non-uniform, our rate improves upon accelerated SVRG [LMH15, FGKS15] and Katyusha [All17] by up to a $\sqrt{n}$ factor. (3) Minimax finite sums. We generalize our algorithms for minimax and finite sum optimization to solve a natural family of minimax finite sum optimization problems at an accelerated rate, encapsulating both above results up to a logarithmic factor.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑