FedCSL: a scalable and accurate approach to federated causal structure learning
Files
(Published version)
Date
2024
Authors
Guo, X.
Yu, K.
Liu, L.
Li, J.
Editors
Wooldridge, M.
Dy, J.
Natarajan, S.
Dy, J.
Natarajan, S.
Advisors
Journal Title
Journal ISSN
Volume Title
Type:
Conference paper
Citation
Proceedings of the ... AAAI Conference on Artificial Intelligence. AAAI Conference on Artificial Intelligence, 2024 / Wooldridge, M., Dy, J., Natarajan, S. (ed./s), vol.38, iss.11, pp.12235-12243
Statement of Responsibility
Conference Name
Thirty-Eighth AAAI Conference on Artificial Intelligence (20 Feb 2024 - 27 Feb 2024 : Vancouver)
Abstract
As an emerging research direction, federated causal structure learning (CSL) aims at learning causal relationships from decentralized data across multiple clients while preserving data privacy. Existing federated CSL algorithms suffer from scalability and accuracy issues, since they require computationally expensive CSL algorithms to be executed at each client. Furthermore, in real-world scenarios, the number of samples held by each client varies significantly, and existing methods still assign equal weights to the learned structural information from each client, which severely harms the learning accuracy of those methods.
To address these two limitations, we propose FedCSL, a scalable and accurate method for federated CSL. Specifically, FedCSL consists of two novel strategies: (1) a federated local-to-global learning strategy that enables FedCSL to scale to high-dimensional data for tackling the scalability issue, and (2) a novel weighted aggregation strategy that does not rely on any complex encryption techniques while preserving data privacy for tackling the accuracy issue. Extensive experiments on benchmark datasets, high-dimensional synthetic datasets and a real-world dataset verify the efficacy of the proposed FedCSL method. The source code is available at https://github.com/Xianjie-Guo/FedCSL.
School/Discipline
Dissertation Note
Provenance
Description
Access Status
Rights
Copyright 2024 Association for the Advancement of Artificial Intelligence
Access Condition Notes: Accepted manuscript available after 1 April 2025