返回
VertiBayes: learning Bayesian network parameters from vertically partitioned data with missing values
DOI:10.1007/s40747-024-01424-0.png)
摘要
En 中文
Federated learning makes it possible to train a machine learning model on decentralized data. Bayesian networks are widely used probabilistic graphical models. While some research has been published on the federated learning of Bayesian networks, publications on Bayesian networks in a vertically partitioned data setting are limited, with important omissions, such as handling missing data. We propose a novel method called VertiBayes to train Bayesian networks (structure and parameters) on vertically partitioned data, which can handle missing values as well as an arbitrary number of parties. For structure learning we adapted the K2 algorithm with a privacy-preserving scalar product protocol. For parameter learning, we use a two-step approach: first, we learn an intermediate model using maximum likelihood, treating missing values as a special value, then we train a model on synthetic data generated by the intermediate model using the EM algorithm. The privacy guarantees of VertiBayes are equivalent to those provided by the privacy preserving scalar product protocol used. We experimentally show VertiBayes produces models comparable to those learnt using traditional algorithms. Finally, we propose two alternative approaches to estimate the performance of the model using vertically partitioned data and we show in experiments that these give accurate estimates.
Keyword:
Federated Learning
Bayesian network
Privacy preserving
Vertically partitioned data
Parameter learning
Structure learning
期刊
IF:
4.6
论文数:
2.1K
被引数:
6.6K
机构
引用论文
Asymptotics of reaction–diffusion fronts with one static and one diffusing reactant具有一个静态和一个扩散反应物的反应扩散前沿的渐近性
Preparation and Characterization of Inclusion Complexes of Poly(propylene glycol) with Cyclodextrins

