Efficient Distributed Online Prediction and Stochastic Optimization With Approximate Distributed Averaging

摘要

We study distributed methods for online prediction and stochastic optimization. Our approach is iterative: In each round, nodes first perform local computations and then communicate in order to aggregate information and synchronize their decision variables. Synchronization is accomplished through the use of a distributed averaging protocol. When an exact distributed averaging protocol is used, it is known that the optimal regret bound of O(m−−√) can be achieved using the distributed minibatch algorithm in [“Optimal distributed online prediction using mini-batches,” J. Mach. Learn. Res., vol. 13, pp. 165–202, 2012], where m is the total number of samples processed across the network. We focus on methods using approximate distributed averaging protocols and show that the optimal regret bound can also be achieved in this setting. In particular, we propose a gossip-based optimization method that achieves the optimal regret bound. The amount of communication required depends on the network topology through the second largest eigenvalue of the transition matrix of a random walk on the network. In the setting of stochastic optimization, the proposed gossip-based approach achieves nearly linear scaling: The optimization error is guaranteed to be no more than ϵ after O(1nϵ2) rounds, each of which involves O(logn) gossip iterations, when n nodes communicate over a well-connected graph.