关于非凸-凹极小极大问题的梯度下降上升法
机器学习
2024-05-06 v9 最优化与控制
机器学习
摘要
我们考虑非凸-凹极小极大问题,,其中关于非凸但关于凹,且是凸且有界集合。求解该问题最流行的算法之一是为学界所熟知的梯度下降上升(GDA)算法,其已广泛应用于机器学习、控制理论与经济学。尽管在凸-凹设定下已有大量收敛结果,等步长GDA在一般设定下会收敛到极限环甚至发散。本文中,我们给出双时间尺度GDA求解非凸-凹极小极大问题的复杂度结果,表明该算法能高效找到函数的驻点。据我们所知,这是该设定下双时间尺度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