Convergence Analysis of Continuous-Time Distributed Stochastic Gradient Algorithms
Jianhua Sun, Kaihong Lu, Xin Yu
TL;DR
This work introduces a continuous-time distributed stochastic gradient framework where multiple agents cooperatively minimize a sum of convex functions using consensus dynamics and stochastic gradient feedback modeled by Brownian noise. The convergence analysis leverages Itô calculus, Lyapunov methods, and switching balanced directed graphs to show that agent states converge in expectation to a common minimizer, with explicit rates governed by the step-size decay exponent $a$; the fastest polynomial rate is $\mathcal{O}(t^{-1/4}\sqrt{\ln t})$ at $a=3/4$. Theoretical results are supported by simulations on a six-agent network with a time-varying topology, illustrating convergence of both the agent states and objective value to the optimum. These findings advance the understanding of continuous-time distributed stochastic optimization under realistic communication constraints and stochastic gradient information. Potential future directions include non-convex objectives, constraints, and high-probability convergence analyses.
Abstract
In this paper, we propose a new framework to study distributed optimization problems with stochastic gradients by employing a multi-agent system with continuous-time dynamics. Here the goal of the agents is to cooperatively minimize the sum of convex objective functions. When making decisions, each agent only has access to a stochastic gradient of its own objective function rather than the real gradient, and can exchange local state information with its immediate neighbors via a time-varying directed graph. Particularly, the stochasticity is depicted by the Brownian motion. To handle this problem, we propose a continuous-time distributed stochastic gradient algorithm based on the consensus algorithm and the gradient descent strategy. Under mild assumptions on the connectivity of the graph and objective functions, using convex analysis theory, the Lyapunov theory and Ito formula, we prove that the states of the agents asymptotically reach a common minimizer in expectation. Finally, a simulation example is worked out to demonstrate the effectiveness of our theoretical results.
