Proceedings

Optimizing Resource-Constrained Non-Pharmaceutical Interventions for Multi-Cluster Outbreak Control Using Hierarchical Reinforcement Learning

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. AI and Health. Pages 6850-6858.

https://doi.org/10.24963/ijcai.2026/762

Abstract

Non-pharmaceutical interventions (NPIs), such as diagnostic testing and quarantine, are crucial for controlling infectious disease outbreaks but are often constrained by limited resources, particularly in the early outbreak stages. In real-world public health settings, resources must be allocated across multiple outbreak clusters that emerge asynchronously, vary in size and risk, and compete for a shared resource budget. We define a cluster as a group of close contacts generated by a single infected index case. Thus, decisions must be made under uncertainty and heterogeneous demands while respecting operational constraints. We formulate this problem as a constrained restless multi-armed bandit and propose a hierarchical reinforcement learning framework. A global controller learns a continuous action cost multiplier that adjusts global resource demand, while a generalized local policy estimates the marginal value of allocating resources to individuals within each cluster. We evaluate the proposed framework in a realistic agent-based simulator of SARS-CoV-2 with dynamically arriving clusters. Across a wide range of system scales and testing budgets, our method consistently outperforms RMAB-inspired and heuristic baselines, improving outbreak control effectiveness by 20--30%. Experiments on up to 40 concurrently active clusters further demonstrate that the hierarchical framework is highly scalable and enables faster decision-making than the RMAB-inspired method.

  1. [Alamo et al., 2021] Teodoro Alamo, Daniel G Reina, Pablo Millán Gata, Victor M Preciado, and Giulia Gior- dano. Data-driven methods for present and future pan- demics: Monitoring, modelling and managing. Annual Reviews in Control, 52:448-464, 2021.
  2. [Amann et al., 2020] Julia Amann, Alessandro Blasimme, Effy Vayena, Dietmar Frey, Vince I Madai, and Precise4Q Consortium. Explainability for artificial intelligence in healthcare: a multidisciplinary perspective. BMC medical informatics and decision making, 20(1):310, 2020.
  3. [Argyle et al., 2023] Lisa P Argyle, Ethan C Busby, Nancy Fulda, Joshua R Gubler, Christopher Rytting, and David Wingate. Out of one, many: Using language models to simulate human samples. Political Analysis, 31(3):337- 351, 2023.
  4. [Avrachenkov and Borkar, 2022] Konstantin E Avrachenkov and Vivek S Borkar. Whittle index based q-learning for restless bandits with average reward. Automatica, 139:110186, 2022.
  5. [Barto and Mahadevan, 2003] Andrew G Barto and Srid- har Mahadevan. Recent advances in hierarchical re- inforcement learning. Discrete event dynamic systems, 13(4):341-379, 2003.
  6. [Bertsekas, 1997] Dimitri P Bertsekas. Nonlinear Program- ming. Athena Scientific, 1997.
  7. [Biswas et al., 2021] Arpita Biswas, Gaurav Aggarwal, Pradeep Varakantham, and Milind Tambe. Learn to in- tervene: An adaptive learning policy for restless bandits in application to preventive healthcare. arXiv preprint arXiv:2105.07965, 2021. IJCAI
  8. [Brauer et al., 2017] Fred Brauer, Carlos Castillo-Chavez, and Zhilan Feng. Mathematical epidemiology. Springer, 2017.
  9. [Funk et al., 2010] Sebastian Funk, Marcel Salathé, and Vin- cent AA Jansen. Modelling the influence of human be- haviour on the spread of infectious diseases: a review. Journal of the Royal Society Interface, 7(50):1247-1256, 2010.
  10. [Glazebrook et al., 2006] Kevin D Glazebrook, Diego Ruiz- Hernandez, and Christopher Kirkbride. Some indexable families of restless bandit problems. Advances in Applied Probability, 38(3):643-672, 2006.
  11. [Holzinger et al., 2019] Andreas Holzinger, Georg Langs, Helmut Denk, Kurt Zatloukal, and Heimo Müller. Caus- ability and explainability of artificial intelligence in medicine. Wiley interdisciplinary reviews: data mining and knowledge discovery, 9(4):e1312, 2019.
  12. [Killian et al., 2021] Jackson A Killian, Andrew Perrault, and Milind Tambe. Beyond" to act or not to act": Fast lagrangian approaches to general multi-action restless ban- dits. In Proceedings of the 20th International Conference on Autonomous Agents and MultiAgent Systems, pages 710-718, 2021.
  13. [Kompella et al., 2020] Varun Kompella, Roberto Capo- bianco, Stacy Jong, Jonathan Browne, Spencer Fox, Lau- ren Meyers, Peter Wurman, and Peter Stone. Reinforce- ment learning for optimization of covid-19 mitigation poli- cies. arXiv preprint arXiv:2010.10560, 2020.
  14. [Liu and Zhao, 2010] Keqin Liu and Qing Zhao. Indexabil- ity of restless bandit problems and optimality of whittle in- dex for dynamic multichannel access. IEEE Transactions on Information Theory, 56(11):5547-5567, 2010.
  15. [Mao and Perrault, 2026] Yi Mao and Andrew Perrault. Op- timizing urban service allocation with time-constrained restless bandits. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 40, pages 39025-39032, 2026.
  16. [Mate et al., 2020] Aditya Mate, Jackson Killian, Haifeng Xu, Andrew Perrault, and Milind Tambe. Collapsing bandits and their application to public health interven- tion. Advances in Neural Information Processing Systems, 33:15639-15650, 2020.
  17. [Mnih et al., 2015] Volodymyr Mnih, Koray Kavukcuoglu, David Silver, Andrei A Rusu, Joel Veness, Marc G Belle- mare, Alex Graves, Martin Riedmiller, Andreas K Fidje- land, Georg Ostrovski, et al. Human-level control through deep reinforcement learning. nature, 518(7540):529-533, 2015.
  18. [Nachum et al., 2018] Ofir Nachum, Shixiang Shane Gu, Honglak Lee, and Sergey Levine. Data-efficient hierar- chical reinforcement learning. Advances in neural infor- mation processing systems, 31, 2018.
  19. [Nakhleh et al., 2021] Khaled Nakhleh, Santosh Ganji, Ping-Chun Hsieh, I-Hong Hou, and Srinivas Shakkottai. Neurwin: Neural whittle index network for restless bandits via deep rl. In M. Ranzato, A. Beygelzimer, Y. Dauphin, P.S. Liang, and J. Wortman Vaughan, editors, Advances in Neural Information Processing Systems, volume 34, pages 828-839.
  20. Curran Associates, Inc., 2021.
  21. [Neely, 2010] Michael J Neely. Stochastic network op- timization with application to resource allocation and queueing. Synthesis Lectures on Communication Net- works, 3:1-211, 2010.
  22. [Nowzari et al., 2016] Cameron Nowzari, Victor M Preci- ado, and George J Pappas. Analysis and control of epi- demics: A survey of spreading processes on complex net- works. IEEE Control Systems Magazine, 36(1):26-46, 2016.
  23. [Park et al., 2023] Joon Sung Park, Joseph O'Brien, Car- rie Jun Cai, Meredith Ringel Morris, Percy Liang, and Michael S Bernstein. Generative agents: Interactive sim- ulacra of human behavior. In Proceedings of the 36th an- nual acm symposium on user interface software and tech- nology, pages 1-22, 2023.
  24. [Peng et al., 2023] Xueqiao Peng, Jiaqi Xu, Xi Chen, Dinh Song An Nguyen, and Andrew Perrault. Using rein- forcement learning for multi-objective cluster-level opti- mization of non-pharmaceutical interventions for infec- tious disease. In Machine Learning for Health (ML4H), pages 445-460.
  25. PMLR, 2023.
  26. [Probert et al., 2016] William JM Probert, Katriona Shea, Christopher J Fonnesbeck, Michael C Runge, Tim E Car- penter, Salome Dürr, M Graeme Garner, Neil Harvey, Mark A Stevenson, and Colleen T Webb. Decision-making for foot-and-mouth disease control: objectives matter. Epi- demics, 15:10-19, 2016.
  27. [Rahmandad et al., 2021] Hazhir Rahmandad, Tse Yang Lim, and John Sterman. Behavioral dynamics of covid- 19: estimating underreporting, multiple waves, and adher- ence fatigue across 92 nations. System dynamics review, 37(1):5-31, 2021.
  28. [Schulman et al., 2017] John Schulman, Filip Wolski, Pra- fulla Dhariwal, Alec Radford, and Oleg Klimov. Prox- imal policy optimization algorithms. arXiv preprint arXiv:1707.06347, 2017.
  29. [Vaswani et al., 2017] Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N Gomez, Łukasz Kaiser, and Illia Polosukhin. Attention is all you need. Advances in neural information processing systems, 30, 2017.
  30. [Verelst et al., 2016] Frederik Verelst, Lander Willem, and Philippe Beutels. Behavioural change models for infectious disease transmission: a systematic review (2010-2015). Journal of The Royal Society Interface, 13(125):20160820, 2016.
  31. [Vezhnevets et al., 2017] Alexander Sasha Vezhnevets, Si- mon Osindero, Tom Schaul, Nicolas Heess, Max Jader- berg, David Silver, and Koray Kavukcuoglu. Feudal net- works for hierarchical reinforcement learning. In Interna- tional conference on machine learning, pages 3540-3549.
  32. PMLR, 2017.
  33. [Weber and Weiss, 1990] Richard R Weber and Gideon Weiss. On an index policy for restless bandits. Journal of applied probability, 27(3):637-648, 1990.
  34. [Whittle, 1988] Peter Whittle. Restless bandits: Activity al- location in a changing world. Journal of applied probabil- ity, 25(A):287-298, 1988.
  35. [Yang et al., 2018] Yaodong Yang, Rui Luo, Minne Li, Ming Zhou, Weinan Zhang, and Jun Wang. Mean field multi- agent reinforcement learning. In International conference on machine learning, pages 5571-5580. PMLR, 2018.

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

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