arrow
Return

Asynchronous Parallel, Sparse Approximated SVRG for High-Dimensional Machine Learning

delete2022-12-01
delete7
PRE
AI
F
Fanhua Shang
H
Hua Huang
J
Jun Fan *
刘圆圆 (Yuanyuan Liu) *
刘红英 cover
刘红英 (Hongying Liu)
DOI:10.1109/TKDE.2021.3070539delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
With the increasing of the data size and the development of multi-core computers, asynchronous parallel stochastic optimization algorithms such as KroMagnon have gained significant attention. In this paper, we propose a new Sparse approximation and asynchronous parallel Stochastic Variance Reduced Gradient (SSVRG) method for sparse and high-dimensional machine learning problems. Unlike standard SVRG and its asynchronous parallel variant, KroMagnon, the snapshot point of SSVRG is set to the average of all the iterates in the previous epoch, which allows it to take much larger learning rates and also makes it more robust to the choice of learning rates. In particular, we use the sparse approximation of the popular SVRG estimator to perform completely sparse updates at all iterations. Therefore, SSVRG has a much lower per-iteration computational cost than its dense counterpart, SVRG++, and is very friendly to asynchronous parallel implementation. Moreover, we provide the convergence guarantees of SSVRG for both strongly convex and non-strongly convex problems, while existing asynchronous algorithms (e.g., KroMagnon and ASAGA) only have convergence guarantees for strongly convex problems. Finally, we extend SSVRG to non-smooth and asynchronous parallel settings. Numerical experimental results demonstrate that SSVRG converges significantly faster than the state-of-the-art asynchronous parallel methods, e.g., KroMagnon, and is usually more than three orders of magnitude faster than SVRG++.
Keywords:
Convergence
Machine learning
Stochastic processes
Acceleration
Radio frequency
Parallel algorithms
Optimization
Empirical risk minimization
stochastic optimization
variance reduction
asynchronous parallel
sparse approximation
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

IEEE Transactions on Knowledge and Data Engineering cover
IEEE Transactions on Knowledge and Data Engineering
IF:
10.4
Papers:
6.7K
Citations:
3.2W

Organization

H
hebei university of technology
Scholars:
1.8W
Papers: 1.2W
Citations: 10
X
Xidian University
Scholars:
2.4W
Papers: 1.9W
Citations: 9.7K