Predicting and Visualizing Covid-19 Identification by a Hybrid Machine Learning and Pre-trained Model

Ruilin Cai, Weimei Li, Han Lu, Jiangang Chen, Dongdong Zou, Xin Chen, Yongchao Pan, Liang Feng*, Jun Qi*

*Corresponding author for this work

Research output: Chapter in Book or Report/Conference proceedingConference Proceedingpeer-review

Abstract

The emergence of the Omicron variant in Shanghai in 2022 has highlighted the need for effective diagnostic tools for COVID-19. Recent studies have indicated that cough sounds generate distinctive features capable of distinguishing infected individuals from healthy ones. This study proposes a hybrid approach that combines machine learning and pre-trained models, utilizing audio features such as Mel-frequency cepstral coefficients and waveforms as inputs to train both types of models for accurate identification of COVID-19 patients and healthy individuals. The study employs a dataset collected from hospitals in Shanghai, China, comprising 78 participants, including COVID-19 positive and negative individuals. The proposed method demonstrates superior performance in diagnosing COVID-19 compared to existing mainstream machine learning algorithms. Furthermore, decisive important audio features for the COVID-19 positive classifier are identified via SHAP values for feature importance. Overall, the proposed approach achieves excellent diagnostic accuracy for COVID-19, outperforming current mainstream machine learning methods. With its multiple strengths in performance, speed, and usability, this algorithm shows great promise in enabling large-scale screening and aiding the containment of future widespread infections.

Original languageEnglish
Title of host publicationProceedings - 2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024
EditorsMario Cannataro, Huiru Zheng, Lin Gao, Jianlin Cheng, Joao Luis de Miranda, Ester Zumpano, Xiaohua Hu, Young-Rae Cho, Taesung Park
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages5936-5943
Number of pages8
ISBN (Electronic)9798350386226
DOIs
Publication statusPublished - 2024
Event2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024 - Lisbon, Portugal
Duration: 3 Dec 20246 Dec 2024

Publication series

NameProceedings - 2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024

Conference

Conference2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024
Country/TerritoryPortugal
CityLisbon
Period3/12/246/12/24

Keywords

  • Audio classification
  • COVID-19 diagnosis
  • Machine learning
  • Pre trained model

Fingerprint

Dive into the research topics of 'Predicting and Visualizing Covid-19 Identification by a Hybrid Machine Learning and Pre-trained Model'. Together they form a unique fingerprint.

Cite this