# Towards Privacy-Aware Causal Structure Learning in Federated Setting

arXiv: https://arxiv.org/abs/2211.06919 · HTML: https://mlsanity.com/paper/2211.06919/

## Key facts

- Cited by 11 papers in the map (data through 14 Oct 2026).
- Builds on 17 of its 54 references that are in the map.
- First posted on arXiv 13 Nov 2022.

## Builds on (most cited first, 17 of 17)

- [Communication-Efficient Learning of Deep Networks from Decentralized Data](https://mlsanity.com/paper/1602.05629/) (2016)
- [DAGs with NO TEARS: Continuous Optimization for Structure Learning](https://mlsanity.com/paper/1803.01422/) (2018)
- [DAG-GNN: DAG Structure Learning with Graph Neural Networks](https://mlsanity.com/paper/1904.10098/) (2019)
- [Gradient-Based Neural DAG Learning](https://mlsanity.com/paper/1906.02226/) (2019)
- [D'ya like DAGs? A Survey on Structure Learning and Causal Discovery](https://mlsanity.com/paper/2103.02582/) (2021)
- [Learning Sparse Nonparametric DAGs](https://mlsanity.com/paper/1909.13189/) (2019)
- [On the Role of Sparsity and DAG Constraints for Learning Linear DAGs](https://mlsanity.com/paper/2006.10201/) (2020)
- [Causal Inference and Causal Explanation with Background Knowledge](https://mlsanity.com/paper/1302.4972/) (2013)
- [Masked Gradient-Based Causal Structure Learning](https://mlsanity.com/paper/1910.08527/) (2019)
- [DAGs with No Curl: An Efficient DAG Structure Learning Approach](https://mlsanity.com/paper/2106.07197/) (2021)
- [Causal Discovery Toolbox: Uncover causal relationships in Python](https://mlsanity.com/paper/1903.02278/) (2019)
- [Constraint-based Causal Discovery from Multiple Interventions over Overlapping Variable Sets](https://mlsanity.com/paper/1403.2150/) (2014)
- [Towards Federated Bayesian Network Structure Learning with Continuous Optimization](https://mlsanity.com/paper/2110.09356/) (2021)
- [Federated Causal Inference in Heterogeneous Observational Data](https://mlsanity.com/paper/2107.11732/) (2021)
- [Learning causal representations for robust domain adaptation](https://mlsanity.com/paper/2011.06317/) (2020)
- [FedDAG: Federated DAG Structure Learning](https://mlsanity.com/paper/2112.03555/) (2021)
- [Towards Efficient Local Causal Structure Learning](https://mlsanity.com/paper/2103.00378/) (2021)

## Built on by (most cited first, 11 of 11)

- [Federated Causal Discovery from Heterogeneous Data](https://mlsanity.com/paper/2402.13241/) (2024)
- [FedGES: A Federated Learning Approach for BN Structure Learning](https://mlsanity.com/paper/2502.01538/) (2025)
- [Regret-Based Federated Causal Discovery with Unknown Interventions](https://mlsanity.com/paper/2512.23626/) (2025)
- [CCBNet: Confidential Collaborative Bayesian Networks Inference](https://mlsanity.com/paper/2405.15055/) (2024)
- [Interventional Causal Structure Discovery over Graphical Models with Convergence and Optimality Guarantees](https://mlsanity.com/paper/2408.04819/) (2024)
- [Federated Concept-Based Models: Interpretable models with distributed supervision](https://mlsanity.com/paper/2602.04093/) (2026)
- [Uncertainty Quantification in Federated Granger Causality Learning](https://mlsanity.com/paper/2602.13004/) (2026)
- [Preserving Data Privacy in Learning Causal Structure with Fully Homomorphic Encryption](https://mlsanity.com/paper/2606.05129/) (2026)
- [A Survey on Federated Causal Discovery and Inference](https://mlsanity.com/paper/2606.23741/) (2026)
- [Federated Causal Discovery via Regression-Directed Cumulants](https://mlsanity.com/paper/2609.03705/) (2026)
- [Fed-ReMasker: Federated Tabular Imputation under Feature-Level Missingness](https://mlsanity.com/paper/2609.28105/) (2026)

## Abstract (arXiv)

Causal structure learning has been extensively studied and widely used in machine learning and various applications. To achieve an ideal performance, existing causal structure learning algorithms often need to centralize a large amount of data from multiple data sources. However, in the privacy-preserving setting, it is impossible to centralize data from all sources and put them together as a single dataset. To preserve data privacy, federated learning as a new learning paradigm has attracted much attention in machine learning in recent years. In this paper, we study a privacy-aware causal structure learning problem in the federated setting and propose a novel Federated PC (FedPC) algorithm with two new strategies for preserving data privacy without centralizing data. Specifically, we first propose a novel layer-wise aggregation strategy for a seamless adaptation of the PC algorithm into the federated learning paradigm for federated skeleton learning, then we design an effective strategy for learning consistent separation sets for federated edge orientation. The extensive experiments validate that FedPC is effective for causal structure learning in a federated learning setting.
