arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.03463cs.LGcs.SYeess.SY

学习具有可扩展遗憾的不安分老虎机 Whittle 索引策略

On learning Whittle index policy for restless bandits with scalable regret

McGill University(麦吉尔大学)

Nima Akbarzadeh, Aditya Mahajan

AI总结:

针对不安分老虎机资源调度问题,提出贴合模型结构的基于 Thompson 采样的模型驱动 RL 算法,并证明其相对 Whittle 索引策略具有可扩展遗憾。

摘要:

当系统模型未知时,强化学习是一种颇具吸引力的方法,可基于数据学习良好的资源分配与调度策略。然而,大多数强化学习算法的累积遗憾规模为 $\tilde O(\mathsf{S} \sqrt{\mathsf{A} T})$,其中 $\mathsf{S}$ 表示状态空间大小,$\mathsf{A}$ 表示动作空间大小,$T$ 表示时域长度,而 $\tilde{O}(\cdot)$ 记号隐藏了对数项。由于对状态空间大小存在线性依赖,这些遗憾界对于资源分配和调度问题而言大得难以接受。本文针对此类问题提出了一种基于模型的强化学习算法,该算法具有可扩展的遗憾。具体而言,我们考虑不安分老虎机模型,并提出一种基于 Thompson 采样的学习算法,该算法针对模型的底层结构进行了调整。我们相对于 Whittle 索引策略,对所提算法的遗憾给出了两种刻画。首先,我们证明,对于具有 $n$ 条臂且每个时刻至多激活 $m$ 条臂的不安分老虎机,取决于奖励模型,其遗憾规模为 $\tilde{O}(mn\sqrt{T})$ 或 $\tilde{O}(n^2 \sqrt{T})$。其次,在一个附加技术假设下,我们证明遗憾规模为 $\tilde{O}(n^{1.5} \sqrt{T})$ 或 $\tilde{O}(\max\{m\sqrt{n}, n\} \sqrt{T})$。我们给出数值示例以说明该算法的显著特征。

英文摘要:

Reinforcement learning is an attractive approach to learn good resource allocation and scheduling policies based on data when the system model is unknown. However, the cumulative regret of most RL algorithms scales as $\tilde O(\mathsf{S} \sqrt{\mathsf{A} T})$, where $\mathsf{S}$ is the size of the state space, $\mathsf{A}$ is the size of the action space, $T$ is the horizon, and the $\tilde{O}(\cdot)$ notation hides logarithmic terms. Due to the linear dependence on the size of the state space, these regret bounds are prohibitively large for resource allocation and scheduling problems. In this paper, we present a model-based RL algorithm for such problems which has scalable regret. In particular, we consider a restless bandit model, and propose a Thompson-sampling based learning algorithm which is tuned to the underlying structure of the model. We present two characterizations of the regret of the proposed algorithm with respect to the Whittle index policy. First, we show that for a restless bandit with $n$ arms and at most $m$ activations at each time, the regret scales either as $\tilde{O}(mn\sqrt{T})$ or $\tilde{O}(n^2 \sqrt{T})$ depending on the reward model. Second, under an additional technical assumption, we show that the regret scales as $\tilde{O}(n^{1.5} \sqrt{T})$ or $\tilde{O}(\max\{m\sqrt{n}, n\} \sqrt{T})$. We present numerical examples to illustrate the salient features of the algorithm.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑