Planning Multiple Epidemic Interventions with Reinforcement Learning

Planning Multiple Epidemic Interventions with Reinforcement Learning

Anh Mai, Nikunj Gupta, Azza Abouzied, Dennis Shasha

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence
AI for Good. Pages 6147-6155. https://doi.org/10.24963/ijcai.2023/682

Combating an epidemic entails finding a plan that describes when and how to apply different interventions, such as mask-wearing mandates, vaccinations, school or workplace closures. An optimal plan will curb an epidemic with minimal loss of life, disease burden, and economic cost. Finding an optimal plan is an intractable computational problem in realistic settings. Policy-makers, however, would greatly benefit from tools that can efficiently search for plans that minimize disease and economic costs especially when considering multiple possible interventions over a continuous and complex action space given a continuous and equally complex state space. We formulate this problem as a Markov decision process. Our formulation is unique in its ability to represent multiple continuous interventions over any disease model defined by ordinary differential equations. We illustrate how to effectively apply state-of-the-art actor-critic reinforcement learning algorithms (PPO and SAC) to search for plans that minimize overall costs. We empirically evaluate the learning performance of these algorithms and compare their performance to hand-crafted baselines that mimic plans constructed by policy-makers. Our method outperforms baselines. Our work confirms the viability of a computational approach to support policy-makers.
Keywords:
AI for Good: Machine Learning