Horizontal and Vertical Federated Causal Structure Learning via Higher-order Cumulants
Wei Chen, Wanyang Gu, Linjun Peng, Ting Yan, Ruichu Cai, Zhifeng Hao, Kun Zhang
Abstract
Federated causal discovery aims to uncover causal relationships while protecting data privacy, with significant real-world applications. Existing methods focus on horizontal federated settings where clients share the same variables but have different samples. However, in practice, clients may have different variables, leading to spurious causal relationships. To address this issue, we comprehensively consider causal structure learning methods under both horizontal and vertical federated settings. Interestingly, we find that, higher-order cumulants rely solely on the joint distribution of the relevant variables and are useful to solve the above problem in the linear non-Gaussian case. This motivates us to provide the identification theories for determining the causal order over observed variables, leveraging the difference in the product of the (cross) cumulants of the specific variables. Based on these theories, we develop a method for learning causal order in the horizontal and vertical federated scenarios. Specifically, we first obtain local (cross) cumulant matrices of observed variables from all participating clients to construct a global cumulant matrix. This global cumulant matrix is then used for recursive source variable identification, ultimately yielding a causal strength matrix of the union of variables from all clients. Our algorithm demonstrates superior performance in experiments on both synthetic and real-world data.
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 on6
- Causal Discovery from Multiple Data Sets with Non-Identical Variable SetsBiwei Huang, Kun Zhang, Mingming Gong, Clark GlymourAAAI 2020 · 40 citations
- Causal Discovery with Latent Confounders Based on Higher-Order CumulantsRuichu Cai, Zhiyi Huang, Wei Chen, Zhifeng Hao et al.ICML 2023 · 22 citations
- Federated Causality Learning with Explainable Adaptive OptimizationDezhi Yang, Xintong He, Jun Wang, Guoxian Yu et al.AAAI 2024 · 21 citations
- FedCSL: A Scalable and Accurate Approach to Federated Causal Structure LearningXianjie Guo, Kui Yu, Lin Liu, Jiuyong LiAAAI 2024 · 17 citations
- Federated Causal Discovery from Heterogeneous DataLoka Li, Ignavier Ng, Gongxu Luo, Biwei Huang et al.ICLR 2024 · 16 citations
Related papers
- Federated Causal Structure Learning with Non-identical Variable SetsYunxia Wang, Fuyuan Cao, Kui Yu, Jiye LiangICML 2025
- Identification of Causal Structure with Latent Variables Based on Higher Order CumulantsWei Chen, Zhiyi Huang, Ruichu Cai, Zhifeng Hao et al.AAAI 2024 · 10 citations
- Learning Discrete Latent Variable Structures with Tensor Rank ConditionsZhengming Chen, Ruichu Cai, Feng Xie, Jie Qiao et al.NeurIPS 2024 · 7 citations
- Federated Nonlinear Causal Discovery via Divide-and-Conquer LearningXianjie Guo, Shuai Yang, Lin Ma, Xi Cheng et al.KDD 2026 · 1 citation
- Regret-Based Federated Causal Discovery with Unknown InterventionsFederico Baldo, Charles AssaadICML 2026 · 2 citations
