Abstract: In this paper, we consider the problem of stochastic optimization, where the objective function is in terms of the expectation of a (possibly non-convex) cost function that is parametrized by a random variable. While the convergence speed is critical for many emerging applications, most existing stochastic optimization methods suffer from slow convergence. Furthermore, the emerging technology of parallel computing has motivated an increasing demand for designing new stochastic optimization schemes that can handle parallel optimization for implementation in distributed systems. We propose a fast parallel stochastic optimization framework that can solve a large class of possibly non-convex stochastic optimization problems that may arise in applications with multi-agent systems. In the proposed method, each agent updates its control variable in parallel, by solving a convex quadratic subproblem independently. The convergence of the proposed method to the optimal solution for convex problems and to a stationary point for general non-convex problems is established. The proposed algorithm can be applied to solve a large class of optimization problems arising in important applications from various fields, such as machine learning and wireless networks. As a representative application of our proposed stochastic optimization framework, we focus on large-scale support vector machines and demonstrate how our algorithm can efficiently solve this problem, especially in modern applications with huge datasets. Using popular real-world datasets, we present experimental results to demonstrate the merits of our proposed framework by comparing its performance to the state-of-the-art in the literature. Numerical results show that the proposed method can significantly outperform the state-of-the-art methods in terms of the convergence speed while having the same or lower complexity and storage requirement.

Large-Scale Machine Learning with Stochastic Gradient Descent

Parallel Stochastic Optimization Framework for Large-Scale Non-Convex Stochastic Problems

Painless Stochastic Conjugate Gradient for Large-Scale Machine Learning

On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima

Stochastic Proximal Gradient Algorithm with Minibatches. Application to Large Scale Learning Models

Accelerated Almost-Sure Convergence Rates for Nonconvex Stochastic Gradient Descent using Stochastic Learning Rates

Stochastic Gradient Made Stable: A Manifold Propagation Approach for Large-Scale Optimization

Stochastic gradient descent methods for estimation with large data sets

Convergence Analysis of Asynchronous Stochastic Recursive Gradient Algorithms

Asymptotic Analysis via Stochastic Differential Equations of Gradient Descent Algorithms in Statistical and Computational Paradigms

Stochastic Ratios Tracking Algorithm for Large Scale Machine Learning Problems

Demystifying SGD with Doubly Stochastic Gradients

Understanding the unstable convergence of gradient descent.

Stochastic normalized gradient descent with momentum for large-batch training

Optimal Adaptive and Accelerated Stochastic Gradient Descent

Scaling up stochastic gradient descent for non-convex optimisation

Gradient Descent, Stochastic Optimization, and Other Tales

Novel Convergence Results of Adaptive Stochastic Gradient Descents

Accelerated Doubly Stochastic Gradient Algorithm for Large-scale Empirical Risk Minimization

Beyond Single-Model Views for Deep Learning: Optimization versus Generalizability of Stochastic Optimization Algorithms

Analysis of Stochastic Gradient Descent in Continuous Time