中文

关于Polyak-Łojasiewicz条件下有限和光滑优化的复杂度

最优化与控制 2024-02-06 v1 机器学习

摘要

本文考虑形如minxRdf(x)1ni=1nfi(x)\min_{{\bf x}\in{\mathbb R}^d} f({\bf x})\triangleq \frac{1}{n}\sum_{i=1}^n f_i({\bf x})的优化问题,其中f()f(\cdot)满足参数为μ\mu的Polyak--Łojasiewicz (PL)条件,且{fi()}i=1n\{f_i(\cdot)\}_{i=1}^nLL-均方光滑的。我们证明,任何梯度方法至少需要Ω(n+κnlog(1/ϵ))\Omega(n+\kappa\sqrt{n}\log(1/\epsilon))次增量一阶预言机(IFO)调用才能找到ϵ\epsilon-次优解,其中κL/μ\kappa\triangleq L/\mu是问题的条件数。这一结果几乎匹配了最知名一阶方法的IFO复杂度上界。我们还研究了分布式设置中最小化PL函数的问题,其中个体f1(),,fn()f_1(\cdot),\dots,f_n(\cdot)位于由nn个智能体组成的连通网络上。我们给出了通信轮次、时间成本和局部一阶预言机调用的下界,分别为Ω(κ/γlog(1/ϵ))\Omega(\kappa/\sqrt{\gamma}\,\log(1/\epsilon))Ω((κ+τκ/γ)log(1/ϵ))\Omega((\kappa+\tau\kappa/\sqrt{\gamma}\,)\log(1/\epsilon))Ω(n+κnlog(1/ϵ))\Omega\big(n+\kappa\sqrt{n}\log(1/\epsilon)\big),其中γ(0,1]\gamma\in(0,1]是与网络相关的混合矩阵的谱间隙,τ>0\tau>0是每轮通信的时间成本。此外,我们提出了一种分散式一阶方法,该方法在期望上几乎匹配上述下界。

关键词

引用

@article{arxiv.2402.02569,
  title  = {On the Complexity of Finite-Sum Smooth Optimization under the Polyak-{\L}ojasiewicz Condition},
  author = {Yunyan Bai and Yuxing Liu and Luo Luo},
  journal= {arXiv preprint arXiv:2402.02569},
  year   = {2024}
}