Optimization for Classical Machine Learning Problems on the GPU
Sören Laue, Mark Blacher, Joachim Giesen
[AAAI-22] Main Track
Abstract:
Constrained optimization problems arise frequently in classical machine learning. There exist frameworks addressing constrained optimization, for instance, CVXPY, but so far, in contrast to deep learning frameworks, GPU support is limited. Here, we provide a framework for solving constrained optimization problems on the GPU. The framework allows the user to specify constrained optimization problems in an easy-to-read modeling language. A solver is then automatically generated from this specification. When run on the GPU, the solver outperforms state-of-the-art approaches like CVXPY combined with a GPU-accelerated solver such as cuOSQP or SCS by several orders of magnitude.
Introduction Video
Sessions where this paper appears
-
Poster Session 1
Thu, February 24 4:45 PM - 6:30 PM (+00:00)
Blue 3
-
Poster Session 12
Mon, February 28 8:45 AM - 10:30 AM (+00:00)
Blue 3