Proceedings

GATES: Cost-aware Dynamic Workflow Scheduling via Graph Attention Networks and Evolution Strategy

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 8635-8643.

https://doi.org/10.24963/ijcai.2025/960

Abstract

Cost-aware Dynamic Workflow Scheduling (CADWS) is a key challenge in cloud computing, focusing on devising an effective scheduling policy to efficiently schedule dynamically arriving workflow tasks, represented as Directed Acyclic Graphs (DAG), to suitable virtual machines (VMs). Deep reinforcement learning (DRL) has been widely employed for automated scheduling policy design. However, the performance of DRL is heavily influenced by the design of the problem-tailored policy network and is highly sensitive to hyperparameters and the design of reward feedback. Considering the above-mentioned issues, this study proposes a novel DRL method combining Graph Attention Networks-based policy network and Evolution Strategy, referred to as GATES. The contributions of GATES are summarized as follows: (1) GATES can capture the impact of current task scheduling on subsequent tasks by learning the topological relationships between tasks in a DAG. (2) GATES can assess the importance of each VM to the ready task, enabling it to adapt to dynamically changing VM resources. (3) Utilizing Evolution Strategy's robustness, exploratory nature, and tolerance for delayed rewards, GATES achieves stable policy learning in CADWS. Extensive experimental results demonstrate the superiority of the proposed GATES in CADWS, outperforming several state-of-the-art algorithms. The source code is available at: https://github.com/YaShen998/GATES.

  1. Scenarios
  2. ProLis GRP-HEFT ES-RL SPN-CWS GATES (our) ⟨1.00, S⟩ 773.69 1685.01(-) 223.68(51.51)(+)(+) 158.28(14.45)(+)(+)(+) 143.95(18.94)(+)(+)(+)(+) ⟨1.00, M ⟩ 1829.25 2867.64(-) 422.26(100.92)(+)(+) 308.93(75.88)(+)(+)(+) 225.41(37.21)() ⟨1.00, L⟩ 3641.79 5873.20(-) 829.64(262.91)(+)(+) 524.83(163.15)(+)(+)(+) 324.93(65.))(+) ⟨1.25, S⟩ 923.15 1967.31(-) 320.75(83.69)(+)(+) 143.12(15.73)(+)(+)(+) 124.14
  3. ⟨1., M ⟩ 2068.78 3578.04(-) 565.53(213.00)(+)(+) 272.77(60.65)(+)(+)(+) 193.26(24.)) ⟨1.25, L⟩ 4213.29 6868.22(-) 989.79(458.04)(+)(+) 496.63(155.02)(+)(+)(+) 300.19(76.))(+) ⟨1.50, S⟩ 901.40 1963.39(-) 270.93(67.20)(+)(+) 144.13(18.97)(+)(+)(+) 121.56
  4. ⟨1., M ⟩ 2035.52 3567.97(-) 505.44(222.45)(+)(+) 276.09(60.63)(+)(+)(+) 196.90(33.)) ⟨1.50, L⟩ 4066.84 6863.62(-) 886.87(380.26)(+)(+) 511.69(161.85)(+)(+)(+) 316.23(99.))(+) ⟨1.75, S⟩ 804.33 1959.71(-) 241.09(51.43)(+)(+) 137.36(14.10)(+)(+)(+) 118.45
  5. ⟨1., M ⟩ 1977.10 3560.22(-) 491.97(229.48)(+)(+) 263.12(57.61)(+)(+)(+) 195.25(35.)) ⟨1.75, L⟩ 3898.42 6854.40(-) 828.06(359.74)(+)(+) 502.67(155.64)(+)(+)(+) 315.84(92.))(+) ⟨2.00, S⟩ 789.71 1957.25(-) 225.70(50.89)(+)(+) 129.75(12.72)(+)(+)(+) 113.48
  6. ⟨2., M ⟩ 1921.42 3554.07(-) 427.16(213.71)(+)(+) 251.83(52.01)(+)(+)(+) 185.55(30.)) ⟨2.00, L⟩ 4024.03 6847.49(-) 780.77(408.14)(+)(+) 497.65(158.52)(+)(+)(+) 322.01(110.))(+) ⟨2.25, S⟩ 710.18 1954.71(-) 201.54(50.56)(+)(+) 126.18(13.02)(+)(+)(+) 110.90
  7. ⟨2., M ⟩ 1924.27 3551.31(-) 374.96(190.21)(+)(+) 238.35(48.26)(+)(+)(+) 179.17(24.)) ⟨2.25, L⟩ 3957.08 6842.88(-) 678.94(396.79)(+)(+) 475.46(139.99)(+)(+)(+) 318.19(112al., 2015]
  8. Jürgen Branke, Su Nguyen, Christoph W Pickardt, and Mengjie Zhang. Auto- mated design of production scheduling heuristics: A review. IEEE Transactions on Evolutionary Computation, 20(1):110-124, 2015.
  9. [Cunha et al., 2020] Bruno Cunha, Ana M Madureira, Ben- jamim Fonseca, and Duarte Coelho. Deep reinforcement learning as a job shop scheduling solver: A literature re- view. In 18th International Conference on Hybrid Intelli- gent Systems, pages 350-359. Springer, 2020. Deelman et al., 2015]
  10. Ewa Deelman, Karan Vahi, Gideon Juve, Mats Rynge, Scott Callaghan, Philip J Maechling, Rajiv Mayani, Weiwei Chen, Rafael Ferreira Da Silva, Miron Livny, et al. Pegasus, a workflow management sys- tem for science automation. Future Generation Computer Systems, 46:17-35, 2015.
  11. [Dong et al., 2021] Hang Dong, Boshi Wang, Bo Qiao, Wen- qian Xing, Chuan Luo, Si Qin, Qingwei Lin, Dongmei Zhang, Gurpreet Virdi, and Thomas Moscibroda. Predic- tive job scheduling under uncertain constraints in cloud computing. In IJCAI, pages 3627-3634, 2021. IJCAI
  12. [Faragardi et al., 2019] Hamid Reza Faragardi, Mohammad Reza Saleh Sedghpour, Saber Fazliahmadi, Thomas Fahringer, and Nayereh Rasouli. Grp-heft: A budget- constrained resource provisioning scheme for workflow scheduling in iaas clouds. IEEE Transactions on Paral- lel and Distributed Systems, 31(6):1239-1254, 2019.
  13. [Geiger et al., 2006] Christopher D Geiger, Reha Uzsoy, and Haldun Aytug. Rapid modeling and discovery of prior- ity dispatching rules: An autonomous learning approach. Journal of Scheduling, 9:7-34, 2006.
  14. [HasanzadeZonuzy et al., 2021] Aria HasanzadeZonuzy, Dileep Kalathil, and Srinivas Shakkottai. Model-based reinforcement learning for infinite-horizon discounted constrained markov decision processes. IJCAI 2021, 2021. IJCAI
  15. [Huang et al., 2022] Victoria Huang, Chen Wang, Hui Ma, Gang Chen, and Kameron Christopher. Cost-aware dy- namic multi-workflow scheduling in cloud data center us- ing evolutionary reinforcement learning. In International Conference on Service-Oriented Computing, pages 449- 464. Springer, 2022.
  16. [Iklassov et al., 2023] Zangir Iklassov, Dmitrii Medvedev, Ruben Solozabal Ochoa De Retana, and Martin Takac. On the study of curriculum learning for inferring dispatching policies on the job shop scheduling. In IJCAI, pages 5350- 5358, 2023. IJCAI
  17. [Jadeja and Modi, 2012] Yashpalsinh Jadeja and Kirit Modi. Cloud computing-concepts, architecture and challenges. In 2012 international conference on computing, electron- ics and electrical technologies (ICCEET), pages 877-880. IEEE, 2012.
  18. [Kayhan and Yildiz, 2023] Behice Meltem Kayhan and Gokalp Yildiz. Reinforcement learning applications to machine scheduling problems: a comprehensive literature review. Journal of Intelligent Manufacturing, 34(3):905-929, 2023.
  19. [Khadka and Tumer, 2018] Shauharda Khadka and Kagan Tumer. Evolution-guided policy gradient in reinforcement learning. Advances in Neural Information Processing Sys- tems, 31, 2018.
  20. [Khalil et al., 2017] Elias Khalil, Hanjun Dai, Yuyu Zhang, Bistra Dilkina, and Le Song. Learning combinatorial op- timization algorithms over graphs. Advances in neural in- formation processing systems, 30, 2017.
  21. [Ning et al., 2020] Xuefei Ning, Yin Zheng, Tianchen Zhao, Yu Wang, and Huazhong Yang. A generic graph-based neural architecture encoding scheme for predictor-based nas. In European Conference on Computer Vision, pages 189-204. Springer, 2020.
  22. [Ou et al., 2023] Junwei Ou, Lining Xing, Feng Yao, Mengjun Li, Jimin Lv, Yongming He, Yanjie Song, Jian Wu, and Guoting Zhang. Deep reinforcement learning method for satellite range scheduling problem. Swarm and Evolutionary Computation, 77:101233, 2023.
  23. [Pallathadka et al., 2022] Harikumar Pallathadka, Guna Sekhar Sajja, Khongdet Phasinam, Mahyudin Ritonga, Mohd Naved, Rajni Bansal, and Jose Quiñonez- Choquecota. An investigation of various applications and related challenges in cloud computing. Materials Today: Proceedings, 51:2245-2248, 2022.
  24. [Pham and Fahringer, 2020] Thanh-Phuong Pham and Thomas Fahringer. Evolutionary multi-objective work- flow scheduling for volatile resources in the cloud. IEEE Transactions on Cloud Computing, 10(3):1780-1791, 2020.
  25. [Price et al., 2025] Ilan Price, Alvaro Sanchez-Gonzalez, Ferran Alet, Tom R Andersson, Andrew El-Kadi, Dominic Masters, Timo Ewalds, Jacklynn Stott, Shakir Mohamed, Peter Battaglia, et al. Probabilistic weather forecasting with machine learning. Nature, 637(8044):84-90, 2025.
  26. [Qazi et al., 2024] Faiza Qazi, Daehan Kwak, Fiaz Gul Khan, Farman Ali, and Sami Ullah Khan. Service level agreement in cloud computing: Taxonomy, prospects, and challenges. Internet of Things, page 101126, 2024.
  27. [Salimans et al., 2017] Tim Salimans, Jonathan Ho, Xi Chen, Szymon Sidor, and Ilya Sutskever. Evolution strategies as a scalable alternative to reinforcement learning. arXiv preprint arXiv:1703.03864, 2017.
  28. [Shen et al., 2024] Ya Shen, Gang Chen, Hui Ma, and Mengjie Zhang. Cost-aware dynamic cloud workflow scheduling using self-attention and evolutionary reinforce- ment learning. In International Conference on Service- Oriented Computing, pages 3-18. Springer, 2024.
  29. [Song et al., 2022] Wen Song, Xinyang Chen, Qiqiang Li, and Zhiguang Cao. Flexible job-shop scheduling via graph neural network and deep reinforcement learning. IEEE Transactions on Industrial Informatics, 19(2):1600-1610, 2022.
  30. [Sun and Yang, 2023] Zhiqing Sun and Yiming Yang. Di- fusco: Graph-based diffusion solvers for combinatorial optimization. Advances in Neural Information Processing Systems, 36:3706-3731, 2023.
  31. [Thekumparampil et al., 2018] Kiran K Thekumparampil, Chong Wang, Sewoong Oh, and Li-Jia Li. Attention-based graph neural network for semi-supervised learning. arXiv preprint arXiv:1803.03735, 2018.
  32. [Vaswani, 2017] A Vaswani. Attention is all you need. Ad- vances in Neural Information Processing Systems, 2017.
  33. [Veličković et al., 2017] Petar Veličković, Guillem Cucurull, Arantxa Casanova, Adriana Romero, Pietro Lio, and Yoshua Bengio. Graph attention networks. arXiv preprint arXiv:1710.10903, 2017.
  34. [Wang et al., 2019] Xiao Wang, Houye Ji, Chuan Shi, Bai Wang, Yanfang Ye, Peng Cui, and Philip S Yu. Hetero- geneous graph attention network. In The world wide web conference, pages 2022-2032, 2019.
  35. [Wu et al., 2017] Quanwang Wu, Fuyuki Ishikawa, Qing- sheng Zhu, Yunni Xia, and Junhao Wen. Deadline- constrained cost optimization approaches for workflow scheduling in clouds. IEEE Transactions on Parallel and Distributed Systems, 28(12):3401-3412, 2017.
  36. [Yang et al., 2022] Yifan Yang, Gang Chen, Hui Ma, and Mengjie Zhang. Dual-tree genetic programming for deadline-constrained dynamic workflow scheduling in cloud. In Service-Oriented Computing, pages 433-448. Springer, 2022.
  37. [Yang et al., 2025] Yifan Yang, Gang Chen, Hui Ma, Cong Zhang, Zhiguang Cao, and Mengjie Zhang. Graph as- sisted offline-online deep reinforcement learning for dy- namic workflow scheduling. In The Thirteenth Interna- tional Conference on Learning Representations, 2025.
  38. [Zhang et al., 2020] Cong Zhang, Wen Song, Zhiguang Cao, Jie Zhang, Puay Siew Tan, and Xu Chi. Learning to dispatch for job shop scheduling via deep reinforcement learning. Advances in neural information processing sys- tems, 33:1621-1632, 2020.
  39. [Zhang et al., 2024] Cong Zhang, Zhiguang Cao, Wen Song, Yaoxin Wu, and Jie Zhang. Deep reinforcement learning guided improvement heuristic for job shop scheduling. In The Twelfth International Conference on Learning Repre- sentations, 2024.
  40. [Zhu et al., 2016] Xiaomin Zhu, Ji Wang, Hui Guo, Dakai Zhu, Laurence T Yang, and Ling Liu. Fault-tolerant scheduling for real-time scientific workflows with elastic resource provisioning in virtualized clouds. IEEE Trans- actions on Parallel and Distributed Systems, 27(12):3501- 3517, 2016.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards