We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Accelerating Quadratic Optimization with Reinforcement Learning

Abstract: First-order methods for quadratic optimization such as OSQP are widely used for large-scale machine learning and embedded optimal control, where many related problems must be rapidly solved. These methods face two persistent challenges: manual hyperparameter tuning and convergence time to high-accuracy solutions. To address these, we explore how Reinforcement Learning (RL) can learn a policy to tune parameters to accelerate convergence. In experiments with well-known QP benchmarks we find that our RL policy, RLQP, significantly outperforms state-of-the-art QP solvers by up to 3x. RLQP generalizes surprisingly well to previously unseen problems with varying dimension and structure from different applications, including the QPLIB, Netlib LP and Maros-Meszaros problems. Code for RLQP is available at this https URL
Comments: 25 pages, 7 figures. Code available at this https URL
Subjects: Machine Learning (cs.LG); Optimization and Control (math.OC)
Cite as: arXiv:2107.10847 [cs.LG]
  (or arXiv:2107.10847v1 [cs.LG] for this version)

Submission history

From: Paras Jain [view email]
[v1] Thu, 22 Jul 2021 17:59:10 GMT (286kb,D)

Link back to: arXiv, form interface, contact.