Stochastic Primal-Dual Method for Empirical Risk Minimization with O(1) Per-Iteration Complexity

被引:0
|
作者
Tan, Conghui [1 ]
Zhang, Tong [2 ]
Ma, Shiqian [3 ]
Liu, Ji [4 ]
机构
[1] Chinese Univ Hong Kong, Hong Kong, Peoples R China
[2] Tencent AI Lab, Bellevue, WA USA
[3] Univ Calif Davis, Davis, CA 95616 USA
[4] Univ Rochester, Tencent AI Lab, Rochester, NY 14627 USA
基金
中国国家自然科学基金;
关键词
D O I
暂无
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Regularized empirical risk minimization problem with linear predictor appears frequently in machine learning. In this paper, we propose a new stochastic primaldual 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.
引用
收藏
页数:10
相关论文
共 50 条
  • [1] Stochastic Primal-Dual Coordinate Method for Regularized Empirical Risk Minimization
    Zhang, Yuchen
    Xiao, Lin
    [J]. INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 37, 2015, 37 : 353 - 361
  • [2] Stochastic Primal-Dual Coordinate Method for Regularized Empirical Risk Minimization
    Zhang, Yuchen
    Xiao, Lin
    [J]. JOURNAL OF MACHINE LEARNING RESEARCH, 2017, 18
  • [3] Distributed Primal-Dual Proximal Method for Regularized Empirical Risk Minimization
    Khuzani, Masoud Badiei
    [J]. 2018 17TH IEEE INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND APPLICATIONS (ICMLA), 2018, : 938 - 945
  • [4] Decentralized Stochastic Bilevel Optimization with Improved per-Iteration Complexity
    Chen, Xuxing
    Huang, Minhui
    Ma, Shiqian
    Balasubramanian, Krishnakumar
    [J]. INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 202, 2023, 202
  • [5] A primal-dual algorithm for risk minimization
    Kouri, Drew P.
    Surowiec, Thomas M.
    [J]. MATHEMATICAL PROGRAMMING, 2022, 193 (01) : 337 - 363
  • [6] On the iteration complexity analysis of Stochastic Primal-Dual Hybrid Gradient approach with high probability
    Qiao, Linbo
    Lin, Tianyi
    Qin, Qi
    Lu, Xicheng
    [J]. NEUROCOMPUTING, 2018, 307 : 78 - 90
  • [7] Doubly Greedy Primal-Dual Coordinate Descent for Sparse Empirical Risk Minimization
    Lei, Qi
    Yen, Ian E. H.
    Wu, Chao-yuan
    Dhillon, Inderjit S.
    Ravikumar, Pradeep
    [J]. INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 70, 2017, 70
  • [8] Primal-dual first-order methods with O(1/ε) iteration-complexity for cone programming
    Lan, Guanghui
    Lu, Zhaosong
    Monteiro, Renato D. C.
    [J]. MATHEMATICAL PROGRAMMING, 2011, 126 (01) : 1 - 29
  • [9] A Stochastic Primal-Dual Method for Optimization with Conditional Value at Risk Constraints
    Avinash N. Madavan
    Subhonmesh Bose
    [J]. Journal of Optimization Theory and Applications, 2021, 190 : 428 - 460
  • [10] On Iteration Complexity of a First-Order Primal-Dual Method for Nonlinear Convex Cone Programming
    Zhao, Lei
    Zhu, Dao-Li
    [J]. JOURNAL OF THE OPERATIONS RESEARCH SOCIETY OF CHINA, 2022, 10 (01) : 53 - 87