Embracing Discrete Search: A Reasonable Approach to Causal Structure Learning
Marcel Wienöbst, Leonard Henckel, Sebastian Weichwald
Abstract
We present FLOP (Fast Learning of Order and Parents), a score-based causal discovery algorithm for linear models. It pairs fast parent selection with iterative Cholesky-based score updates, cutting run-times over prior algorithms. This makes it feasible to fully embrace discrete search, enabling iterated local search with principled order initialization to find graphs with scores at or close to the global optimum. The resulting structures are highly accurate across benchmarks, with near-perfect recovery in standard settings. This performance calls for revisiting discrete search over graphs as a reasonable approach to causal discovery.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on7
- On the Role of Sparsity and DAG Constraints for Learning Linear DAGsIgnavier Ng, AmirEmad Ghassami, Kun ZhangNeurIPS 2020 · 306 citations
- DAGMA: Learning DAGs via M-matrices and a Log-Determinant Acyclicity CharacterizationKevin Bello, Bryon Aragam, Pradeep RavikumarNeurIPS 2022 · 222 citations
- Beware of the Simulated DAG! Causal Discovery Benchmarks May Be Easy to GameAlexander G. Reisach, Christof Seiler, Sebastian WeichwaldNeurIPS 2021 · 213 citations
- Score Matching Enables Causal Discovery of Nonlinear Additive Noise ModelsPaul Rolland, Volkan Cevher, Matthäus Kleindessner, Chris Russell et al.ICML 2022 · 123 citations
- Fast Scalable and Accurate Discovery of DAGs Using the Best Order Score Search and Grow Shrink TreesBryan Andrews, Joseph D. Ramsey, Ruben Sanchez-Romero, Jazmin Camchong et al.NeurIPS 2023 · 61 citations
Related papers
- Hybrid Top-Down Global Causal Discovery with Local Search for Linear and Nonlinear Additive Noise ModelsSujai Hiremath, Jacqueline R. M. A. Maasch, Mengxiao Gao, Promit Ghosal et al.NeurIPS 2024 · 11 citations
- Ordering-based Causal Discovery via Generalized Score MatchingVy Vo, Trung Le, He Zhao, Edwin V. Bonilla et al.KDD 2026 · 1 citation
- Improving Causal Discovery By Optimal Bayesian Network LearningNi Y. Lu, Kun Zhang, Changhe YuanAAAI 2021 · 25 citations
- Reliable Causal Discovery with Improved Exact Search and Weaker AssumptionsIgnavier Ng, Yujia Zheng, Jiji Zhang, Kun ZhangNeurIPS 2021 · 35 citations
- QWO: Speeding Up Permutation-Based Causal Discovery in LiGAMsMohammad Shahverdikondori, Ehsan Mokhtarian, Negar KiyavashNeurIPS 2024 · 1 citation
