An Efficient Unsupervised Framework for Convex Quadratic Programs via Deep Unrolling

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Yang, Linxin, Li, Bingheng, Ding, Tian, Wu, Jianghua, Wang, Akang, Wang, Yuyi, Tang, Jiliang, Sun, Ruoyu, Luo, Xiaodong
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909411528671232
author Yang, Linxin
Li, Bingheng
Ding, Tian
Wu, Jianghua
Wang, Akang
Wang, Yuyi
Tang, Jiliang
Sun, Ruoyu
Luo, Xiaodong
author_facet Yang, Linxin
Li, Bingheng
Ding, Tian
Wu, Jianghua
Wang, Akang
Wang, Yuyi
Tang, Jiliang
Sun, Ruoyu
Luo, Xiaodong
contents Quadratic programs (QPs) arise in various domains such as machine learning, finance, and control. Recently, learning-enhanced primal-dual hybrid gradient (PDHG) methods have shown great potential in addressing large-scale linear programs; however, this approach has not been extended to QPs. In this work, we focus on unrolling "PDQP", a PDHG algorithm specialized for convex QPs. Specifically, we propose a neural network model called "PDQP-net" to learn optimal QP solutions. Theoretically, we demonstrate that a PDQP-net of polynomial size can align with the PDQP algorithm, returning optimal primal-dual solution pairs. We propose an unsupervised method that incorporates KKT conditions into the loss function. Unlike the standard learning-to-optimize framework that requires optimization solutions generated by solvers, our unsupervised method adjusts the network weights directly from the evaluation of the primal-dual gap. This method has two benefits over supervised learning: first, it helps generate better primal-dual gap since the primal-dual gap is in the objective function; second, it does not require solvers. We show that PDQP-net trained in this unsupervised manner can effectively approximate optimal QP solutions. Extensive numerical experiments confirm our findings, indicating that using PDQP-net predictions to warm-start PDQP can achieve up to 45% acceleration on QP instances. Moreover, it achieves 14% to 31% acceleration on out-of-distribution instances.
format Preprint
id arxiv_https___arxiv_org_abs_2412_01051
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle An Efficient Unsupervised Framework for Convex Quadratic Programs via Deep Unrolling
Yang, Linxin
Li, Bingheng
Ding, Tian
Wu, Jianghua
Wang, Akang
Wang, Yuyi
Tang, Jiliang
Sun, Ruoyu
Luo, Xiaodong
Optimization and Control
Machine Learning
Quadratic programs (QPs) arise in various domains such as machine learning, finance, and control. Recently, learning-enhanced primal-dual hybrid gradient (PDHG) methods have shown great potential in addressing large-scale linear programs; however, this approach has not been extended to QPs. In this work, we focus on unrolling "PDQP", a PDHG algorithm specialized for convex QPs. Specifically, we propose a neural network model called "PDQP-net" to learn optimal QP solutions. Theoretically, we demonstrate that a PDQP-net of polynomial size can align with the PDQP algorithm, returning optimal primal-dual solution pairs. We propose an unsupervised method that incorporates KKT conditions into the loss function. Unlike the standard learning-to-optimize framework that requires optimization solutions generated by solvers, our unsupervised method adjusts the network weights directly from the evaluation of the primal-dual gap. This method has two benefits over supervised learning: first, it helps generate better primal-dual gap since the primal-dual gap is in the objective function; second, it does not require solvers. We show that PDQP-net trained in this unsupervised manner can effectively approximate optimal QP solutions. Extensive numerical experiments confirm our findings, indicating that using PDQP-net predictions to warm-start PDQP can achieve up to 45% acceleration on QP instances. Moreover, it achieves 14% to 31% acceleration on out-of-distribution instances.
title An Efficient Unsupervised Framework for Convex Quadratic Programs via Deep Unrolling
topic Optimization and Control
Machine Learning
url https://arxiv.org/abs/2412.01051