arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2505.08085cs.LG

一种用于安全分布式机器学习的联邦随机森林解决方案

A Federated Random Forest Solution for Secure Distributed Machine Learning

IEETA/DETI, LASI University of Aveiro(IEETA/DETI实验室,阿维罗大学)

Alexandre Cotorobai, Jorge Miguel Silva, Jose Luis Oliveira

AI总结:

本文提出基于PySyft的联邦随机森林框架,通过加权平均、增量学习和本地评估,在保护隐私的前提下实现接近集中式精度的分布式树模型训练。

摘要:

隐私和监管障碍常常阻碍集中式机器学习解决方案,尤其是在医疗保健等数据无法自由共享的领域。联邦学习已成为应对这些问题的强大范式;然而,现有框架主要支持基于梯度的模型,在更具可解释性的树基方法方面留下了空白。本文介绍了一种用于随机森林分类器的联邦学习框架,该框架能够保护数据隐私,并在分布式环境中提供稳健的性能。通过利用PySyft进行安全的、具有隐私意识的计算,我们的方法使多个机构能够在本地存储的数据上协作训练随机森林模型,而不会暴露敏感信息。该框架支持加权模型平均以应对不同的数据分布,支持增量学习以逐步优化模型,并支持本地评估以衡量异构数据集上的性能。在两个真实世界医疗保健基准上的实验表明,联邦方法保持了具有竞争力的预测准确率——与集中式方法相比最大差距在9%以内——同时满足严格的隐私要求。这些发现强调了树基联邦学习在因监管、竞争或技术限制而无法集中数据的场景中的可行性。所提出的解决方案填补了现有联邦学习库中的一个显著空白,为既要求透明度又要求可靠性能的安全分布式机器学习任务提供了一种适应性强的工具。该工具可在https://github.com/ieeta-pt/fed_rf获取。

英文摘要:

Privacy and regulatory barriers often hinder centralized machine learning solutions, particularly in sectors like healthcare where data cannot be freely shared. Federated learning has emerged as a powerful paradigm to address these concerns; however, existing frameworks primarily support gradient-based models, leaving a gap for more interpretable, tree-based approaches. This paper introduces a federated learning framework for Random Forest classifiers that preserves data privacy and provides robust performance in distributed settings. By leveraging PySyft for secure, privacy-aware computation, our method enables multiple institutions to collaboratively train Random Forest models on locally stored data without exposing sensitive information. The framework supports weighted model averaging to account for varying data distributions, incremental learning to progressively refine models, and local evaluation to assess performance across heterogeneous datasets. Experiments on two real-world healthcare benchmarks demonstrate that the federated approach maintains competitive predictive accuracy - within a maximum 9\% margin of centralized methods - while satisfying stringent privacy requirements. These findings underscore the viability of tree-based federated learning for scenarios where data cannot be centralized due to regulatory, competitive, or technical constraints. The proposed solution addresses a notable gap in existing federated learning libraries, offering an adaptable tool for secure distributed machine learning tasks that demand both transparency and reliable performance. The tool is available at https://github.com/ieeta-pt/fed_rf.

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

↑