Distributed Stochastic Optimization under Heavy-Tailed Noise: A Federated Mirror Descent Approach with High Probability Convergence

Abstract

We study the distributed stochastic optimization (DSO) problem under a heavy-tailed noise condition by utilizing a multi-agent system. Despite the extensive research on DSO algorithms used to solve DSO problems under light-tailed noise conditions (such as Gaussian noise), there is a significant lack of study of DSO algorithms in the context of heavy-tailed random noise. Classical DSO approaches in a heavy-tailed setting may present poor convergence behaviors. Therefore, developing DSO methods in the context of heavy-tailed noises is of importance. This work follows this path and we consider the setting that the gradient noises associated with each agent can be heavy-tailed, potentially having unbounded variance. We propose a clipped federated stochastic mirror descent algorithm to solve the DSO problem. We rigorously present a convergence theory and show that, under appropriate rules on the stepsize and the clipping parameter associated with the local noisy gradient influenced by the heavy-tailed noise, the algorithm is able to achieve satisfactory high probability convergence.