Estimating Interventional Distributions with Uncertain Causal Graphs through Meta-Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Dhir, Anish, Diaconu, Cristiana, Lungu, Valentinian Mihai, Requeima, James, Turner, Richard E., van der Wilk, Mark
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910017159954432
author Dhir, Anish
Diaconu, Cristiana
Lungu, Valentinian Mihai
Requeima, James
Turner, Richard E.
van der Wilk, Mark
author_facet Dhir, Anish
Diaconu, Cristiana
Lungu, Valentinian Mihai
Requeima, James
Turner, Richard E.
van der Wilk, Mark
contents In scientific domains -- from biology to the social sciences -- many questions boil down to \textit{What effect will we observe if we intervene on a particular variable?} If the causal relationships (e.g.~a causal graph) are known, it is possible to estimate the intervention distributions. In the absence of this domain knowledge, the causal structure must be discovered from the available observational data. However, observational data are often compatible with multiple causal graphs, making methods that commit to a single structure prone to overconfidence. A principled way to manage this structural uncertainty is via Bayesian inference, which averages over a posterior distribution on possible causal structures and functional mechanisms. Unfortunately, the number of causal structures grows super-exponentially with the number of nodes in the graph, making computations intractable. We propose to circumvent these challenges by using meta-learning to create an end-to-end model: the Model-Averaged Causal Estimation Transformer Neural Process (MACE-TNP). The model is trained to predict the Bayesian model-averaged interventional posterior distribution, and its end-to-end nature bypasses the need for expensive calculations. Empirically, we demonstrate that MACE-TNP outperforms strong Bayesian baselines. Our work establishes meta-learning as a flexible and scalable paradigm for approximating complex Bayesian causal inference, that can be scaled to increasingly challenging settings in the future.
format Preprint
id arxiv_https___arxiv_org_abs_2507_05526
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Estimating Interventional Distributions with Uncertain Causal Graphs through Meta-Learning
Dhir, Anish
Diaconu, Cristiana
Lungu, Valentinian Mihai
Requeima, James
Turner, Richard E.
van der Wilk, Mark
Machine Learning
Methodology
In scientific domains -- from biology to the social sciences -- many questions boil down to \textit{What effect will we observe if we intervene on a particular variable?} If the causal relationships (e.g.~a causal graph) are known, it is possible to estimate the intervention distributions. In the absence of this domain knowledge, the causal structure must be discovered from the available observational data. However, observational data are often compatible with multiple causal graphs, making methods that commit to a single structure prone to overconfidence. A principled way to manage this structural uncertainty is via Bayesian inference, which averages over a posterior distribution on possible causal structures and functional mechanisms. Unfortunately, the number of causal structures grows super-exponentially with the number of nodes in the graph, making computations intractable. We propose to circumvent these challenges by using meta-learning to create an end-to-end model: the Model-Averaged Causal Estimation Transformer Neural Process (MACE-TNP). The model is trained to predict the Bayesian model-averaged interventional posterior distribution, and its end-to-end nature bypasses the need for expensive calculations. Empirically, we demonstrate that MACE-TNP outperforms strong Bayesian baselines. Our work establishes meta-learning as a flexible and scalable paradigm for approximating complex Bayesian causal inference, that can be scaled to increasingly challenging settings in the future.
title Estimating Interventional Distributions with Uncertain Causal Graphs through Meta-Learning
topic Machine Learning
Methodology
url https://arxiv.org/abs/2507.05526