返回
Variance-reduced reshuffling gradient descent for nonconvex optimization: Centralized and distributed algorithms
DOI:10.1016/j.automatica.2024.111954.png)
摘要
En 中文
Nonconvex finite-sum optimization plays a crucial role in signal processing and machine learning, fueling the development of numerous centralized and distributed stochastic algorithms. However, existing stochastic optimization algorithms often suffer from high stochastic gradient variance due to the use of random sampling with replacement. To address this issue, this paper introduces an explicit variance-reduction step and proposes variance-reduced reshuffling gradient algorithms with a sampling-without-replacement scheme. Specifically, this paper proves that the proposed centralized variance-reduced reshuffling gradient algorithm (VR-RG) with constant step sizes converges to a stationary point for nonconvex optimization under the Kurdyka-& Lstrok;ojasiewicz condition. Moreover, for nonconvex optimization over connected multi-agent networks, the proposed distributed variance- reduced reshuffling gradient algorithm (DVR-RG) converges to a neighborhood of stationary points, where the neighborhood can be made arbitrarily small under mild conditions. Notably, the proposed DVR-RG requires only one communication round at each epoch. Finally, numerical simulations demonstrate the efficiency of the proposed algorithms. (c) 2024 Elsevier Ltd. All rights are reserved, including those for text and data mining, AI training, and similar technologies.
Keyword:
Reshuffling gradient descent
Variance reduction
Nonconvex optimization
Sampling without replacement
Multi-agent networks
期刊
IF:
5.9
论文数:
1.2W
被引数:
5.2W
机构
引用论文
Convergence analysis of distributed stochastic gradient descent with shuffling带shuffling的分布随机梯度下降算法的收敛性分析
NEUROCOMPUTING
IF6.5

