A Recursive Decomposition Framework for Causal Structure Learning in the Presence of Latent Variables
Zheng Li, Feng Xie, Shenglan Nie, Xichen Guo, Ruxin Wang, Hao Zhang
Abstract
Constraint-based causal discovery is widely used for learning causal structures, but heavy reliance on conditional independence (CI) testing makes it computationally expensive in high-dimensional settings. To mitigate this limitation, many divideand-conquer frameworks have been proposed, but most assume causal sufficiency, i.e., no latent variables. In this paper, we show that divide-andconquer strategies can be theoretically generalized beyond causal sufficiency to settings with latent variables. Specifically, we propose a recursive decomposition framework, termed DICOLA, that enables divide-and-conquer causal discovery in the presence of latent variables. It recursively decomposes the global learning task into smaller subproblems and integrates their solutions through a principled reconstruction step to recover the global structure. We theoretically establish the soundness and completeness of the proposed framework. Extensive experiments on synthetic data demonstrate that our approach significantly improves computational efficiency across a range of causal discovery algorithms, while experiments on a real-world dataset further illustrate its practical effectiveness. Our source code is available at github.com/zhengli0060/DiCoLa-ICML2026.
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 on9
- Generalized Independent Noise Condition for Estimating Latent Variable Causal GraphsFeng Xie, Ruichu Cai, Biwei Huang, Clark Glymour et al.NeurIPS 2020 · 119 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
- A Versatile Causal Discovery Framework to Allow Causally-Related Hidden VariablesXinshuai Dong, Biwei Huang, Ignavier Ng, Xiangchen Song et al.ICLR 2024 · 29 citations
- Learning Bayesian Networks in the Presence of Structural Side InformationEhsan Mokhtarian, Sina Akbari, Fateme Jamshidi, Jalal Etesami et al.AAAI 2022 · 16 citations
Related papers
- Latent Hierarchical Causal Structure Discovery with Rank ConstraintsBiwei Huang, Charles Jia Han Low, Feng Xie, Clark Glymour et al.NeurIPS 2022 · 78 citations
- Differentiable Causal Discovery for Latent Hierarchical Causal ModelsParjanya Prajakta Prashant, Ignavier Ng, Kun Zhang, Biwei HuangICLR 2025
- Score-based Greedy Search for Structure Identification of Partially Observed Causal ModelsXinshuai Dong, Ignavier Ng, Haoyue Dai, Jiaqi Sun et al.ICLR 2026 · 1 citation
- Structural Estimation of Partially Observed Linear Non-Gaussian Acyclic Model: A Practical Approach with IdentifiabilitySongyao Jin, Feng Xie, Guangyi Chen, Biwei Huang et al.ICLR 2024 · 7 citations
- Causal Structure Recovery with Latent Variables under Milder Distributional and Graphical AssumptionsXiu-Chuan Li, Kun Zhang, Tongliang LiuICLR 2024 · 5 citations
