NVIDIA FLARE: Federated Learning from Simulation to Real-World
Holger R. Roth,Yan Cheng,Yuhong Wen,Isaac Yang,Ziyue Xu,Yuan-Ting Hsieh,Kristopher Kersten,Ahmed Harouni,Can Zhao,Kevin Lu,Zhihong Zhang,Wenqi Li,Andriy Myronenko,Dong Yang,Sean Yang,Nicola Rieke,Abood Quraini,Chester Chen,Daguang Xu,Nic Ma,Prerna Dogra,Mona Flores,Andrew Feng
DOI: https://doi.org/10.48550/arXiv.2210.13291
2023-04-29
Abstract:Federated learning (FL) enables building robust and generalizable AI models by leveraging diverse datasets from multiple collaborators without centralizing the data. We created NVIDIA FLARE as an open-source software development kit (SDK) to make it easier for data scientists to use FL in their research and real-world applications. The SDK includes solutions for state-of-the-art FL algorithms and federated machine learning approaches, which facilitate building workflows for distributed learning across enterprises and enable platform developers to create a secure, privacy-preserving offering for multiparty collaboration utilizing homomorphic encryption or differential privacy. The SDK is a lightweight, flexible, and scalable Python package. It allows researchers to apply their data science workflows in any training libraries (PyTorch, TensorFlow, XGBoost, or even NumPy) in real-world FL settings. This paper introduces the key design principles of NVFlare and illustrates some use cases (e.g., COVID analysis) with customizable FL workflows that implement different privacy-preserving algorithms.
Code is available at <a class="link-external link-https" href="https://github.com/NVIDIA/NVFlare" rel="external noopener nofollow">this https URL</a>.
Machine Learning,Artificial Intelligence,Computer Vision and Pattern Recognition,Networking and Internet Architecture,Software Engineering