中文

基于原始-对偶外梯度方法的可分离极小极大与有限和优化更优收敛速率

最优化与控制 2022-02-10 v1 数据结构与算法 机器学习

摘要

我们为几类基本优化问题设计了具有改进速率的加速算法。这些算法均基于 [CST21] 最近提出的通过相对 Lipschitz 性分析原始-对偶外梯度方法的相关技术。(1) 可分离极小极大优化。我们研究可分离极小极大优化问题 minxmaxyf(x)g(y)+h(x,y)\min_x \max_y f(x) - g(y) + h(x, y),其中 ffgg 具有光滑性和强凸性参数 (Lx,μx)(L^x, \mu^x)(Ly,μy)(L^y, \mu^y),且 hh 是凸-凹的,其 Hessian 矩阵具有 (Λxx,Λxy,Λyy)(\Lambda^{xx}, \Lambda^{xy}, \Lambda^{yy})-分块算子范数有界。我们提供了一个梯度查询复杂度为 O~(Lxμx+Lyμy+Λxxμx+Λxyμxμy+Λyyμy)\tilde{O}\left(\sqrt{\frac{L^{x}}{\mu^{x}}} + \sqrt{\frac{L^{y}}{\mu^{y}}} + \frac{\Lambda^{xx}}{\mu^{x}} + \frac{\Lambda^{xy}}{\sqrt{\mu^{x}\mu^{y}}} + \frac{\Lambda^{yy}}{\mu^{y}}\right) 的算法。值得注意的是,对于具有双线性耦合的凸-凹极小极大问题(例如二次型),其中 Λxx=Λyy=0\Lambda^{xx} = \Lambda^{yy} = 0,我们的速率匹配了 [ZHZ19] 的下界。(2) 联合极小优化。我们研究联合极小优化问题 minx1ni[n]fi(x)\min_x \frac{1}{n}\sum_{i\in[n]} f_i(x),其中每个 fif_iLiL_i-光滑的,且整体问题是 μ\mu-强凸的。我们提供了一个梯度查询复杂度为 O~(n+i[n]Linμ)\tilde{O}\left(n + \sum_{i\in[n]} \sqrt{\frac{L_i}{n\mu}} \right) 的算法。值得注意的是,当光滑性界 {Li}i[n]\{L_i\}_{i\in[n]} 非均匀时,我们的速率比加速 SVRG [LMH15, FGKS15] 和 Katyusha [All17] 提高了最多 n\sqrt{n} 倍。(3) 极小极大联合极小化。我们将用于极小极大和联合极小优化的算法推广,以加速速率解决一类自然的极小极大联合极小优化问题,将上述两个结果囊括在内,仅相差一个对数因子。

关键词

引用

@article{arxiv.2202.04640,
  title  = {Sharper Rates for Separable Minimax and Finite Sum Optimization via Primal-Dual Extragradient Methods},
  author = {Yujia Jin and Aaron Sidford and Kevin Tian},
  journal= {arXiv preprint arXiv:2202.04640},
  year   = {2022}
}