English

Evaluating and Learning Robust Bandit Policies Under Uncertain Causal Mechanisms

Machine Learning 2026-04-07 v2

Abstract

Causal graphical models can encode large amounts structural knowledge, both from the background knowledge of domain experts and the structural knowledge discovered from randomized experiments or observational data. However, though we may know the general structure of causal relationships, we often do not know the exact causal mechanisms. In this work, we propose a causal multi-armed bandit evaluation and learning algorithm that can reason effectively despite uncertainty over conditional probability distributions. Further, we show how conditional independence testing can be used to choose variables for modeling. We find that the structural equation model (SEM) approach gives more accurate evaluations compared to traditional approaches, particularly as the range of possible causal mechanisms grows. Further, the SEM approach learns low-variance policies, and it learns an optimal policy, assuming the model is sufficiently well-specified. Traditional approaches can converge to local extrema or fail to converge at all.

Keywords

Cite

@article{arxiv.2508.02812,
  title  = {Evaluating and Learning Robust Bandit Policies Under Uncertain Causal Mechanisms},
  author = {Katherine Avery and Chinmay Pendse and David Jensen},
  journal= {arXiv preprint arXiv:2508.02812},
  year   = {2026}
}

Comments

13 pages main text, 34 pages total

R2 v1 2026-07-01T04:34:03.553Z