Additive Causal Bandits with Unknown Graph

Alan Malek, Virginia Aglietti, Silvia Chiappa
6/13/2023
cs.LGstat.ML

No official code found. Best implementation: google-deepmind/additive_cbug (5 stars, Apache-2.0, last commit 2023-06-06).

Abstract

We explore algorithms to select actions in the causal bandit setting where the learner can choose to intervene on a set of random variables related by a causal graph, and the learner sequentially chooses interventions and observes a sample from the interventional distribution. The learner's goal is to quickly find the intervention, among all interventions on observable variables, that maximizes the expectation of an outcome variable. We depart from previous literature by assuming no knowledge of the causal graph except that latent confounders between the outcome and its ancestors are not present. We first show that the unknown graph problem can be exponentially hard in the parents of the outcome. To remedy this, we adopt an additional additive assumption on the outcome which allows us to solve the problem by casting it as an additive combinatorial linear bandit problem with full-bandit feedback. We propose a novel action-elimination algorithm for this setting, show how to apply this algorithm to the causal bandit problem, provide sample complexity bounds, and empirically validate our findings on a suite of randomly generated causal models, effectively showing that one does not need to explicitly learn the parents of the outcome to identify the best intervention.

View on arXivView PDF

Code Implementations(1)

Apache-2.0
50Python, Jupyter NotebookJun 6, 20233 years agoApache-2.0

Similar Papers

Cite this paper

@article{malek2023additive,
  title  = {Additive Causal Bandits with Unknown Graph},
  author = {Alan Malek and Virginia Aglietti and Silvia Chiappa},
  year   = {2023},
  eprint = {2306.07858},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2306.07858v1},
  journal = {ICML 2023 2023}
}

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/2306.07858.svg)](https://codeofpaper.com/paper/2306.07858)

Discussion