Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Learning DAGs from Data with Few Root Causes

About

We present a novel perspective and algorithm for learning directed acyclic graphs (DAGs) from data generated by a linear structural equation model (SEM). First, we show that a linear SEM can be viewed as a linear transform that, in prior work, computes the data from a dense input vector of random valued root causes (as we will call them) associated with the nodes. Instead, we consider the case of (approximately) few root causes and also introduce noise in the measurement of the data. Intuitively, this means that the DAG data is produced by few data-generating events whose effect percolates through the DAG. We prove identifiability in this new setting and show that the true DAG is the global minimizer of the $L^0$-norm of the vector of root causes. For data with few root causes, with and without noise, we show superior performance compared to prior DAG learning methods.

Panagiotis Misiakos, Chris Wendler, Markus P\"uschel• 2023

Related benchmarks

TaskDatasetResultRank
Causal DiscoverySynthetic DAGs
TPR1
125
DAG learningSynthetic (test)
SID51
101
DAG learningSynthetic DAGs (100 nodes, 400 edges) v1
SHD60
51
Causal DiscoverySynthetic DAG data--
40
Causal DiscoverySynthetic DAG data (test)--
40
DAG learningSynthetic DAG data
Runtime (s)101.8
26
Causal DiscoverySynthetic Data--
21
Causal DiscoverySynthetic DAG Datasets
Runtime (s)101.8
14
DAG learningSynthetic Large DAGs
Runtime (s)21
12
Learning DAGsSynthetic Larger DAGs v1 (test)
SHD22
12
Showing 10 of 20 rows

Other info

Code

Follow for update