Marginal Causal Flows for Validation and Inference

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Manela, Daniel de Vassimon, Battaglia, Laura, Evans, Robin J.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912144797204480
author Manela, Daniel de Vassimon
Battaglia, Laura
Evans, Robin J.
author_facet Manela, Daniel de Vassimon
Battaglia, Laura
Evans, Robin J.
contents Investigating the marginal causal effect of an intervention on an outcome from complex data remains challenging due to the inflexibility of employed models and the lack of complexity in causal benchmark datasets, which often fail to reproduce intricate real-world data patterns. In this paper we introduce Frugal Flows, a novel likelihood-based machine learning model that uses normalising flows to flexibly learn the data-generating process, while also directly inferring the marginal causal quantities from observational data. We propose that these models are exceptionally well suited for generating synthetic data to validate causal methods. They can create synthetic datasets that closely resemble the empirical dataset, while automatically and exactly satisfying a user-defined average treatment effect. To our knowledge, Frugal Flows are the first generative model to both learn flexible data representations and also exactly parameterise quantities such as the average treatment effect and the degree of unobserved confounding. We demonstrate the above with experiments on both simulated and real-world datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2411_01295
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Marginal Causal Flows for Validation and Inference
Manela, Daniel de Vassimon
Battaglia, Laura
Evans, Robin J.
Machine Learning
Artificial Intelligence
Methodology
Investigating the marginal causal effect of an intervention on an outcome from complex data remains challenging due to the inflexibility of employed models and the lack of complexity in causal benchmark datasets, which often fail to reproduce intricate real-world data patterns. In this paper we introduce Frugal Flows, a novel likelihood-based machine learning model that uses normalising flows to flexibly learn the data-generating process, while also directly inferring the marginal causal quantities from observational data. We propose that these models are exceptionally well suited for generating synthetic data to validate causal methods. They can create synthetic datasets that closely resemble the empirical dataset, while automatically and exactly satisfying a user-defined average treatment effect. To our knowledge, Frugal Flows are the first generative model to both learn flexible data representations and also exactly parameterise quantities such as the average treatment effect and the degree of unobserved confounding. We demonstrate the above with experiments on both simulated and real-world datasets.
title Marginal Causal Flows for Validation and Inference
topic Machine Learning
Artificial Intelligence
Methodology
url https://arxiv.org/abs/2411.01295