Data-driven Projection Generation for Efficiently Solving Heterogeneous Quadratic Programming Problems
Tomoharu Iwata, Futoshi Futami
TL;DR
This work presents a data-driven framework for efficiently solving large, heterogeneous QP problems by learning instance-specific projection matrices that reduce dimensionality from $N$ to $K$ via a graph neural network. The projected QP is solved with a solver, and the original solution is obtained by back-projecting, ensuring feasibility while aiming to minimize the original objective. Training uses a bilevel formulation with gradients computed through the envelope theorem, avoiding backpropagation through the inner solver, and the approach is backed by a generalization bound based on covering numbers and Lipschitz properties. Empirical results on Regression, Portfolio, and Control datasets show strong solution quality, feasibility, and substantial speedups over solving the full QP, with robust generalization across varying problem sizes and structures.
Abstract
We propose a data-driven framework for efficiently solving quadratic programming (QP) problems by reducing the number of variables in high-dimensional QPs using instance-specific projection. A graph neural network-based model is designed to generate projections tailored to each QP instance, enabling us to produce high-quality solutions even for previously unseen problems. The model is trained on heterogeneous QPs to minimize the expected objective value evaluated on the projected solutions. This is formulated as a bilevel optimization problem; the inner optimization solves the QP under a given projection using a QP solver, while the outer optimization updates the model parameters. We develop an efficient algorithm to solve this bilevel optimization problem, which computes parameter gradients without backpropagating through the solver. We provide a theoretical analysis of the generalization ability of solving QPs with projection matrices generated by neural networks. Experimental results demonstrate that our method produces high-quality feasible solutions with reduced computation time, outperforming existing methods.
