Characterization and Learning of Causal Graphs with Latent Confounders and Post-treatment Selection from Interventional Data
Gongxu Luo, Loka Li, Guangyi Chen, Haoyue Dai, Kun Zhang
Abstract
Interventional causal discovery seeks to identify causal relations by leveraging distributional changes introduced by interventions, even in the presence of latent confounders. Beyond the spurious dependencies induced by latent confounders, we highlight a common yet often overlooked challenge in the problem due to post-treatment selection, in which samples are selectively included in datasets after interventions. This fundamental challenge widely exists in biological studies; for example, in gene expression analysis, both observational and interventional samples are retained only if they meet quality control criteria (e.g., highly active cells). Neglecting post-treatment selection may introduce spurious dependencies and distributional changes under interventions, which can mimic causal responses, thereby distorting causal discovery results and challenging existing causal formulations. To address this, we introduce a novel causal formulation that explicitly models post-treatment selection and reveals how its differential reactions to interventions can distinguish causal relations from selection patterns, allowing us to go beyond traditional equivalence classes toward the underlying true causal structure. We then characterize its Markov properties and propose a ine-grained nterventional equivalence class, named -Markov equivalence, represented by a new graphical diagram, -PAG. Finally, we develop a provably sound and complete algorithm, -FCI, to identify causal relations, latent confounders, and post-treatment selection up to -Markov equivalence, using both observational and interventional data. Experimental results on synthetic and real-world datasets demonstrate that our method recovers causal relations despite the presence of both selection and latent confounders.
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.
Cited by top-tier papers3
- Distributional Equivalence in Linear Non-Gaussian Latent-Variable Cyclic Causal Models: Characterization and LearningHaoyue Dai, Immanuel Albrecht, Peter Spirtes, Kun ZhangICLR 2026 · 4 citations
- PersonaX: Multimodal Datasets with LLM-Inferred Behavior TraitsLoka Li, Wong Yu Kang, Minghao Fu, Guangyi Chen et al.ICLR 2026 · 2 citations
- Causal Modeling of Selection in EvolutionHaoyue Dai, Zeyu Tang, Peter Spirtes, Kun ZhangICML 2026
Builds on6
- Causal Discovery from Soft Interventions with Unknown Targets: Characterization and LearningAmin Jaber, Murat Kocaoglu, Karthikeyan Shanmugam, Elias BareinboimNeurIPS 2020 · 136 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
- Causal discovery from observational and interventional data across multiple environmentsAdam Li, Amin Jaber, Elias BareinboimNeurIPS 2023 · 41 citations
- Gene Regulatory Network Inference in the Presence of Selection Bias and Latent ConfoundersGongxu Luo, Haoyue Dai, Longkang Li, Chengqian Gao et al.NeurIPS 2025 · 9 citations
- Characterization and Learning of Causal Graphs from Hard InterventionsZihan Zhou, Muhammad Qasim Elahi, Murat KocaogluNeurIPS 2025 · 4 citations
Related papers
- When Selection Meets Intervention: Additional Complexities in Causal DiscoveryHaoyue Dai, Ignavier Ng, Jianle Sun, Zeyu Tang et al.ICLR 2025
- Towards Completeness in Causal Discovery from Soft Interventions with Known TargetsZihan Zhou, Murat KocaogluICML 2026
- Learning Mixtures of Unknown Causal InterventionsAbhinav Kumar, Kirankumar Shiragur, Caroline UhlerNeurIPS 2024 · 1 citation
- Efficient Causal Structure Learning from Multiple Interventional Datasets with Unknown TargetsYunxia Wang, Fuyuan Cao, Kui Yu, Jiye LiangAAAI 2022 · 7 citations
- Less Greedy Equivalence SearchAdiba Ejaz, Elias BareinboimNeurIPS 2025 · 1 citation
