摘要
图神经网络 (GNN) 在节点分类、图分类和链路预测等多种任务中取得了巨大成功。然而,利用 GNN(以及更普遍的机器学习)来解决组合优化 (CO) 问题却鲜有探索。在这里,我们介绍了 GCON,这是一种新颖的 GNN 架构,它利用复杂的滤波器组和局部注意力机制来解决图上的组合优化问题。我们展示了我们的方法如何区别于之前的基于 GNN 的组合优化求解器,以及如何将其有效地应用于最大割、最小支配集和最大团问题,且无需监督学习。GCON 在所有任务中都具有竞争力,并且始终优于其他专门的基于 GNN 的方法,在最大割问题上与强大的 Gurobi 求解器不相上下。我们在 https://github.com/WenkelF/copt 提供了我们工作的开源实现。