Freya PAGE:面向大规模非凸有限和优化问题的首个最优时间复杂度方法,支持异构异步计算
最优化与控制
2024-11-05 v2 机器学习
机器学习
摘要
在实际分布式系统中,工作进程通常具有异构性,由于硬件配置和网络条件的差异,处理时间可能高度变化。我们考虑此场景下的光滑非凸有限和问题(即经验风险最小化),并引入一种新型并行方法——Freya PAGE,旨在处理任意异构和异步计算。该方法对“缓慢计算节点”具有鲁棒性,通过自适应忽略缓慢计算,Freya PAGE 的时间复杂度优于所有先前方法,包括 Asynchronous SGD、Rennala SGD、SPIDER 和 PAGE,同时要求的假设更弱。该算法依赖于具有理论保证的新型随机梯度收集策略,这些策略本身也可能为未来的优化方法设计提供 inspiration。此外,我们为光滑非凸有限和问题建立了下界,提供了该异步设置下的基本时间复杂度限制。该下界紧凑,表明在大规模 regime(即 ,其中 为工作进程数, 为数据样本数)下,Freya PAGE 是最优的。
引用
@article{arxiv.2405.15545,
title = {Freya PAGE: First Optimal Time Complexity for Large-Scale Nonconvex Finite-Sum Optimization with Heterogeneous Asynchronous Computations},
author = {Alexander Tyurin and Kaja Gruntkowska and Peter Richtárik},
journal= {arXiv preprint arXiv:2405.15545},
year = {2024}
}
备注
43 pages, 2 figures