Using causal abstractions to accelerate decision-making in complex bandit problems

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Dyer, Joel, Bishop, Nicholas, Calinescu, Anisoara, Wooldridge, Michael, Zennaro, Fabio Massimo
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916933914329088
author Dyer, Joel
Bishop, Nicholas
Calinescu, Anisoara
Wooldridge, Michael
Zennaro, Fabio Massimo
author_facet Dyer, Joel
Bishop, Nicholas
Calinescu, Anisoara
Wooldridge, Michael
Zennaro, Fabio Massimo
contents Although real-world decision-making problems can often be encoded as causal multi-armed bandits (CMABs) at different levels of abstraction, a general methodology exploiting the information and computational advantages of each abstraction level is missing. In this paper, we propose AT-UCB, an algorithm which efficiently exploits shared information between CMAB problem instances defined at different levels of abstraction. More specifically, AT-UCB leverages causal abstraction (CA) theory to explore within a cheap-to-simulate and coarse-grained CMAB instance, before employing the traditional upper confidence bound (UCB) algorithm on a restricted set of potentially optimal actions in the CMAB of interest, leading to significant reductions in cumulative regret when compared to the classical UCB algorithm. We illustrate the advantages of AT-UCB theoretically, through a novel upper bound on the cumulative regret, and empirically, by applying AT-UCB to epidemiological simulators with varying resolution and computational cost.
format Preprint
id arxiv_https___arxiv_org_abs_2509_04296
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Using causal abstractions to accelerate decision-making in complex bandit problems
Dyer, Joel
Bishop, Nicholas
Calinescu, Anisoara
Wooldridge, Michael
Zennaro, Fabio Massimo
Machine Learning
Although real-world decision-making problems can often be encoded as causal multi-armed bandits (CMABs) at different levels of abstraction, a general methodology exploiting the information and computational advantages of each abstraction level is missing. In this paper, we propose AT-UCB, an algorithm which efficiently exploits shared information between CMAB problem instances defined at different levels of abstraction. More specifically, AT-UCB leverages causal abstraction (CA) theory to explore within a cheap-to-simulate and coarse-grained CMAB instance, before employing the traditional upper confidence bound (UCB) algorithm on a restricted set of potentially optimal actions in the CMAB of interest, leading to significant reductions in cumulative regret when compared to the classical UCB algorithm. We illustrate the advantages of AT-UCB theoretically, through a novel upper bound on the cumulative regret, and empirically, by applying AT-UCB to epidemiological simulators with varying resolution and computational cost.
title Using causal abstractions to accelerate decision-making in complex bandit problems
topic Machine Learning
url https://arxiv.org/abs/2509.04296