COVID-19 detection using federated machine learning.

Mustafa Abdul Salam, Sanaa Taha, Mohamed Ramadan
Author Information
  1. Mustafa Abdul Salam: Artificial Intelligence Department, Faculty of Computers and Artificial intelligence, Benha University, Benha, Egypt. ORCID
  2. Sanaa Taha: Information Technology Department, Faculty of Computers and Artificial Intelligence, Cairo University, Giza, Egypt.
  3. Mohamed Ramadan: Computer Science Department, Faculty of Computers and Information- Egyptian E-Learning University, Giza, Egypt.

Abstract

The current COVID-19 pandemic threatens human life, health, and productivity. AI plays an essential role in COVID-19 case classification as we can apply machine learning models on COVID-19 case data to predict infectious cases and recovery rates using chest x-ray. Accessing patient's private data violates patient privacy and traditional machine learning model requires accessing or transferring whole data to train the model. In recent years, there has been increasing interest in federated machine learning, as it provides an effective solution for data privacy, centralized computation, and high computation power. In this paper, we studied the efficacy of federated learning versus traditional learning by developing two machine learning models (a federated learning model and a traditional machine learning model)using Keras and TensorFlow federated, we used a descriptive dataset and chest x-ray (CXR) images from COVID-19 patients. During the model training stage, we tried to identify which factors affect model prediction accuracy and loss like activation function, model optimizer, learning rate, number of rounds, and data Size, we kept recording and plotting the model loss and prediction accuracy per each training round, to identify which factors affect the model performance, and we found that softmax activation function and SGD optimizer give better prediction accuracy and loss, changing the number of rounds and learning rate has slightly effect on model prediction accuracy and prediction loss but increasing the data size did not have any effect on model prediction accuracy and prediction loss. finally, we build a comparison between the proposed models' loss, accuracy, and performance speed, the results demonstrate that the federated machine learning model has a better prediction accuracy and loss but higher performance time than the traditional machine learning model.

References

  1. Exp Ther Med. 2020 Aug;20(2):727-735 [PMID: 32742318]
  2. Wellcome Open Res. 2020 Mar 31;5:56 [PMID: 32587900]
  3. J Crohns Colitis. 2020 Sep 16;14(9):1330-1333 [PMID: 32211765]
  4. Multimed Syst. 2022;28(4):1223-1237 [PMID: 33814730]
  5. J Bone Joint Surg Am. 2020 Jul 1;102(13):e70 [PMID: 32618918]
  6. Biocybern Biomed Eng. 2021 Jul-Sep;41(3):867-879 [PMID: 34108787]
  7. IEEE Internet Things J. 2021 Feb 04;8(21):15884-15891 [PMID: 35663640]
  8. Appl Intell (Dordr). 2021;51(5):2777-2789 [PMID: 34764562]
  9. Arch Comput Methods Eng. 2021;28(4):2645-2653 [PMID: 32837183]
  10. Chaos Solitons Fractals. 2020 Nov;140:110203 [PMID: 32834663]
  11. Ing Rech Biomed. 2022 Apr;43(2):114-119 [PMID: 32837679]
  12. Biomed Signal Process Control. 2020 Sep;62:102149 [PMID: 32834831]
  13. Chaos Solitons Fractals. 2020 Sep;138:109944 [PMID: 32536759]
  14. Int J Inf Technol. 2020;12(3):731-739 [PMID: 32838125]
  15. Sustain Cities Soc. 2021 Feb;65:102589 [PMID: 33169099]
  16. Pattern Recognit Lett. 2020 Oct;138:638-643 [PMID: 32958971]

MeSH Term

COVID-19
Computer Simulation
Datasets as Topic
Humans
Image Processing, Computer-Assisted
Machine Learning
Thorax
Tomography, X-Ray Computed

Word Cloud

Created with Highcharts 10.0.0learningmodelmachinepredictionaccuracylossdatafederatedCOVID-19traditionalusingperformancecasemodelschestx-rayprivacyincreasingcomputationtrainingidentifyfactorsaffectactivationfunctionoptimizerratenumberroundsbettereffectcurrentpandemicthreatenshumanlifehealthproductivityAIplaysessentialroleclassificationcanapplypredictinfectiouscasesrecoveryratesAccessingpatient'sprivateviolatespatientrequiresaccessingtransferringwholetrainrecentyearsinterestprovideseffectivesolutioncentralizedhighpowerpaperstudiedefficacyversusdevelopingtwoKerasTensorFlowuseddescriptivedatasetCXRimagespatientsstagetriedlikeSizekeptrecordingplottingperroundfoundsoftmaxSGDgivechangingslightlysizefinallybuildcomparisonproposedmodels'speedresultsdemonstratehighertimedetection

Similar Articles

Cited By