Causal Representation Learning in Temporal Data via Single-Parent Decoding

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Brouillard, Philippe, Lachapelle, Sébastien, Kaltenborn, Julia, Gurwicz, Yaniv, Sridhar, Dhanya, Drouin, Alexandre, Nowack, Peer, Runge, Jakob, Rolnick, David
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916429479018496
author Brouillard, Philippe
Lachapelle, Sébastien
Kaltenborn, Julia
Gurwicz, Yaniv
Sridhar, Dhanya
Drouin, Alexandre
Nowack, Peer
Runge, Jakob
Rolnick, David
author_facet Brouillard, Philippe
Lachapelle, Sébastien
Kaltenborn, Julia
Gurwicz, Yaniv
Sridhar, Dhanya
Drouin, Alexandre
Nowack, Peer
Runge, Jakob
Rolnick, David
contents Scientific research often seeks to understand the causal structure underlying high-level variables in a system. For example, climate scientists study how phenomena, such as El Niño, affect other climate processes at remote locations across the globe. However, scientists typically collect low-level measurements, such as geographically distributed temperature readings. From these, one needs to learn both a mapping to causally-relevant latent variables, such as a high-level representation of the El Niño phenomenon and other processes, as well as the causal model over them. The challenge is that this task, called causal representation learning, is highly underdetermined from observational data alone, requiring other constraints during learning to resolve the indeterminacies. In this work, we consider a temporal model with a sparsity assumption, namely single-parent decoding: each observed low-level variable is only affected by a single latent variable. Such an assumption is reasonable in many scientific applications that require finding groups of low-level variables, such as extracting regions from geographically gridded measurement data in climate research or capturing brain regions from neural activity data. We demonstrate the identifiability of the resulting model and propose a differentiable method, Causal Discovery with Single-parent Decoding (CDSD), that simultaneously learns the underlying latents and a causal graph over them. We assess the validity of our theoretical results using simulated data and showcase the practical validity of our method in an application to real-world data from the climate science field.
format Preprint
id arxiv_https___arxiv_org_abs_2410_07013
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Causal Representation Learning in Temporal Data via Single-Parent Decoding
Brouillard, Philippe
Lachapelle, Sébastien
Kaltenborn, Julia
Gurwicz, Yaniv
Sridhar, Dhanya
Drouin, Alexandre
Nowack, Peer
Runge, Jakob
Rolnick, David
Machine Learning
Scientific research often seeks to understand the causal structure underlying high-level variables in a system. For example, climate scientists study how phenomena, such as El Niño, affect other climate processes at remote locations across the globe. However, scientists typically collect low-level measurements, such as geographically distributed temperature readings. From these, one needs to learn both a mapping to causally-relevant latent variables, such as a high-level representation of the El Niño phenomenon and other processes, as well as the causal model over them. The challenge is that this task, called causal representation learning, is highly underdetermined from observational data alone, requiring other constraints during learning to resolve the indeterminacies. In this work, we consider a temporal model with a sparsity assumption, namely single-parent decoding: each observed low-level variable is only affected by a single latent variable. Such an assumption is reasonable in many scientific applications that require finding groups of low-level variables, such as extracting regions from geographically gridded measurement data in climate research or capturing brain regions from neural activity data. We demonstrate the identifiability of the resulting model and propose a differentiable method, Causal Discovery with Single-parent Decoding (CDSD), that simultaneously learns the underlying latents and a causal graph over them. We assess the validity of our theoretical results using simulated data and showcase the practical validity of our method in an application to real-world data from the climate science field.
title Causal Representation Learning in Temporal Data via Single-Parent Decoding
topic Machine Learning
url https://arxiv.org/abs/2410.07013