arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.01780cs.LGstat.ML

Neural Contextual Bandits with Deep Representation and Shallow Exploration

University of California, Los Angeles(加州大学洛杉矶分校);DeepMind(深度思考(DeepMind));Adobe Research(Adobe研究院)

Pan Xu, Zheng Wen, Handong Zhao, Quanquan Gu

英文摘要:

We study a general class of contextual bandits, where each context-action pair is associated with a raw feature vector, but the reward generating function is unknown. We propose a novel learning algorithm that transforms the raw feature vector using the last hidden layer of a deep ReLU neural network (deep representation learning), and uses an upper confidence bound (UCB) approach to explore in the last linear layer (shallow exploration). We prove that under standard assumptions, our proposed algorithm achieves $\tilde{O}(\sqrt{T})$ finite-time regret, where $T$ is the learning time horizon. Compared with existing neural contextual bandit algorithms, our approach is computationally much more efficient since it only needs to explore in the last layer of the deep neural network.

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

↑