Enhancing multiclass COVID-19 prediction with ESN-MDFS: Extreme smart network using mean dropout feature selection technique.

Journal: PloS one
Published Date:

Abstract

Deep learning and artificial intelligence offer promising tools for improving the accuracy and efficiency of diagnosing various lung conditions using portable chest x-rays (CXRs). This study explores this potential by leveraging a large dataset containing over 6,000 CXR images from publicly available sources. These images encompass COVID-19 cases, normal cases, and patients with viral or bacterial pneumonia. The research proposes a novel approach called "Enhancing COVID Prediction with ESN-MDFS" that utilizes a combination of an Extreme Smart Network (ESN) and a Mean Dropout Feature Selection Technique (MDFS). This study aimed to enhance multi-class lung condition detection in portable chest X-rays by combining static texture features with dynamic deep learning features extracted from a pre-trained VGG-16 model. To optimize performance, preprocessing, data imbalance, and hyperparameter tuning were meticulously addressed. The proposed ESN-MDFS model achieved a peak accuracy of 96.18% with an AUC of 1.00 in a six-fold cross-validation. Our findings demonstrate the model's superior ability to differentiate between COVID-19, bacterial pneumonia, viral pneumonia, and normal conditions, promising significant advancements in diagnostic accuracy and efficiency.

Authors

  • Saghir Ahmed
    Department of Computer Science, COMSATS University, Islamabad Capital Territory, Islamabad, Pakistan.
  • Basit Raza
    Medical Imaging and Diagnostics (MID) Lab, National Centre of Artificial Intelligence (NCAI), Department of Computer Science, COMSATS University Islamabad (CUI), Islamabad, 45550, Pakistan. basit.raza@comsats.edu.pk.
  • Lal Hussain
    Department of Computer Sciences & Information Technology, University of Azad Jammu and Kashmir, City Campus 13100, Muzaffarabad, Azad Kashmir, Pakistan.
  • Touseef Sadiq
    Centre for Artificial Intelligence Research, Department of Information and Communication Technology, University of Agder, Jon Lilletuns vei 9, 4879 Grimstad, Norway.
  • Ashit Kumar Dutta
    Department of Computer Science and Information Systems, College of Applied Sciences, Almaarefa University, Riyadh 11597, Saudi Arabia.