跨机构协作学习动态贝叶斯网络,保护隐私且适应数据差异。
Federated Learning of Dynamic Bayesian Network via Continuous Optimization from Time Series Data
- 用连续优化方法在客户端间交换参数,不共享原始数据。
- 在小样本多客户端场景下,结构学习准确率优于现有方法。
- 适用于医疗、金融等数据分散且分布不一的领域。
传统上,动态贝叶斯网络(DBN)的结构学习依赖集中式数据,需将所有数据汇聚至一处。然而在真实场景中,数据通常分布在多个实体(如企业、设备)之间,各方希望协作学习DBN结构,同时保障数据隐私与安全。更重要的是,由于客户端多样性,数据可能呈现不同分布,导致数据异质性,进一步加剧集中式方法的挑战。本文首先提出一种针对水平分布同质时间序列数据的联邦学习框架,用于估计动态贝叶斯网络结构;随后通过引入近端算子作为正则项,将该方法扩展至异质时间序列数据,构建个性化联邦学习机制。为此,我们提出 exttt{FDBNL} 与 exttt{PFDBNL} 框架,均采用连续优化策略,仅在训练过程中交换模型参数。在合成数据与真实世界数据集上的实验表明,本方法在多客户端、单个客户端样本量有限的场景下显著优于当前最优技术。
原文摘要 · Abstract (English)
Traditionally, learning the structure of a Dynamic Bayesian Network has been centralized, requiring all data to be pooled in one location. However, in real-world scenarios, data are often distributed across multiple entities (e.g., companies, devices) that seek to collaboratively learn a Dynamic Bayesian Network while preserving data privacy and security. More importantly, due to the presence of diverse clients, the data may follow different distributions, resulting in data heterogeneity. This heterogeneity poses additional challenges for centralized approaches. In this study, we first introduce a federated learning approach for estimating the structure of a Dynamic Bayesian Network from homogeneous time series data that are horizontally distributed across different parties. We then extend this approach to heterogeneous time series data by incorporating a proximal operator as a regularization term in a personalized federated learning framework. To this end, we propose \texttt{FDBNL} and \texttt{PFDBNL}, which leverage continuous optimization, ensuring that only model parameters are exchanged during the optimization process. Experimental results on synthetic and real-world datasets demonstrate that our method outperforms state-of-the-art techniques, particularly in scenarios with many clients and limited individual sample sizes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。