GINO-Q: Learning an Asymptotically Optimal Index Policy for Restless Multi-armed Bandits
This addresses scalability issues in RMABs for applications like resource allocation, offering a more flexible solution than prior index-based methods.
The paper tackles the problem of solving restless multi-armed bandits (RMABs) with large state and action spaces by proposing GINO-Q, a three-timescale stochastic approximation algorithm that learns an asymptotically optimal index policy. It demonstrates that GINO-Q learns near-optimal policies for non-indexable RMABs where other methods fail and converges faster than baselines.
The restless multi-armed bandit (RMAB) framework is a popular model with applications across a wide variety of fields. However, its solution is hindered by the exponentially growing state space (with respect to the number of arms) and the combinatorial action space, making traditional reinforcement learning methods infeasible for large-scale instances. In this paper, we propose GINO-Q, a three-timescale stochastic approximation algorithm designed to learn an asymptotically optimal index policy for RMABs. GINO-Q mitigates the curse of dimensionality by decomposing the RMAB into a series of subproblems, each with the same dimension as a single arm, ensuring that complexity increases linearly with the number of arms. Unlike recently developed Whittle-index-based algorithms, GINO-Q does not require RMABs to be indexable, enhancing its flexibility and applicability. Our experimental results demonstrate that GINO-Q consistently learns near-optimal policies, even for non-indexable RMABs where Whittle-index-based algorithms perform poorly, and it converges significantly faster than existing baselines.