arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.04495cs.LGcs.AI
收藏

通过非自回归神经网络学习求解正线性约束下的组合优化问题

Learning to Solve Combinatorial Optimization under Positive Linear Constraints via Non-Autoregressive Neural Networks

Shanghai Jiao Tong University(上海交通大学)

Runzhong Wang, Yang Li, Junchi Yan, Xiaokang Yang

AI总结:

本文提出一种非自回归神经网络框架,利用无监督学习和在线可微搜索求解正线性约束下的组合优化问题,在设施选址、最大集合覆盖和旅行商问题中达到与SCIP、Gurobi相竞争甚至更优的性能。

摘要:

组合优化(CO)是计算机科学、应用数学等交叉领域中的基本问题。CO问题固有的困难性给精确求解带来了挑战,使得基于深度神经网络的求解器成为研究前沿。本文设计了一族非自回归神经网络,用于求解正线性约束下的CO问题,其具有以下优点。第一,正线性约束覆盖了广泛的CO问题,表明我们的方法突破了现有非自回归网络的通用性瓶颈。第二,与现有的自回归神经网络求解器相比,我们的非自回归网络具有更高效率和保持置换不变性的优势。第三,我们的离线无监督学习对高质量标签的需求较低,摆脱了监督学习中对最优标签的需求。第四,我们的在线可微搜索方法显著提高了神经网络求解器对未见问题的泛化能力。我们验证了该框架在求解代表性CO问题(包括设施选址、最大集合覆盖和旅行商问题)中的有效性。我们的非自回归神经求解器可与SCIP和Gurobi等最先进求解器相竞争,甚至更优,尤其是在同时考虑效率和效果时。代码可在https://github.com/Thinklab-SJTU/NAR-CO-Solver获取。

英文摘要:

Combinatorial optimization (CO) is the fundamental problem at the intersection of computer science, applied mathematics, etc. The inherent hardness in CO problems brings up challenge for solving CO exactly, making deep-neural-network-based solvers a research frontier. In this paper, we design a family of non-autoregressive neural networks to solve CO problems under positive linear constraints with the following merits. First, the positive linear constraint covers a wide range of CO problems, indicating that our approach breaks the generality bottleneck of existing non-autoregressive networks. Second, compared to existing autoregressive neural network solvers, our non-autoregressive networks have the advantages of higher efficiency and preserving permutation invariance. Third, our offline unsupervised learning has lower demand on high-quality labels, getting rid of the demand of optimal labels in supervised learning. Fourth, our online differentiable search method significantly improves the generalizability of our neural network solver to unseen problems. We validate the effectiveness of this framework in solving representative CO problems including facility location, max-set covering, and traveling salesman problem. Our non-autoregressive neural solvers are competitive to and can be even superior to state-of-the-art solvers such as SCIP and Gurobi, especially when both efficiency and efficacy are considered. Code is available at https://github.com/Thinklab-SJTU/NAR-CO-Solver

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑