Proceedings

No Regret Reinforcement Learning Algorithms for Online Scheduling with Multi-Stage Tasks

Download BibTeX Cite

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

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

Abstract

We study online task scheduling problems where tasks arrive sequentially and are processed by the platform or server. The service processes for tasks are multi-stage and are modeled as episodic Markov Decision Processes (MDPs). While processing a task, the system acquires rewards by consuming resources. The goal of the platform is to maximize the reward-to-cost ratio over a sequence of K tasks.

Online scheduling with multi-stage tasks faces two major challenges: intra-dependence among the different stages within a task and inter-dependence among different tasks. These challenges are further exacerbated by the unknown rewards, costs, and task arrival distribution. To address these challenges, we propose the Robbins-Monro-based Value Iteration for Ratio Maximization (RM^2VI) algorithm. Specifically,RM^2VI addresses ``intra-dependence'' through optimistic value iteration and handles ``inter-dependence'' using the Robbins-Monro method. The algorithm has a greedy structure and achieves a sub-linear regret of O(K^(3/4)), establishing the no-regret property (per-task).

We test RM^2VI in two synthetic experiments of sale promotion in E-commerce and machine learning job training in cloud computing. The results show RM^2VI achieves the best reward-to-cost ratio compared with the baselines.

  1. [Abdul Qayyum et al., 2019] Alif Bin Abdul Qayyum, Asi- ful Arefeen, and Celia Shahnaz. Convolutional neural net- work (cnn) based speech-emotion recognition. In 2019 IEEE International Conference on Signal Processing, Information, Communication & Systems (SPICSCON), pages 122-125, 2019.
  2. [Abounadi et al., 2001] Jinane Abounadi, Dimitrib Bert- sekas, and VS Borkar. Learning algorithms for Markov decision processes with average cost. SIAM Journal on Control and Optimization, 40(3):681-698, 2001.
  3. [Agarwal and Jain, 2014] Dr Amit Agarwal and Saloni Jain. Efficient optimal algorithm of task scheduling in cloud computing environment. International Journal of Com- puter Trends and Technology, 9(7):344-349, 2014.
  4. [Agrawal and Devanur, 2014] Shipra Agrawal and Nikhil R. Devanur. Bandits with concave rewards and convex knap- sacks. In Proceedings of the Fifteenth ACM Conference on Economics and Computation, page 989-1006, New York, NY, USA, 2014. Association for Computing Machinery.
  5. [Agrawal and Devanur, 2016] Shipra Agrawal and Nikhil Devanur. Linear contextual bandits with knapsacks. In Advances in Neural Information Processing Systems. Cur- ran Associates, Inc., 2016.
  6. [Alabbadi and Abulkhair, 2021] Afra A Alabbadi and Maysoon F Abulkhair. Multi-objective task scheduling optimization in spatial crowdsourcing. Algorithms, 14(3):77-97, 2021.
  7. [Altman, 1999] Eitan Altman. Constrained Markov Deci- sion Processes. CRC Press, 1999.
  8. [Arunarani et al., 2019] AR Arunarani, Dhanabalachandran Manjula, and Vijayan Sugumaran. Task scheduling tech- niques in cloud computing: A literature survey. Future Generation Computer Systems, 91:407-415, 2019.
  9. [Auer et al., 2002] P. Auer, Paul Fischer, and N. Cesa- Bianchi. Finite-time analysis of the multiarmed bandit problem. Machine Learning, 47(3):235-256, 2002.
  10. [Azar et al., 2017] Mohammad Gheshlaghi Azar, Ian Os- band, and Rémi Munos. Minimax regret bounds for re- inforcement learning. In Proceedings of the 34th Interna- tional Conference on Machine Learning -Volume 70, page 263-272, Sydney, NSW, Australia, 2017. JMLR.org.
  11. [Badanidiyuru et al., 2014] Ashwinkumar Badanidiyuru, John Langford, and Aleksandrs Slivkins. Resourceful contextual bandits. In Proceedings of The 27th Conference on Learning Theory, pages 1109-1134, Barcelona, Spain, June 2014. PMLR. Badanidiyuru et al., 2018] Ashwinkumar Badanidiyuru, Robert Kleinberg, and Aleksandrs Slivkins. Bandits with knapsacks. J. ACM, 65(3), March 2018.
  12. [Bhatnagar, 2010] Shalabh Bhatnagar. An actor-critic al- gorithm with function approximation for discounted cost constrained markov decision processes. Systems & Con- trol Letters, 59(12):760-766, 2010.
  13. [Borkar, 2005] Vivek S Borkar. An actor-critic algorithm for constrained markov decision processes. Systems & control letters, 54(3):207-213, 2005.
  14. [Cayci et al., 2020] Semih Cayci, Atilla Eryilmaz, and R Srikant. Budget-constrained bandits over general cost and reward distributions. In Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics, pages 4388-4398. PMLR, August 2020.
  15. [Cesa-Bianchi et al., 2021] Nicolò Cesa-Bianchi, Tom Ce- sari, Yishay Mansour, and Vianney Perchet. Roi max- imization in stochastic online decision-making. In Ad- vances in Neural Information Processing Systems, pages 9152-9166.
  16. Curran Associates, Inc., 2021.
  17. [Chen et al., 2020] Xuan Chen, Long Cheng, Cong Liu, Qingzhi Liu, Jinwei Liu, Ying Mao, and John Murphy. A woa-based optimization approach for task scheduling in cloud computing systems. IEEE Systems Journal, 14(3):3117-3128, 2020.
  18. [Cho et al., 2015] Keng-Mao Cho, Pang-Wei Tsai, Chun- Wei Tsai, and Chu-Sing Yang. A hybrid meta-heuristic algorithm for vm scheduling with load balancing in cloud computing. Neural Computing and Applications, 26:1297-1309, 2015.
  19. [Deane, 2012] Jason Deane. Hybrid genetic algorithm and augmented neural network application for solving the on- line advertisement scheduling problem with contextual tar- geting. Expert Systems with Applications, 39(5):5168- 5177, 2012.
  20. [Deng et al., 2015] Dingxiong Deng, Cyrus Shahabi, and Linhong Zhu. Task matching and scheduling for multi- ple workers in spatial crowdsourcing. In Proceedings of ACM SIGSPATIAL, 2015.
  21. [Ding et al., 2013] Wenkui Ding, Tao Qin, Xu-Dong Zhang, and Tie-Yan Liu. Multi-armed bandit with budget con- straint and variable costs. In Proceedings of the AAAI Con- ference on Artificial Intelligence, volume 27, pages 232- 238, 2013.
  22. [Guo and Liu, 2024] Hengquan Guo and Xin Liu. Stochastic constrained contextual bandits via lyapunov optimization based estimation to decision framework. In The Thirty Sev- enth Annual Conference on Learning Theory, pages 2204- 2231, 2024.
  23. [Guo et al., 2012] Lizheng Guo, Shuguang Zhao, Shigen Shen, and Changyuan Jiang. Task scheduling optimization in cloud computing based on heuristic algorithm. Journal of Networks, 7(3):547-553, 2012.
  24. [Han et al., 2023] Yuxuan Han, Jialin Zeng, Yang Wang, Yang Xiang, and Jiheng Zhang. Optimal contextual ban- dits with knapsacks under realizability via regression ora- cles. In Proceedings of The 26th International Conference on Artificial Intelligence and Statistics, pages 5011-5035. PMLR, April 2023.
  25. [Held and Karp, 1962] Michael Held and Richard M Karp. A dynamic programming approach to sequencing prob- lems. Journal of the Society for Industrial and Applied mathematics, 10(1):196-210, 1962.
  26. [Heyden et al., 2024] Marco Heyden, Vadim Arzamasov, Edouard Fouché, and Klemens Böhm. Budgeted multi- armed bandits with asymmetric confidence intervals. In Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, page 1073-1084, New York, NY, USA, 2024. Association for Computing Machinery.
  27. [Jin et al., 2018] Chi Jin, Zeyuan Allen-Zhu, Sebastien Bubeck, and Michael I. Jordan. Is Q-learning provably efficient? In Proceedings of the 32nd International Con- ference on Neural Information Processing Systems, page 4868-4878, Red Hook, NY, USA, 2018.
  28. Curran Asso- ciates Inc. [Kanuri et al., 2018]
  29. Vamsi K Kanuri, Yixing Chen, and Shrihari Sridhar. Scheduling content on social media: Theory, evidence, and application. Journal of Marketing, 82(6):89-108, 2018.
  30. [Khazankin et al., 2011] Roman Khazankin, Harald Psaier, Daniel Schall, and Schahram Dustdar. Qos-based task scheduling in crowdsourcing environments. In Proceed- ings of ICSOC, 2011.
  31. [Martineau et al., 2018] Maxime Martineau, Romain Raveaux, Clément Chatelain, Donatello Conte, and Gilles Venturini. Effective training of convolutional neural networks for insect image recognition. In Advanced Con- cepts for Intelligent Vision Systems: 19th International Conference, ACIVS 2018, Poitiers, France, September 24-27, 2018, Proceedings 19, pages 426-437. Springer, 2018.
  32. [Neely, 2010] Michael J Neely. Stochastic network opti- mization with application to communication and queueing systems. Synthesis Lectures on Communication Networks, 3(1):1-211, 2010.
  33. [Neely, 2021] Michael J Neely. Fast learning for renewal op- timization in online task scheduling. Journal of Machine Learning Research, 22(1):12785-12828, 2021.
  34. [Neely, 2024] Michael J Neely. Adaptive optimiza- tion for stochastic renewal systems. arXiv preprint arXiv:2401.07170, 2024.
  35. [REN and KROGH, 2005] ZHIYUAN REN and Bruce H KROGH. Markov decision processes with fractional costs. IEEE transactions on automatic control, 50(5):646-650, 2005.
  36. [Slivkins et al., 2023] Aleksandrs Slivkins, Karthik Abinav Sankararaman, and Dylan J Foster. Contextual bandits with packing and covering constraints: A modular la- grangian approach via regression. In Proceedings of Thirty Sixth Conference on Learning Theory, pages 4633-4656. PMLR, 12-15 Jul 2023.
  37. [Srikant and Ying, 2014] Rayadurgam Srikant and Lei Ying. Communication Networks: An Optimization, Control, and Stochastic Networks Perspective. Cambridge University Press, 2014.
  38. [Suttle et al., 2021] Wesley Suttle, Kaiqing Zhang, Zhuoran Yang, Ji Liu, and David Kraemer. Reinforcement Learn- ing for Cost-Aware Markov Decision Processes. In Pro- ceedings of the 38th International Conference on Machine Learning, pages 9989-9999. PMLR, 2021.
  39. [Tanaka, 2017] Teruo Tanaka. A partially observable dis- crete time markov decision process with a fractional dis- counted reward. Journal of Information and Optimization Sciences, 38(1):21-37, 2017.
  40. [Tran-Thanh et al., 2012] Long Tran-Thanh, Archie Chap- man, Alex Rogers, and Nicholas Jennings. Knapsack based optimal policies for budget-limited multi-armed bandits. In Proceedings of the AAAI Conference on Ar- tificial Intelligence, volume 26, pages 1134-1140, 2012.
  41. [Wachi et al., 2024] Akifumi Wachi, Xun Shen, and Yanan Sui. A survey of constraint formulations in safe reinforce- ment learning. arXiv preprint arXiv:2402.02025, 2024. IJCAI
  42. [Watanabe et al., 2018] Ryo Watanabe, Junpei Komiyama, Atsuyoshi Nakamura, and Mineichi Kudo. Ucb-sc: A fast variant of kl-ucb-sc for budgeted multi-armed bandit prob- lem. IEICE Transactions on Fundamentals of Electronics, Communications and Computer Sciences, 101(3):662- 667, 2018.
  43. [Wei et al., 2022] Honghao Wei, Xin Liu, and Lei Ying. Triple-q: A model-free algorithm for constrained rein- forcement learning with sublinear regret and zero con- straint violation. In Proceedings of The 25th International Conference on Artificial Intelligence and Statistics, pages 3274-3307. PMLR, 2022.
  44. [Xia et al., 2017] Yingce Xia, Tao Qin, Wenkui Ding, Haifang Li, Xudong Zhang, Nenghai Yu, and Tie-Yan Liu. Finite budget analysis of multi-armed bandit problems. Neurocomputing, 258:13-29, 2017.
  45. [Xu et al., 2024] Yongxin Xu, Shangshang Wang, Hengquan Guo, Xin Liu, and Ziyu Shao. Learning to Schedule Online Tasks with Bandit Feedback. In Proceedings of the 23rd International Conference on Autonomous Agents and Multiagent Systems, page 1975-1983, Richland, SC, 2024. International Foundation for Autonomous Agents and Multiagent Systems.
  46. [Zhang et al., 2024] Juntao Zhang, Juan Ren, and Shuiqing Hu. Afm imaging defect detection and classification using deep learning. IEEE Access, 12:132027-132037, 2024.
  47. [Zhao et al., 2022] Zihui Zhao, Xiaoyu Shi, and Mingsheng Shang. Performance and cost-aware task scheduling via deep reinforcement learning in cloud environment. In Service-Oriented Computing, pages 600-615, Cham, 2022. Springer Nature Switzerland.

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

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