472

Stochastic Rank-1 Bandits

Abstract

We propose stochastic rank-11 bandits, a class of online learning problems where at each step a learning agent chooses a pair of row and column arms, and receives the product of their values as a reward. The challenge is that the values of the row and column are unobserved. These values are stochastic and drawn independently of each other. We propose an efficient algorithm for solving our problem, Rank1Elim, and derive a O((K+L)(1/Δ)logn)O((K + L) (1 / \Delta) \log n) upper bound on its nn-step regret, where KK is the number of rows, LL is the number of columns, and Δ\Delta is the minimum of the row and column gaps. This is the first bandit algorithm for finding the maximum entry of a rank-11 matrix whose regret is linear in K+LK + L, 1/Δ1 / \Delta, and logn\log n. We evaluate our proposed algorithm on both synthetic and real-world problems, and observe that it leverages the structure of our problems and can learn near-optimal solutions even when our modeling assumptions are mildly violated.

View on arXiv
Comments on this paper