Reinforcement Learning for Finite-Horizon Restless Multi-Armed Multi-Action Bandits

Abstract

We study a finite-horizon restless multi-armed bandit problem with multiple actions, dubbed R(MA)^2B. The state of each arm evolves according to a controlled Markov decision process (MDP), and the reward of pulling an arm depends on both the current state of the corresponding MDP and the action taken. The goal is to sequentially choose actions for arms so as to maximize the expected value of the cumulative rewards collected. Since finding the optimal policy is typically intractable, we propose a computationally appealing index policy which we call Occupancy-Measured-Reward Index Policy. Our policy is well-defined even if the underlying MDPs are not indexable. We prove that it is asymptotically optimal when the activation budget and number of arms are scaled up, while keeping their ratio as a constant. For the case when the system parameters are unknown, we develop a learning algorithm. Our learning algorithm uses the principle of optimism in the face of uncertainty and further uses a generative model in order to fully exploit the structure of Occupancy-Measured-Reward Index Policy. We call it the R(MA)^2B-UCB algorithm. As compared with the existing algorithms, R(MA)^2B-UCB performs close to an offline optimum policy, and also achieves a sub-linear regret with a low computational complexity. Experimental results show that R(MA)^2B-UCB outperforms the existing algorithms in both regret and run time.

Appeared in AAAI 2022 with title "Reinforcement Learning Augmented Asymptotically Optimal Index Policy for Finite-Horizon Restless Bandits"

References in corpus (11)

• Empirical Bernstein Bounds and Sample Variance Penalization
• Dynamic priority allocation via restless bandit marginal productivity indices
• Asymptotically optimal priority policies for indexable and nonindexable restless bandits
• Online Convex Optimization in Adversarial Markov Decision Processes
• Regret Bounds for Restless Markov Bandits
• Exploration-Exploitation in Constrained MDPs
• A Sample-Efficient Algorithm for Episodic Finite-Horizon MDP with Constraints
• Restless Bandits with Many Arms: Beating the Central Limit Theorem
• Learning Augmented Index Policy for Optimal Service Placement at the Network Edge
• An Asymptotically Optimal Index Policy for Finite-Horizon Restless Bandits
• Thompson Sampling in Non-Episodic Restless Bandits