Decision Transformer for Enhancing Neural Local Search on the Job Shop Scheduling Problem

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: de Puiseau, Constantin Waubert, Wolz, Fabian, Montag, Merlin, Peters, Jannik, Tercan, Hasan, Meisen, Tobias
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917911802675200
author de Puiseau, Constantin Waubert
Wolz, Fabian
Montag, Merlin
Peters, Jannik
Tercan, Hasan
Meisen, Tobias
author_facet de Puiseau, Constantin Waubert
Wolz, Fabian
Montag, Merlin
Peters, Jannik
Tercan, Hasan
Meisen, Tobias
contents The job shop scheduling problem (JSSP) and its solution algorithms have been of enduring interest in both academia and industry for decades. In recent years, machine learning (ML) is playing an increasingly important role in advancing existing and building new heuristic solutions for the JSSP, aiming to find better solutions in shorter computation times. In this paper we build on top of a state-of-the-art deep reinforcement learning (DRL) agent, called Neural Local Search (NLS), which can efficiently and effectively control a large local neighborhood search on the JSSP. In particular, we develop a method for training the decision transformer (DT) algorithm on search trajectories taken by a trained NLS agent to further improve upon the learned decision-making sequences. Our experiments show that the DT successfully learns local search strategies that are different and, in many cases, more effective than those of the NLS agent itself. In terms of the tradeoff between solution quality and acceptable computational time needed for the search, the DT is particularly superior in application scenarios where longer computational times are acceptable. In this case, it makes up for the longer inference times required per search step, which are caused by the larger neural network architecture, through better quality decisions per step. Thereby, the DT achieves state-of-the-art results for solving the JSSP with ML-enhanced search.
format Preprint
id arxiv_https___arxiv_org_abs_2409_02697
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Decision Transformer for Enhancing Neural Local Search on the Job Shop Scheduling Problem
de Puiseau, Constantin Waubert
Wolz, Fabian
Montag, Merlin
Peters, Jannik
Tercan, Hasan
Meisen, Tobias
Artificial Intelligence
Machine Learning
The job shop scheduling problem (JSSP) and its solution algorithms have been of enduring interest in both academia and industry for decades. In recent years, machine learning (ML) is playing an increasingly important role in advancing existing and building new heuristic solutions for the JSSP, aiming to find better solutions in shorter computation times. In this paper we build on top of a state-of-the-art deep reinforcement learning (DRL) agent, called Neural Local Search (NLS), which can efficiently and effectively control a large local neighborhood search on the JSSP. In particular, we develop a method for training the decision transformer (DT) algorithm on search trajectories taken by a trained NLS agent to further improve upon the learned decision-making sequences. Our experiments show that the DT successfully learns local search strategies that are different and, in many cases, more effective than those of the NLS agent itself. In terms of the tradeoff between solution quality and acceptable computational time needed for the search, the DT is particularly superior in application scenarios where longer computational times are acceptable. In this case, it makes up for the longer inference times required per search step, which are caused by the larger neural network architecture, through better quality decisions per step. Thereby, the DT achieves state-of-the-art results for solving the JSSP with ML-enhanced search.
title Decision Transformer for Enhancing Neural Local Search on the Job Shop Scheduling Problem
topic Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2409.02697