中文

一种高效且鲁棒的垂直联邦随机森林系统

机器学习 2022-01-27 v1 密码学与安全 分布式、并行与集群计算

摘要

随着利用多源数据构建更优机器学习模型的需求日益增长,许多垂直联邦学习算法被提出以保护参与机构的数据隐私。然而,现有垂直联邦学习算法的效率仍是一大问题,尤其在应用于大规模真实世界数据集时。本文提出一种快速、准确、可扩展且鲁棒的垂直联邦随机森林系统。通过广泛优化,我们在训练与推理任务上相较 SOTA 的 SecureBoost 模型 \cite{cheng2019secureboost} 分别实现了 5×5\times83×83\times 的加速。此外,所提系统可达到相近精度,同时具备良好的可扩展性与分区容错性。我们的代码已公开,以促进社区发展与用户数据隐私保护。

关键词

引用

@article{arxiv.2201.10761,
  title  = {An Efficient and Robust System for Vertically Federated Random Forest},
  author = {Houpu Yao and Jiazhou Wang and Peng Dai and Liefeng Bo and Yanqing Chen},
  journal= {arXiv preprint arXiv:2201.10761},
  year   = {2022}
}