Application of Federated Learning in Building a Robust COVID-19 Chest X-ray Classification Model

04/22/2022
by   Amartya Bhattacharya, et al.
12

While developing artificial intelligence (AI)-based algorithms to solve problems, the amount of data plays a pivotal role - large amount of data helps the researchers and engineers to develop robust AI algorithms. In the case of building AI-based models for problems related to medical imaging, these data need to be transferred from the medical institutions where they were acquired to the organizations developing the algorithms. This movement of data involves time-consuming formalities like complying with HIPAA, GDPR, etc.There is also a risk of patients' private data getting leaked, compromising their confidentiality. One solution to these problems is using the Federated Learning framework. Federated Learning (FL) helps AI models to generalize better and create a robust AI model by using data from different sources having different distributions and data characteristics without moving all the data to a central server. In our paper, we apply the FL framework for training a deep learning model to solve a binary classification problem of predicting the presence or absence of COVID-19. We took three different sources of data and trained individual models on each source. Then we trained an FL model on the complete data and compared all the model performances. We demonstrated that the FL model performs better than the individual models. Moreover, the FL model performed at par with the model trained on all the data combined at a central server. Thus Federated Learning leads to generalized AI models without the cost of data transfer and regulatory overhead.

READ FULL TEXT

page 1

page 2

page 4

research
04/11/2022

CXR-FL: Deep Learning-based Chest X-ray Image Analysis Using Federated Learning

Federated learning enables building a shared model from multicentre data...
research
11/18/2021

Advancing COVID-19 Diagnosis with Privacy-Preserving Collaboration in Artificial Intelligence

Artificial intelligence (AI) provides a promising substitution for strea...
research
03/28/2023

A Comparative Study of Federated Learning Models for COVID-19 Detection

Deep learning is effective in diagnosing COVID-19 and requires a large a...
research
12/12/2021

Improving Performance of Federated Learning based Medical Image Analysis in Non-IID Settings using Image Augmentation

Federated Learning (FL) is a suitable solution for making use of sensiti...
research
05/01/2022

Reward Systems for Trustworthy Medical Federated Learning

Federated learning (FL) has received high interest from researchers and ...
research
11/24/2022

Collaborative Training of Medical Artificial Intelligence Models with non-uniform Labels

Artificial intelligence (AI) methods are revolutionizing medical image a...
research
03/23/2023

Federated Learning on Heterogenous Data using Chest CT

Large data have accelerated advances in AI. While it is well known that ...

Please sign up or login with your details

Forgot password? Click here to reset