Asynchronous Mini-Batch Algorithm for Regularized Stochastic Optimization
2016 (English)In: IEEE Transactions on Automatic Control, ISSN 0018-9286, E-ISSN 1558-2523, ISSN 0018-9286Article in journal (Refereed) Published
Mini-batch optimization has proven to be a powerful paradigm for large-scale learning. However, the state of the art parallel mini–batch algorithms assume synchronous operation or cyclic update orders. When worker nodes are heterogeneous (due to different computational capabilities or different communication delays), synchronous and cyclic operations are inefficient since they will leave workers idle waiting for the slower nodes to complete their computations. In this paper, we propose an asynchronous mini-batch algorithm for regularized stochastic optimization problems with smooth loss functions that eliminates idle waiting and allows workers to run at their maximal update rates. We show that by suitably choosing the step-size values, the algorithm achieves a rate of the order O(1/ √ T) for general convex regularization functions, and the rate O(1/T ) for strongly convex regularization functions, where T is the number of iterations. In both cases, the impact of asynchrony on the convergence rate of our algorithm is asymptotically negligible, and a nearlinear speedup in the number of workers can be expected. Theoretical results are confirmed in real implementations on a distributed computing infrastructure.
Place, publisher, year, edition, pages
IEEE , 2016.
Optimization, Asynchronous, Delay, Convex, Nonsmooth
Engineering and Technology
Research subject Electrical Engineering
IdentifiersURN: urn:nbn:se:kth:diva-182598DOI: 10.1109/TAC.2016.2525015OAI: oai:DiVA.org:kth-182598DiVA: diva2:904927
QC 201603112016-02-202016-02-202016-03-11Bibliographically approved