Chest X-ray Classification for the Detection of COVID-19 Using Deep Learning Techniques

Sensors

Fawad Ahmed¹  Muhammad Zia Ur Rehman²  Faisal Abdulaziz Alfouzan³  Nouf M. Alzahrani⁴  Jawad Ahmad⁵  Ejaz Khan⁶ 

[1] Department of Cyber Security, Pakistan Navy Engineering College, NUST, Karachi 75350, Pakistan;Department of Electrical Engineering, HITEC University, Taxila 47080, Pakistan;Department of Forensic Sciences, College of Criminal Justice, Naif Arab University for Security Sciences (NAUSS), Riyadh 14812, Saudi Arabia;Information Technology Department, Collage of Computer Science and Information Technology, Al Baha University, Al Bahah 65731, Saudi Arabia;School of Computing, Edinburgh Napier University, Edinburgh EH10 5DT, UK;School of Engineering, RMIT University, Melbourne 3000, Australia;

关键词: COVID-19;    deep learning;    classification;    transfer learning;    chest X-rays;   

DOI  :  10.3390/s22031211

来源: DOAJ

【 摘 要 】

Recent technological developments pave the path for deep learning-based techniques to be used in almost every domain of life. The precision of deep learning techniques make it possible for these to be used in the medical field for the classification and detection of various diseases. Recently, the coronavirus (COVID-19) pandemic has put a lot of pressure on the health system all around the world. The diagnosis of COVID-19 is possible by PCR testing and medical imagining. Since COVID-19 is highly contagious, diagnosis using chest X-ray is considered safe in various situations. In this study, a deep learning-based technique is proposed to classify COVID-19 infection from other non-COVID-19 infections. To classify COVID-19, three different pre-trained models named EfficientNetB1, NasNetMobile and MobileNetV2 are used. The augmented dataset is used for training deep learning models while two different training strategies have been used for classification. In this study, not only are the deep learning model fine-tuned but also the hyperparameters are fine-tuned, which significantly improves the performance of the fine-tuned deep learning models. Moreover, the classification head is regularized to improve the performance. For the evaluation of the proposed techniques, several performance parameters are used to gauge the performance. EfficientNetB1 with regularized classification head outperforms the other models. The proposed technique successfully classifies four classes that include COVID-19, viral pneumonia, lung opacity, and normal, with an accuracy of 96.13%. The proposed technique shows superiority in terms of accuracy when compared with recent techniques present in the literature.

【 授权许可】

Unknown