中文

关于非凸-凹极小极大问题的梯度下降上升法

机器学习 2024-05-06 v9 最优化与控制 机器学习

摘要

我们考虑非凸-凹极小极大问题,minxmaxyYf(x,y)\min_{\mathbf{x}} \max_{\mathbf{y} \in \mathcal{Y}} f(\mathbf{x}, \mathbf{y}),其中ff关于x\mathbf{x}非凸但关于y\mathbf{y}凹,且Y\mathcal{Y}是凸且有界集合。求解该问题最流行的算法之一是为学界所熟知的梯度下降上升(GDA)算法,其已广泛应用于机器学习、控制理论与经济学。尽管在凸-凹设定下已有大量收敛结果,等步长GDA在一般设定下会收敛到极限环甚至发散。本文中,我们给出双时间尺度GDA求解非凸-凹极小极大问题的复杂度结果,表明该算法能高效找到函数Φ():=maxyYf(,y)\Phi(\cdot) := \max_{\mathbf{y} \in \mathcal{Y}} f(\cdot, \mathbf{y})的驻点。据我们所知,这是该设定下双时间尺度GDA的首个非渐近分析,为其训练生成对抗网络(GANs)及其他实际应用中的优越表现提供了启示。

关键词

引用

@article{arxiv.1906.00331,
  title  = {On Gradient Descent Ascent for Nonconvex-Concave Minimax Problems},
  author = {Tianyi Lin and Chi Jin and Michael I. Jordan},
  journal= {arXiv preprint arXiv:1906.00331},
  year   = {2024}
}

备注

Accepted by ICML 2020; 39 pages, 6 figures