NVIDIA FLARE: Federated Learning from Simulation to Real-World

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 https://github.com/NVIDIA/NVFlare.

Accepted at the International Workshop on Federated Learning, NeurIPS 2022, New Orleans, USA (https://federated-learning.org/fl-neurips-2022); Revised version v2: added Key Components list, system metrics for homomorphic encryption experiment; Extended v3 for journal submission

Cited by in corpus (10)

• Do Gradient Inversion Attacks Make Federated Learning Unsafe?
• The Federation Strikes Back: A Survey of Federated Learning Privacy Attacks, Defenses, Applications, and Policy Landscape
• Personalized and privacy-preserving federated heterogeneous medical image analysis with PPPML-HMI
• Decentralized Learning Made Easy with DecentralizePy
• Collaborative Distributed Machine Learning
• A Federated Learning Benchmark on Tabular Data: Comparing Tree-Based Models and Neural Networks
• A Federated Random Forest Solution for Secure Distributed Machine Learning
• Histogram-Based Federated XGBoost using Minimal Variance Sampling for Federated Tabular Data
• FL-APU: A Software Architecture to Ease Practical Implementation of Cross-Silo Federated Learning
• Stalactite: Toolbox for Fast Prototyping of Vertical Federated Learning Systems