FedCSL: a scalable and accurate approach to federated causal structure learning
| dc.contributor.author | Guo, X. | |
| dc.contributor.author | Yu, K. | |
| dc.contributor.author | Liu, L. | |
| dc.contributor.author | Li, J. | |
| dc.contributor.conference | Thirty-Eighth AAAI Conference on Artificial Intelligence (20 Feb 2024 - 27 Feb 2024 : Vancouver) | |
| dc.contributor.editor | Wooldridge, M. | |
| dc.contributor.editor | Dy, J. | |
| dc.contributor.editor | Natarajan, S. | |
| dc.date.issued | 2024 | |
| dc.description.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. | |
| dc.identifier.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 | |
| dc.identifier.doi | 10.1609/aaai.v38i11.29113 | |
| dc.identifier.isbn | 9781577358879 | |
| dc.identifier.issn | 2159-5399 | |
| dc.identifier.issn | 2374-3468 | |
| dc.identifier.uri | https://hdl.handle.net/11541.2/38336 | |
| dc.language.iso | en | |
| dc.publisher | AAAI Press | |
| dc.publisher.place | US | |
| dc.relation.ispartofseries | 38, 2159-5399 | |
| dc.rights | Copyright 2024 Association for the Advancement of Artificial Intelligence Access Condition Notes: Accepted manuscript available after 1 April 2025 | |
| dc.source.uri | https://doi.org/10.1609/aaai.v38i11.29113 | |
| dc.subject | ML: Causal Learning | |
| dc.subject | ML: Distributed Machine Learning & Federated Learning | |
| dc.subject | ML: Dimensionality Reduction/Feature Selection | |
| dc.subject | DMKM: Scalability | |
| dc.subject | Parallel & Distributed Systems | |
| dc.title | FedCSL: a scalable and accurate approach to federated causal structure learning | |
| dc.type | Conference paper | |
| pubs.publication-status | Published | |
| ror.fileinfo | 12284557280001831 13284557270001831 Open Access Postprint | |
| ror.mmsid | 9916847128101831 |
Files
Original bundle
1 - 1 of 1
No Thumbnail Available
- Name:
- 9916847128101831_12284557280001831_AM FedCSL.pdf
- Size:
- 427.02 KB
- Format:
- Adobe Portable Document Format
- Description:
- Published version