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

使用双向图注意力网络学习拓扑表示以解决作业车间调度问题

Learning Topological Representations with Bidirectional Graph Attention Network for Solving Job Shop Scheduling Problem

Nanyang Technological University(南洋理工大学);Singapore Management University(新加坡管理大学);Eindhoven University of Technology(埃因霍温理工大学);Institute of Marine Science and Technology, Shandong University(山东大学海洋科学与技术研究院)

Cong Zhang, Zhiguang Cao, Yaoxin Wu, Wen Song, Jing Sun

AI总结:

本文提出拓扑感知双向图注意力网络(TBGAT),通过正向和反向视图嵌入析取图并利用拓扑排序特征,在局部搜索框架中解决作业车间调度问题,计算复杂度呈线性,并在多个基准上取得新的SOTA结果。

摘要:

现有的基于学习的方法在解决作业车间调度问题(JSSP)时,通常使用为无向图量身定制的现成 GNN 模型,而忽略了析取图(DGs)丰富且有意义的拓扑结构。本文提出了拓扑感知双向图注意力网络(TBGAT),这是一种基于注意力机制的新型 GNN 架构,用于在局部搜索框架中嵌入 DG 以解决 JSSP。具体而言,TBGAT 分别从正向和反向视图嵌入 DG,其中消息通过遵循各视图的不同拓扑进行传播,并通过图注意力进行聚合。然后,我们提出了一种基于消息传递机制的新型算子来计算 DG 的正向和反向拓扑排序,这些排序是表征拓扑结构的特征,并被我们的模型所利用。此外,我们在理论和实验上表明,TBGAT 的计算复杂度分别与作业数量和机器数量呈线性关系,增强了本方法的实用价值。另外,在五个合成数据集和七个经典基准上的广泛实验表明,TBGAT 以较大优势超越了多种神经方法,取得了新的 SOTA 结果。所有代码和数据均已在 https://github.com/zcaicaros/TBGAT 公开。

英文摘要:

Existing learning-based methods for solving job shop scheduling problems (JSSP) usually use off-the-shelf GNN models tailored to undirected graphs and neglect the rich and meaningful topological structures of disjunctive graphs (DGs). This paper proposes the topology-aware bidirectional graph attention network (TBGAT), a novel GNN architecture based on the attention mechanism, to embed the DG for solving JSSP in a local search framework. Specifically, TBGAT embeds the DG from a forward and a backward view, respectively, where the messages are propagated by following the different topologies of the views and aggregated via graph attention. Then, we propose a novel operator based on the message-passing mechanism to calculate the forward and backward topological sorts of the DG, which are the features for characterizing the topological structures and exploited by our model. In addition, we theoretically and experimentally show that TBGAT has linear computational complexity to the number of jobs and machines, respectively, strengthening our method's practical value. Besides, extensive experiments on five synthetic datasets and seven classic benchmarks show that TBGAT achieves new SOTA results by outperforming a wide range of neural methods by a large margin. All the code and data are publicly available online at https://github.com/zcaicaros/TBGAT.

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

↑