Decision-focused Graph Neural Networks for Combinatorial Optimization

التفاصيل البيبلوغرافية
العنوان: Decision-focused Graph Neural Networks for Combinatorial Optimization
المؤلفون: Liu, Yang, Zhou, Chuan, Zhang, Peng, Pan, Shirui, Li, Zhao, Chen, Hongyang
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning, Computer Science - Artificial Intelligence
الوصف: In recent years, there has been notable interest in investigating combinatorial optimization (CO) problems by neural-based framework. An emerging strategy to tackle these challenging problems involves the adoption of graph neural networks (GNNs) as an alternative to traditional algorithms, a subject that has attracted considerable attention. Despite the growing popularity of GNNs and traditional algorithm solvers in the realm of CO, there is limited research on their integrated use and the correlation between them within an end-to-end framework. The primary focus of our work is to formulate a more efficient and precise framework for CO by employing decision-focused learning on graphs. Additionally, we introduce a decision-focused framework that utilizes GNNs to address CO problems with auxiliary support. To realize an end-to-end approach, we have designed two cascaded modules: (a) an unsupervised trained graph predictive model, and (b) a solver for quadratic binary unconstrained optimization. Empirical evaluations are conducted on various classical tasks, including maximum cut, maximum independent set, and minimum vertex cover. The experimental results on classical CO problems (i.e. MaxCut, MIS, and MVC) demonstrate the superiority of our method over both the standalone GNN approach and classical methods.
Comment: 9 pages
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2406.03647
رقم الأكسشن: edsarx.2406.03647
قاعدة البيانات: arXiv