Efficient Ensemble Conditional Independence Test Framework for Causal Discovery
Zhengkang Guan, Kun Kuang
Abstract
Constraint-based causal discovery relies on numerous conditional independence tests (CITs), but its practical applicability is severely constrained by the prohibitive computational cost, especially as CITs themselves have high time complexity with respect to the sample size. To address this key bottleneck, we introduce the Ensemble Conditional Independence Test (E-CIT), a general-purpose and plug-and-play framework. E-CIT operates on an intuitive divide-and-aggregate strategy: it partitions the data into subsets, applies a given base CIT independently to each subset, and aggregates the resulting p-values using a novel method grounded in the properties of stable distributions. This framework reduces the computational complexity of a base CIT to linear in the sample size when the subset size is fixed. Moreover, our tailored p-value combination method offers theoretical consistency guarantees under mild conditions on the subtests. Experimental results demonstrate that E-CIT not only significantly reduces the computational burden of CITs and causal discovery but also achieves competitive performance. Notably, it exhibits an improvement in complex testing scenarios, particularly on real-world datasets.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 4483f3f2-2506-403f-9558-8d665963cd65Builds on12
- On the Role of Sparsity and DAG Constraints for Learning Linear DAGsIgnavier Ng, AmirEmad Ghassami, Kun ZhangNeurIPS 2020 · 306 citations
- Causal Discovery with Reinforcement LearningShengyu Zhu, Ignavier Ng, Zhitang ChenICLR 2020 · 285 citations
- Iterative Causal Discovery in the Possible Presence of Latent Confounders and Selection BiasRaanan Y. Rohekar, Shami Nisimov, Yaniv Gurwicz, Gal NovikNeurIPS 2021 · 43 citations
- Recursive Causal Structure Learning in the Presence of Latent Variables and Selection BiasSina Akbari, Ehsan Mokhtarian, AmirEmad Ghassami, Negar KiyavashNeurIPS 2021 · 37 citations
- Residual Similarity Based Conditional Independence Test and Its Application in Causal DiscoveryHao Zhang, Shuigeng Zhou, Kun Zhang, Jihong GuanAAAI 2022 · 22 citations
Related papers
- Causal Discovery in the Wild: A Voting-Theoretic Ensemble ApproachVy Vo, Haoxuan Li, Mingming GongICLR 2026
- Extracting Rare Dependence Patterns via Adaptive Sample ReweightingYiqing Li, Yewei Xia, Xiaofei Wang, Zhengming Chen et al.ICML 2025
- A Recursive Decomposition Framework for Causal Structure Learning in the Presence of Latent VariablesZheng Li, Feng Xie, Shenglan Nie, Xichen Guo et al.ICML 2026
- Differentially Private Nonlinear Causal Discovery from Numerical DataHao Zhang, Yewei Xia, Yixin Ren, Jihong Guan et al.AAAI 2023 · 6 citations
- Conditional Independence Testing with Heteroskedastic Data and Applications to Causal DiscoveryWiebke Günther, Urmi Ninad, Jonas Wahl, Jakob RungeNeurIPS 2022 · 6 citations
