arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.08479cs.LGcs.CRstat.ML
收藏

用于垂直联邦学习的大规模安全XGB

Large-Scale Secure XGB for Vertical Federated Learning

Ant Group(蚂蚁集团)

Wenjing Fang, Derun Zhao, Jin Tan, Chaochao Chen, Chaofan Yu, Li Wang, Lei Wang, Jun Zhou, Benyu Zhang

摘要:

近年来,隐私保护机器学习受到越来越多的关注,特别是随着各种隐私保护法规的出台。在这种情况下,联邦学习(FL)似乎有助于在多方之间进行隐私保护的联合建模。虽然许多联邦算法已经被广泛研究,但仍然缺乏安全和实用的梯度树提升模型(例如,XGB)在文学中。在本文中,我们的目标是建立大规模的安全XGB垂直联邦学习设置。我们从三个方面保障数据隐私。具体来说,(i)我们采用安全的多方计算技术,以避免在训练过程中泄漏中间信息,(ii)我们以分布式方式存储输出模型,以最小化信息释放,以及(iii)我们提供了一种新的算法,用于安全的XGB预测与分布式模型。此外,通过提出安全的置换协议,我们可以提高训练效率并使框架可扩展到大数据集。我们在公共数据集和真实数据集上进行了广泛的实验,结果表明,我们提出的XGB模型不仅具有竞争力的准确性,而且具有实用性能。

英文摘要:

Privacy-preserving machine learning has drawn increasingly attention recently, especially with kinds of privacy regulations come into force. Under such situation, Federated Learning (FL) appears to facilitate privacy-preserving joint modeling among multiple parties. Although many federated algorithms have been extensively studied, there is still a lack of secure and practical gradient tree boosting models (e.g., XGB) in literature. In this paper, we aim to build large-scale secure XGB under vertically federated learning setting. We guarantee data privacy from three aspects. Specifically, (i) we employ secure multi-party computation techniques to avoid leaking intermediate information during training, (ii) we store the output model in a distributed manner in order to minimize information release, and (iii) we provide a novel algorithm for secure XGB predict with the distributed model. Furthermore, by proposing secure permutation protocols, we can improve the training efficiency and make the framework scale to large dataset. We conduct extensive experiments on both public datasets and real-world datasets, and the results demonstrate that our proposed XGB models provide not only competitive accuracy but also practical performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑