Saved in:
Bibliographic Details
Main Authors: Halder, Budhaditya, Pan, Shubhayan, Khamaru, Koulik
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2505.23260
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912966575652864
author Halder, Budhaditya
Pan, Shubhayan
Khamaru, Koulik
author_facet Halder, Budhaditya
Pan, Shubhayan
Khamaru, Koulik
contents We consider the problem of statistical inference when the data is collected via a Thompson Sampling-type algorithm. While Thompson Sampling (TS) is known to be both asymptotically optimal and empirically effective, its adaptive sampling scheme poses challenges for constructing confidence intervals for model parameters. We propose and analyze a variant of TS, called Stable Thompson Sampling, in which the posterior variance is inflated by a logarithmic factor. We show that this modification leads to asymptotically normal estimates of the arm means, despite the non-i.i.d. nature of the data. Importantly, this statistical benefit comes at a modest cost: the variance inflation increases regret by only a logarithmic factor compared to standard TS. Our results reveal a principled trade-off: by paying a small price in regret, one can enable valid statistical inference for adaptive decision-making algorithms.
format Preprint
id arxiv_https___arxiv_org_abs_2505_23260
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Stable Thompson Sampling: Valid Inference via Variance Inflation
Halder, Budhaditya
Pan, Shubhayan
Khamaru, Koulik
Machine Learning
We consider the problem of statistical inference when the data is collected via a Thompson Sampling-type algorithm. While Thompson Sampling (TS) is known to be both asymptotically optimal and empirically effective, its adaptive sampling scheme poses challenges for constructing confidence intervals for model parameters. We propose and analyze a variant of TS, called Stable Thompson Sampling, in which the posterior variance is inflated by a logarithmic factor. We show that this modification leads to asymptotically normal estimates of the arm means, despite the non-i.i.d. nature of the data. Importantly, this statistical benefit comes at a modest cost: the variance inflation increases regret by only a logarithmic factor compared to standard TS. Our results reveal a principled trade-off: by paying a small price in regret, one can enable valid statistical inference for adaptive decision-making algorithms.
title Stable Thompson Sampling: Valid Inference via Variance Inflation
topic Machine Learning
url https://arxiv.org/abs/2505.23260