OpenFL: An open-source framework for Federated Learning
Authors:
G Anthony Reina,
Alexey Gruzdev,
Patrick Foley,
Olga Perepelkina,
Mansi Sharma,
Igor Davidyuk,
Ilya Trushkin,
Maksim Radionov,
Aleksandr Mokrov,
Dmitry Agapov,
Jason Martin,
Brandon Edwards,
Micah J. Sheller,
Sarthak Pati,
Prakash Narayana Moorthy,
Shih-han Wang,
Prashant Shah,
Spyridon Bakas
Abstract:
Federated learning (FL) is a computational paradigm that enables organizations to collaborate on machine learning (ML) projects without sharing sensitive data, such as, patient records, financial data, or classified secrets. Open Federated Learning (OpenFL https://github.com/intel/openfl) is an open-source framework for training ML algorithms using the data-private collaborative learning paradigm…
▽ More
Federated learning (FL) is a computational paradigm that enables organizations to collaborate on machine learning (ML) projects without sharing sensitive data, such as, patient records, financial data, or classified secrets. Open Federated Learning (OpenFL https://github.com/intel/openfl) is an open-source framework for training ML algorithms using the data-private collaborative learning paradigm of FL. OpenFL works with training pipelines built with both TensorFlow and PyTorch, and can be easily extended to other ML and deep learning frameworks. Here, we summarize the motivation and development characteristics of OpenFL, with the intention of facilitating its application to existing ML model training in a production environment. Finally, we describe the first use of the OpenFL framework to train consensus ML models in a consortium of international healthcare organizations, as well as how it facilitates the first computational competition on FL.
△ Less
Submitted 13 May, 2021;
originally announced May 2021.