English

Differentiable Constraint-Based Causal Discovery

Machine Learning 2026-02-06 v2 Artificial Intelligence Machine Learning

Abstract

Causal discovery from observational data is a fundamental task in artificial intelligence, with far-reaching implications for decision-making, predictions, and interventions. Despite significant advances, existing methods can be broadly categorized as constraint-based or score-based approaches. Constraint-based methods offer rigorous causal discovery but are often hindered by small sample sizes, while score-based methods provide flexible optimization but typically forgo explicit conditional independence testing. This work explores a third avenue: developing differentiable dd-separation scores, obtained through a percolation theory using soft logic. This enables the implementation of a new type of causal discovery method: gradient-based optimization of conditional independence constraints. Empirical evaluations demonstrate the robust performance of our approach in low-sample regimes, surpassing traditional constraint-based and score-based baselines on a real-world dataset. Code and data of the proposed method are publicly available at https://github..com/PurdueMINDS/DAGPA.

Keywords

Cite

@article{arxiv.2510.22031,
  title  = {Differentiable Constraint-Based Causal Discovery},
  author = {Jincheng Zhou and Mengbo Wang and Anqi He and Yumeng Zhou and Hessam Olya and Murat Kocaoglu and Bruno Ribeiro},
  journal= {arXiv preprint arXiv:2510.22031},
  year   = {2026}
}
R2 v1 2026-07-01T07:05:03.284Z