A Zeroth-Order Variance-Reduced Method for Decentralized Stochastic Non-convex Optimization
Abstract
In this paper, we consider a distributed stochastic non-convex optimization problem, which is about minimizing a sum of local cost functions over a network with only zeroth-order information. A novel single-loop Decentralized Zeroth-Order Variance Reduction algorithm, called DZOVR, is proposed, which combines two-point gradient estimation, momentum-based variance reduction technique, and gradient tracking. Under mild assumptions, we show that the algorithm is able to achieve sampling complexity at each node to reach an -accurate stationary point and also exhibits network-independent and linear speedup properties. To the best of our knowledge, this is the first stochastic decentralized zeroth-order algorithm that achieves this sampling complexity. Numerical experiments demonstrate that DZOVR outperforms the other state-of-the-art algorithms and has network-independent and linear speedup properties.
Cite
@article{arxiv.2310.18883,
title = {A Zeroth-Order Variance-Reduced Method for Decentralized Stochastic Non-convex Optimization},
author = {Hongxu Chen and Jinchi Chen and Ke Wei},
journal= {arXiv preprint arXiv:2310.18883},
year = {2023}
}