Comparative analysis of feature selection techniques for COVID-19 dataset
In the context of early disease detection, machine learning (ML) has emerged as a vital tool. Feature selection (FS) algorithms play a crucial role in ensuring the accuracy of predictive models by identifying the most influential variables. This study, focusing on a retrospective cohort of 4778 COVID-19 patients from Iran, explores the performance of various FS methods, including filter, embedded, and hybrid approaches, in predicting mortality outcomes. The researchers leveraged 115 routine clinical, laboratory, and demographic features and employed 13 ML models to assess the effectiveness of these FS methods based on classification accuracy, predictive accuracy, and statistical tests. The results indicate that a Hybrid Boruta-VI model combined with the Random Forest algorithm demonstrated superior performance, achieving an accuracy of 0.89, an F1 score of 0.76, and an AUC value of 0.95 on test data. Key variables identified as important predictors of adverse outcomes include age, oxygen saturation levels, albumin levels, neutrophil counts, platelet levels, and markers of kidney function. These findings highlight the potential of advanced FS techniques and ML models in enhancing early disease detection and informing clinical decision-making.
Authors
- Mohammad Reza Zali (ORCID: https://orcid.org/0000-0002-9027-4560)
- Seyed Saeed Hashemi Nazari (ORCID: https://orcid.org/0000-0002-0883-3408)
- Mohamad Amin Pourhoseingholi (ORCID: https://orcid.org/0000-0002-0121-8031)
- Kaveh Kavousi (ORCID: https://orcid.org/0000-0002-1906-3912)
- Farideh Mohtasham
Institutions
- University of Nottingham (GB)
- University of Tehran (IR)
- Institute of Biochemistry (HU)
- Research Institute for Endocrine Sciences (IR)
- Nottingham Biomedical Research Centre (GB)
- Shahid Beheshti University of Medical Sciences (IR)
Publication Details
- Journal
- Scientific Reports
- Published
- 2024-08-11
- DOI
- https://doi.org/10.1038/s41598-024-69209-6
- Citations
- 16
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 6.47