Skip to content

Federated Learning for COVID-19 Detection with Generative Adversarial Networks

Notifications You must be signed in to change notification settings

Michelle607/FL-GAN_COVID

 
 

Repository files navigation

FL-GAN_COVID

This is source code for the paper: "Federated Learning for COVID-19 Detection with Generative Adversarial Networks in Edge Cloud Computing", published at the IEEE Internet of Things Journal, Nov. 2021 (https://ieeexplore.ieee.org/abstract/document/9580478)

Requirements:

python >=3.5

tensorflow >= 2.6

pytorch >= 0.4

How to run this code:

Install all required libraries and then is ready to run the code. It is recommended to run the standalone GAN code "COVID_GAN3.py" first. Then run the FL-GAN code "Server_COVID.py". Here, it is set for only 1 server and random 50 hospitals as hospital clients for training the COVID X-ray data. Then, use the Classifer Model for COVID detection. - "CNN_COVID_Classification.py" - "EFFICIENTNET_COVID_Classification.py" - "ENSEMBLE_B0B7_Classification.py"

COVID-19 datasets:

I used several open datasets.

Citation:

The paper is available at https://arxiv.org/abs/2110.07136 and the authors using this code should cite as: @article{nguyen2021federatedcovid,

title={Federated learning for covid-19 detection with generative adversarial networks in edge cloud computing},

author={Nguyen, Dinh C and Ding, Ming and Pathirana, Pubudu N and Seneviratne, Aruna and Zomaya, Albert Y},

journal={IEEE Internet of Things Journal},

year={2021}, }

About

Federated Learning for COVID-19 Detection with Generative Adversarial Networks

Topics

Resources

Stars

Watchers

Forks

Languages

  • Python 100.0%