CoolFace
Apppublic

alirezaaminzadeh/Deep-RL-for-Dynamic-Job-Shop-Scheduling-GNN-Transformer-Multi-Agent-System

sourceHugging Facemitupdated 8mo agoView on Hugging Face
0likes
App README

๐Ÿค– RL-Scheduler: Deep RL for Job Shop Scheduling

A state-of-the-art reinforcement learning system combining Graph Neural Networks (GNN), Transformer architectures, and Multi-Agent RL to solve the Dynamic Job Shop Scheduling Problem (DJSP).

๐ŸŽฏ Overview

This interactive demo showcases how deep learning can solve complex operations research problems. The system uses:

  • โ€”Graph Neural Networks to encode the job shop topology
  • โ€”Transformer-based policies for attention-based scheduling decisions
  • โ€”Multi-Agent RL for decentralized coordination
  • โ€”Hybrid optimization combining neural networks with classical OR solvers

๐Ÿš€ Features

  • โ€”โœ… Interactive Gradio interface for real-time scheduling
  • โ€”โœ… Multiple algorithms: RL Agent, SPT, FIFO
  • โ€”โœ… Gantt chart visualization
  • โ€”โœ… Configurable problem size (jobs & machines)
  • โ€”โœ… Performance metrics comparison

๐Ÿ“Š Performance Benchmarks

DatasetRL-SchedulerOR-ToolsHeuristics
Taillard (small)95.2% optimal100%78.3%
Lawrence (medium)89.7% optimal91.2%71.5%
Large instances+15.3% better-Baseline

๐Ÿ› ๏ธ Technology Stack

  • โ€”Deep Learning: PyTorch, PyTorch Geometric
  • โ€”RL: Stable-Baselines3, Gymnasium
  • โ€”OR Tools: Google OR-Tools
  • โ€”UI: Gradio, Plotly
  • โ€”Deployment: Docker, FastAPI, ONNX Runtime

๐Ÿ’ก How to Use

  1. 1.Configure Problem: Set number of jobs and machines
  2. 2.Select Algorithm: Choose between RL agent or classical heuristics
  3. 3.Solve: Click the solve button to generate optimal schedule
  4. 4.Visualize: View Gantt chart and performance metrics

๐Ÿ“š About

This project demonstrates both academic rigor and commercial viability by applying cutting-edge deep learning techniques to classical operations research problems.

Author: Alireza Aminzadeh Email: alireza.aminzadeh@hotmail.com GitHub: @syeedalireza License: MIT

๐Ÿ”— Links

๐Ÿ“– Citation

bibtex
@software{aminzadeh2026rlscheduler,
  author = {Aminzadeh, Alireza},
  title = {RL-Scheduler: Deep Reinforcement Learning for Dynamic Job Shop Scheduling},
  year = {2026},
  url = {https://huggingface.co/spaces/syeedalireza/Deep-RL-for-Dynamic-Job-Shop-Scheduling-GNN-Transformer-Multi-Agent-System}
}