Active Exploration via Autoregressive Generation of Missing Data

Tiffany Tianhui Cai, Hongseok Namkoong, Daniel Russo, Kelly W Zhang
5/29/2024
cs.LGstat.ML

No official code found. Best implementation: namkoong-lab/ps-autoregressive-gen (2 stars, last commit 2025-03-26).

Abstract

We pose uncertainty quantification and exploration in online decision-making as a problem of training and generation from an autoregressive sequence model, an area experiencing rapid innovation. Our approach rests on viewing uncertainty as arising from missing future outcomes that would be revealed through appropriate action choices, rather than from unobservable latent parameters of the environment. This reformulation aligns naturally with modern machine learning capabilities: we can i) train generative models through next-outcome prediction rather than fit explicit priors, ii) assess uncertainty through autoregressive generation rather than parameter sampling, and iii) adapt to new information through in-context learning rather than explicit posterior updating. To showcase these ideas, we formulate a challenging meta-bandit problem where effective performance requires leveraging unstructured prior information (like text features) while exploring judiciously to resolve key remaining uncertainties. We validate our approach through both theory and experiments. Our theory establishes a reduction, showing success at offline next-outcome prediction translates to reliable online uncertainty quantification and decision-making, even with strategically collected data. Semi-synthetic experiments show our insights bear out in a news-article recommendation task, where article text can be leveraged to minimize exploration.

View on arXivView PDF

Code Implementations(1)

Code for Active Exploration via Autoregressive Generation of Missing Data

20Shell, PythonMar 21, 20251 years ago

Similar Papers

Cite this paper

@article{cai2024active,
  title  = {Active Exploration via Autoregressive Generation of Missing Data},
  author = {Tiffany Tianhui Cai and Hongseok Namkoong and Daniel Russo and Kelly W Zhang},
  year   = {2024},
  eprint = {2405.19466},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.19466v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.19466.svg)](https://codeofpaper.com/paper/2405.19466)

Discussion