Optimal and Adaptive Off-policy Evaluation in Contextual Bandits

By Yu-Xiang Wang, Alekh Agarwal, Miroslav Dud'ik. 2016

239 citations recorded by the source corpus.

We study the off-policy evaluation problem---estimating the value of a target policy using data collected by another policy---under the contextual bandit model. We consider the general (agnostic) setting without access to a consistent model of rewards and establish a minimax lower bound on the mean squared error (MSE). The bound is matched up to constants by the inverse propensity scoring (IPS) and doubly robust (DR) estimators. This highlights the difficulty of the agnostic contextual setting, in contrast with multi-armed bandits and contextual bandits with access to a consistent reward model, where IPS is suboptimal. We then propose the SWITCH estimator, which can use an existing reward model (not necessarily consistent) to achieve a better bias-variance tradeoff than IPS and DR. We prove an upper bound on its MSE and demonstrate its benefits empirically on a diverse collection of data sets, often outperforming prior work by orders of magnitude.

Subfields: Artificial Intelligence, Management Science and Operations Research, Statistics and Probability, Reinforcement Learning, Statistical Learning Theory

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Sergey Levine
  2. Ofir Nachum
  3. George Tucker
  4. Pieter Abbeel
  5. Shixiang Shane Gu
  6. Nan Jiang
  7. Nathan Kallus
  8. Chelsea Finn

In Management Science and Operations Research

  1. Nathan Kallus
  2. Masatoshi Uehara
  3. Sergey Levine
  4. Ofir Nachum
  5. Nan Jiang
  6. Susan Athey
  7. George Tucker
  8. Lihong Li

In Statistics and Probability

  1. Nathan Kallus
  2. Masatoshi Uehara
  3. Susan Athey
  4. Nan Jiang
  5. Lihong Li
  6. Ziyang Tang
  7. Stefan Wager
  8. Guido W. Imbens

In Reinforcement Learning

  1. Sergey Levine
  2. Ofir Nachum
  3. Nan Jiang
  4. George Tucker
  5. Aviral Kumar
  6. Shixiang Shane Gu
  7. Nathan Kallus
  8. Masatoshi Uehara

In Statistical Learning Theory

  1. Nan Jiang
  2. Alekh Agarwal
  3. Tengyang Xie
  4. Nathan Kallus
  5. Lihong Li
  6. Masatoshi Uehara
  7. Yifei Ma
  8. Ziyang Tang

Citations in the corpus