Skip to main navigation Skip to search Skip to main content

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
  • Xi'an Jiaotong-Liverpool University
  • Shanghai Jiao Tong University
  • East China Normal University

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