图神经网络求解线性半定规划的表示能力研究 / On the Expressive Power of GNNs to Solve Linear SDPs
1️⃣ 一句话总结
本文研究了图神经网络在求解半定规划问题中的能力,发现标准GNN无法解决这类问题,但提出了一种更强的GNN架构,它不仅能够模拟传统算法的求解过程,还能在预测精度和计算速度上显著优于现有方法,最高可将求解器加速80%。
Semidefinite programs (SDPs) are a powerful framework for convex optimization and for constructing strong relaxations of hard combinatorial problems. However, solving large SDPs can be computationally expensive, motivating the use of machine learning models as fast computational surrogates. Graph neural networks (GNNs) are a natural candidate in this setting due to their sparsity-awareness and ability to model variable-constraint interactions. In this work, we study what expressive power is sufficient to recover optimal SDP solutions. We first prove negative results showing that standard GNN architectures fail on recovering linear SDP solutions. We then identify a more expressive architecture that captures the key structure of SDPs and can, in particular, emulate the updates of a standard first-order solver. Empirically, on both synthetic and \textsc{SdpLib} benchmarks of various classes of SDPs, this more expressive architecture achieves consistently lower prediction error and objective gap than theoretically weaker baselines. Finally, using the learned high-quality predictions to warm-start the first-order solver yields practical speedups of up to 80%.
图神经网络求解线性半定规划的表示能力研究 / On the Expressive Power of GNNs to Solve Linear SDPs
本文研究了图神经网络在求解半定规划问题中的能力,发现标准GNN无法解决这类问题,但提出了一种更强的GNN架构,它不仅能够模拟传统算法的求解过程,还能在预测精度和计算速度上显著优于现有方法,最高可将求解器加速80%。
源自 arXiv: 2604.27786