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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Cong, Cao, Zhiguang, Wu, Yaoxin, Song, Wen, Sun, Jing
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910472295415808
author Zhang, Cong
Cao, Zhiguang
Wu, Yaoxin
Song, Wen
Sun, Jing
author_facet Zhang, Cong
Cao, Zhiguang
Wu, Yaoxin
Song, Wen
Sun, Jing
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2402_17606
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Learning Topological Representations with Bidirectional Graph Attention Network for Solving Job Shop Scheduling Problem
Zhang, Cong
Cao, Zhiguang
Wu, Yaoxin
Song, Wen
Sun, Jing
Machine Learning
Artificial Intelligence
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.
title Learning Topological Representations with Bidirectional Graph Attention Network for Solving Job Shop Scheduling Problem
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2402.17606