Improved Variance-Aware Confidence Sets for Linear Bandits and Linear Mixture MDP

Zihan Zhang,Jiaqi Yang,Xiangyang Ji,Simon S. Du
2021-01-01
Abstract:This paper presents new variance-aware confidence sets for linear bandits and linear mixture Markov Decision Processes (MDPs). With the new confidence sets, we obtain the follow regret bounds: For linear bandits, we obtain an (O) over tilde (poly(d)root 1 Sigma(K)(k=1) sigma(2)(k)) data-dependent regret bound, where d is the feature dimension, K is the number of rounds, and sigma(2)(k) is the unknown variance of the reward at the k-th round. This is the first regret bound that only scales with the variance and the dimension but no explicit polynomial dependency on K. When variances are small, this bound can be significantly smaller than the (Theta) over tilde (d root K) worst-case regret bound. For linear mixture MDPs, we obtain an (O) over tilde (poly(d,log H root K) regret bound, where d is the number of base models, K is the number of episodes, and H is the planning horizon. This is the first regret bound that only scales logarithmically with H in the reinforcement learning with linear function approximation setting, thus exponentially improving existing results, and resolving an open problem in [Zhou et al., 2020a]. We develop three technical ideas that may be of independent interest: 1) applications of the peeling technique to both the input norm and the variance magnitude, 2) a recursion-based estimator for the variance, and 3) a new convex potential lemma that generalizes the seminal elliptical potential lemma.
What problem does this paper attempt to address?