Supervised Whole DAG Causal Discovery

Hebi Li,Qi Xiao,Jin Tian
DOI: https://doi.org/10.48550/arXiv.2006.04697
2020-06-08
Abstract:We propose to address the task of causal structure learning from data in a supervised manner. Existing work on learning causal directions by supervised learning is restricted to learning pairwise relation, and not well suited for whole DAG discovery. We propose a novel approach of modeling the whole DAG structure discovery as a supervised learning. To fit the problem in hand, we propose to use permutation equivariant models that align well with the problem domain. We evaluate the proposed approach extensively on synthetic graphs of size 10,20,50,100 and real data, and show promising results compared with a variety of previous approaches.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?