OpenFL: An open-source framework for Federated Learning
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
DOI: https://doi.org/10.1088/1361-6560/ac97d9
2021-05-14
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 <a class="link-external link-https" href="https://github.com/intel/openfl" rel="external noopener nofollow">this https URL</a>) 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.
Machine Learning,Distributed, Parallel, and Cluster Computing