Development of a classifier with analysis of feature selection methods for COVID-19 diagnosis

Abstract

Purpose

The COVID-19 pandemic situation is increasing day by day and has affected the lifestyle and economy worldwide. Due to the absence of specific treatment, the only way to control a pandemic is by stopping its spread. Early identification of affected persons is urgently in demand. Diagnostic methods applied in hospitals are time-consuming, which delay the identification of positive patients. This study aims to develop machine learning-based diagnosis model which can predict positive cases and helps in decision-making.

Design/methodology/approach

In this research, the authors have developed a diagnosis model to check coronavirus positivity based on an artificial neural network. The authors have trained the model with clinically assessed symptoms, patient-reported symptoms, other medical histories and exposure data of the person. The authors have explored filter-based feature selection methods such as Chi2, ANOVA F-score and Mutual Information for improving performance of a classification model. Metrics used to evaluate performance of the model are accuracy, precision, sensitivity and F1-score.

Findings

The authors got highest classification performance with model trained with features ranked according to ANOVA FS method. Highest scores for accuracy, sensitivity, precision and F1-score of predictions are 0.93, 0.99, 0.94 and 0.93, respectively. The study reveals that most relevant predictors for COVID-19 diagnosis are sob severity, cough severity, sob presence, cough presence, fatigue and number of days since symptom onset.

Originality/value

Treatment for COVID-19 is not available to date. The best way to control this pandemic is the isolation of positive persons. It is very much necessary to identify positive persons at an early stage. RT-PCR test used to check COVID-19 positivity is the time-consuming, expensive and laborious method. Current diagnosis methods used in hospital demand more medical resources with increasing cases of coronavirus that introduce shortage of resources. The developed model provides solution to the problem cheaper and faster decreases the immediate need for medical resources and helps in decision-making.

Citation

Chauhan, H., Modi, K. and Shrivastava, S. (2022), "Development of a classifier with analysis of feature selection methods for COVID-19 diagnosis", World Journal of Engineering, Vol. 19 No. 1, pp. 49-57. https://doi.org/10.1108/WJE-10-2020-0537