OpenFL: the open federated learning library.

Journal: Physics in medicine and biology
Published Date:

Abstract

Federated learning (FL) is a computational paradigm that enables organizations to collaborate on machine learning (ML) and deep learning (DL) projects without sharing sensitive data, such as patient records, financial data, or classified secrets.Open federated learning (OpenFL) framework is an open-source python-based tool for training ML/DL algorithms using the data-private collaborative learning paradigm of FL, irrespective of the use case. OpenFL works with training pipelines built with both TensorFlow and PyTorch, and can be easily extended to other ML and DL frameworks.In this manuscript, we present OpenFL and summarize its motivation and development characteristics, with the intention of facilitating its application to existing ML/DL model training in a production environment. We further provide recommendations to secure a federation using trusted execution environments to ensure explicit model security and integrity, as well as maintain data confidentiality. Finally, we describe the first real-world healthcare federations that use the OpenFL library, and highlight how it can be applied to other non-healthcare use cases.The OpenFL library is designed for real world scalability, trusted execution, and also prioritizes easy migration of centralized ML models into a federated training pipeline. Although OpenFL's initial use case was in healthcare, it is applicable beyond this domain and is now reaching wider adoption both in research and production settings. The tool is open-sourced atgithub.com/intel/openfl.

Authors

  • Patrick Foley
    Intel Corporation, Santa Clara, CA 95052, United States of America.
  • Micah J Sheller
    Intel, Santa Clara, CA, USA.
  • Brandon Edwards
    Intel Corporation, Santa Clara, CA 95052, United States of America.
  • Sarthak Pati
    Perelman School of Medicine, Philadelphia, PA, USA.
  • Walter Riviera
    Intel Corporation, Santa Clara, CA 95052, United States of America.
  • Mansi Sharma
    Intel Corporation, Santa Clara, CA 95052, United States of America.
  • Prakash Narayana Moorthy
    Intel, Santa Clara, CA, USA.
  • Shih-Han Wang
    Intel Corporation, Santa Clara, CA 95052, United States of America.
  • Jason Martin
    Intel Corporation, Santa Clara, CA 95052, United States of America.
  • Parsa Mirhaji
    Albert Einstein College of Medicine, 1300 Morris Park Ave, Bronx, NY 10461, United States of America.
  • Prashant Shah
    Intel, Santa Clara, CA, USA.
  • Spyridon Bakas
    Perelman School of Medicine, Philadelphia, PA, USA.