Stochastic Primal-Dual Method for Empirical Risk Minimization with O(1) Per-Iteration Complexity
Tan, Conghui, Zhang, Tong, Ma, Shiqian, Liu, Ji
–Neural Information Processing Systems
Regularized empirical risk minimization problem with linear predictor appears frequently in machine learning. In this paper, we propose a new stochastic primal-dual method to solve this class of problems. Different from existing methods, our proposed methods only require O(1) operations in each iteration. We also develop a variance-reduction variant of the algorithm that converges linearly. Numerical experiments suggest that our methods are faster than existing ones such as proximal SGD, SVRG and SAGA on high-dimensional problems.
Neural Information Processing Systems
Dec-31-2018
- Country:
- North America > United States > California (0.14)
- Genre:
- Research Report > Experimental Study (0.34)
- Technology: