English

A polynomial-time algorithm for learning nonparametric causal graphs

Machine Learning 2020-11-12 v2 Machine Learning Statistics Theory Statistics Theory

Abstract

We establish finite-sample guarantees for a polynomial-time algorithm for learning a nonlinear, nonparametric directed acyclic graphical (DAG) model from data. The analysis is model-free and does not assume linearity, additivity, independent noise, or faithfulness. Instead, we impose a condition on the residual variances that is closely related to previous work on linear models with equal variances. Compared to an optimal algorithm with oracle knowledge of the variable ordering, the additional cost of the algorithm is linear in the dimension dd and the number of samples nn. Finally, we compare the proposed algorithm to existing approaches in a simulation study.

Keywords

Cite

@article{arxiv.2006.11970,
  title  = {A polynomial-time algorithm for learning nonparametric causal graphs},
  author = {Ming Gao and Yi Ding and Bryon Aragam},
  journal= {arXiv preprint arXiv:2006.11970},
  year   = {2020}
}

Comments

To appear at NeurIPS 2020