Planning and Learning in Risk-Aware Restless Multi-Arm Bandits

Abstract

In restless multi-arm bandits, a central agent is tasked with optimally distributing limited resources across several bandits (arms), with each arm being a Markov decision process. In this work, we generalize the traditional restless multi-arm bandit problem with a risk-neutral objective by incorporating risk-awareness. We establish indexability conditions for the case of a risk-aware objective and provide a solution based on Whittle index. In addition, we address the learning problem when the true transition probabilities are unknown by proposing a Thompson sampling approach and show that it achieves bounded regret that scales sublinearly with the number of episodes and quadratically with the number of arms. The efficacy of our method in reducing risk exposure in restless multi-arm bandits is illustrated through a set of numerical experiments in the contexts of machine replacement and patient scheduling applications under both planning and learning setups.

Cite

Text

Akbarzadeh et al. "Planning and Learning in Risk-Aware Restless Multi-Arm Bandits." Proceedings of The 28th International Conference on Artificial Intelligence and Statistics, 2025.

Markdown

[Akbarzadeh et al. "Planning and Learning in Risk-Aware Restless Multi-Arm Bandits." Proceedings of The 28th International Conference on Artificial Intelligence and Statistics, 2025.](https://mlanthology.org/aistats/2025/akbarzadeh2025aistats-planning/)

BibTeX

@inproceedings{akbarzadeh2025aistats-planning,
  title     = {{Planning and Learning in Risk-Aware Restless Multi-Arm Bandits}},
  author    = {Akbarzadeh, Nima and Adulyasak, Yossiri and Delage, Erick},
  booktitle = {Proceedings of The 28th International Conference on Artificial Intelligence and Statistics},
  year      = {2025},
  pages     = {991-999},
  volume    = {258},
  url       = {https://mlanthology.org/aistats/2025/akbarzadeh2025aistats-planning/}
}