Adaptive Algorithm for Multi-Armed Bandit Problem with High-Dimensional Covariates
Adaptive Algorithm for Multi-Armed Bandit Problem with High-Dimensional Covariates
复制标题
高维协变量多臂老虎机问题的自适应算法
DOI:
10.1080/01621459.2022.2152343
复制
发表时间:
2023
影响因子:
3.7
通讯作者:
Liu, Ji
中科院分区:
文献类型:
--
作者:
Qian, Wei;Ing, Ching-Kang;Liu, Ji
This article studies an important sequential decision making problem known as the multi-armed stochastic bandit problem with covariates. Under a linear bandit framework with high-dimensional covariates, we propose a general multi-stage arm allocation algorithm that integrates both arm elimination and randomized assignment strategies. By employing a class of high-dimensional regression methods for coefficient estimation, the proposed algorithm is shown to have near optimal finite-time regret performance under a new study scope that requires neither a margin condition nor a reward gap condition for competitive arms. Based on the synergistically verified benefit of the margin, our algorithm exhibits adaptive performance that automatically adapts to the margin and gap conditions, and attains optimal regret rates simultaneously for both study scopes, without or with the margin, up to a logarithmic factor. Besides the desirable regret performance, the proposed algorithm simultaneously generates useful coefficient estimation output for competitive arms and is shown to achieve both estimation consistency and variable selection consistency. Promising empirical performance is demonstrated through extensive simulation and two real data evaluation examples. Supplementary materials for this article are available online.