A Practical GPU-Enhanced Matrix-Free Primal-Dual Method for Large-Scale Conic Programs
Zhenwei Lin, Zikai Xiong, Dongdong Ge, Yinyu Ye
In this paper, we introduce a practical GPU-enhanced matrix-free first-order method for solving large-scale conic programming problems, which we refer to as PDCS, standing for the Primal-Dual Conic Programming Solver. Problems that it solves include linear programs, second-order cone programs, convex quadratic programs, and exponential cone programs. The method avoids matrix factorizations and uses sparse matrix-vector multiplication as its core operation, which is memory-efficient and well suited for GPU acceleration. It builds on a variant of the primal-dual hybrid gradient method (PDHG) called reflected restarted Halpern PDHG with averaging, which we prove converges on general conic programs, and further incorporates several practical enhancements. Its GPU implementation, cuPDCS, uses customized computational schemes that assign cones of different types and sizes to different levels of the GPU architecture. Numerical experiments demonstrate that cuPDCS is generally more efficient than state-of-the-art commercial solvers and other first-order methods on large-scale conic program applications, including Fisher market equilibrium problems, Lasso regression, and multi-period portfolio optimization. cuPDCS also exhibits better scalability, efficiency, and robustness compared to other first-order methods on the conic program benchmark dataset CBLIB. These advantages are more pronounced in large-scale settings.
尚无 AI 解读,展示原文摘要