Scalable causal discovery with score matching

Montagna F, Noceti N, Rosasco L, Zhang K, Locatello F. 2023. Scalable causal discovery with score matching. 2nd Conference on Causal Learning and Reasoning. CLeaR: Conference on Causal Learning and Reasoning.

Download (ext.)
Conference Paper | Published | English

Scopus indexed
Author
Montagna, Francesco; Noceti, Nicoletta; Rosasco, Lorenzo; Zhang, Kun; Locatello, FrancescoISTA
Department
Abstract
This paper demonstrates how to discover the whole causal graph from the second derivative of the log-likelihood in observational non-linear additive Gaussian noise models. Leveraging scalable machine learning approaches to approximate the score function ∇logp(X), we extend the work of Rolland et al. (2022) that only recovers the topological order from the score and requires an expensive pruning step removing spurious edges among those admitted by the ordering. Our analysis leads to DAS (acronym for Discovery At Scale), a practical algorithm that reduces the complexity of the pruning by a factor proportional to the graph size. In practice, DAS achieves competitive accuracy with current state-of-the-art while being over an order of magnitude faster. Overall, our approach enables principled and scalable causal discovery, significantly lowering the compute bar.
Publishing Year
Date Published
2023-04-01
Proceedings Title
2nd Conference on Causal Learning and Reasoning
Conference
CLeaR: Conference on Causal Learning and Reasoning
Conference Location
Tübingen, Germany
Conference Date
2023-04-11 – 2023-04-14
IST-REx-ID

Cite this

Montagna F, Noceti N, Rosasco L, Zhang K, Locatello F. Scalable causal discovery with score matching. In: 2nd Conference on Causal Learning and Reasoning. ; 2023.
Montagna, F., Noceti, N., Rosasco, L., Zhang, K., & Locatello, F. (2023). Scalable causal discovery with score matching. In 2nd Conference on Causal Learning and Reasoning. Tübingen, Germany.
Montagna, Francesco, Nicoletta Noceti, Lorenzo Rosasco, Kun Zhang, and Francesco Locatello. “Scalable Causal Discovery with Score Matching.” In 2nd Conference on Causal Learning and Reasoning, 2023.
F. Montagna, N. Noceti, L. Rosasco, K. Zhang, and F. Locatello, “Scalable causal discovery with score matching,” in 2nd Conference on Causal Learning and Reasoning, Tübingen, Germany, 2023.
Montagna F, Noceti N, Rosasco L, Zhang K, Locatello F. 2023. Scalable causal discovery with score matching. 2nd Conference on Causal Learning and Reasoning. CLeaR: Conference on Causal Learning and Reasoning.
Montagna, Francesco, et al. “Scalable Causal Discovery with Score Matching.” 2nd Conference on Causal Learning and Reasoning, 2023.
All files available under the following license(s):
Copyright Statement:
This Item is protected by copyright and/or related rights. [...]

Link(s) to Main File(s)
Access Level
OA Open Access

Export

Marked Publications

Open Data ISTA Research Explorer

Sources

arXiv 2304.03382

Search this title in

Google Scholar