Score: 3

ReLA: Representation Learning and Aggregation for Job Scheduling with Reinforcement Learning

Published: January 7, 2026 | arXiv ID: 2601.03646v1

By: Zhengyi Kwan , Zhang Wei , Aik Beng Ng and more

BigTech Affiliations: NVIDIA

Potential Business Impact:

Makes factory jobs finish much faster.

Business Areas:
Scheduling Information Technology, Software

Job scheduling is widely used in real-world manufacturing systems to assign ordered job operations to machines under various constraints. Existing solutions remain limited by long running time or insufficient schedule quality, especially when problem scale increases. In this paper, we propose ReLA, a reinforcement-learning (RL) scheduler built on structured representation learning and aggregation. ReLA first learns diverse representations from scheduling entities, including job operations and machines, using two intra-entity learning modules with self-attention and convolution and one inter-entity learning module with cross-attention. These modules are applied in a multi-scale architecture, and their outputs are aggregated to support RL decision-making. Across experiments on small, medium, and large job instances, ReLA achieves the best makespan in most tested settings over the latest solutions. On non-large instances, ReLA reduces the optimality gap of the SOTA baseline by 13.0%, while on large-scale instances it reduces the gap by 78.6%, with the average optimality gaps lowered to 7.3% and 2.1%, respectively. These results confirm that ReLA's learned representations and aggregation provide strong decision support for RL scheduling, and enable fast job completion and decision-making for real-world applications.

Country of Origin
🇺🇸 🇸🇬 United States, Singapore

Page Count
15 pages

Category
Computer Science:
Machine Learning (CS)