An Ensemble-based Multi-Criteria Decision Making Method for COVID-19 Cough Classification

Abstract

The objectives of this research are analysing the performance of the state-of-the-art machine learning techniques for classifying COVID-19 from cough sound and identifying the model(s) that consistently perform well across different cough datasets. Different performance evaluation metrics (such as precision, sensitivity, specificity, AUC, accuracy, etc.) make it difficult to select the best performance model. To address this issue, in this paper, we propose an ensemble-based multi-criteria decision making (MCDM) method for selecting top performance machine learning technique(s) for COVID-19 cough classification. We use four cough datasets, namely Cambridge, Coswara, Virufy, and NoCoCoDa to verify the proposed method. At first, our proposed method uses the audio features of cough samples and then applies machine learning (ML) techniques to classify them as COVID-19 or non-COVID-19. Then, we consider a multi-criteria decision-making (MCDM) method that combines ensemble technologies (i.e., soft and hard) to select the best model. In MCDM, we use the technique for order preference by similarity to ideal solution (TOPSIS) for ranking purposes, while entropy is applied to calculate evaluation criteria weights. In addition, we apply the feature reduction process through recursive feature elimination with cross-validation under different estimators. The results of our empirical evaluations show that the proposed method outperforms the state-of-the-art models.

arXiv comment21 pages, 6 figures

References in corpus (11)

• SMOTE: Synthetic Minority Over-sampling Technique
• Exploring Automatic Diagnosis of COVID-19 from Crowdsourced Respiratory Sound Data
• Coswara -- A Database of Breathing, Cough, and Voice Sounds for COVID-19 Diagnosis
• The COUGHVID crowdsourcing dataset: A corpus for the study of large-scale cough analysis algorithms
• COVID-19 Cough Classification using Machine Learning and Global Smartphone Recordings
• A Generic Deep Learning Based Cough Analysis System from Clinically Validated Samples for Point-of-Need Covid-19 Test and Severity Levels
• Cough Against COVID: Evidence of COVID-19 Signature in Cough Sounds
• A Machine Hearing System for Robust Cough Detection Based on a High-Level Representation of Band-Specific Audio Features
• Virufy: Global Applicability of Crowdsourced and Clinical Datasets for AI Detection of COVID-19 from Cough
• Detecting COVID-19 from Breathing and Coughing Sounds using Deep Neural Networks
• Virufy: A Multi-Branch Deep Learning Network for Automated Detection of COVID-19