vix.ing · top · new · best · stats

Uncertainty-Aware COVID-19 Detection from Imbalanced Sound Data

2021/04/05 by Xia Tong, Tong Xia, Jing Han +8 · 2 citations
Computer Science · Engineering · Medicine · #Artificial intelligence #Audio and Speech Processing (eess.AS) #COVID-19 diagnosis using AI #Computer science #Coronavirus disease 2019 (COVID-19) #Data mining #Engineering #FOS: Computer and information sciences #FOS: Electrical engineering #Machine Learning (cs.LG) #Machine learning #Medicine #Music and Audio Processing #Sampling (signal processing) #Scalability #Sensitivity (control systems) #Sound (cs.SD) #Speech and Audio Processing #cs.LG #cs.SD #eess.AS #electronic engineering #information engineering

paper · pdf · doi:10.48550/arxiv.2104.02005

published in arXiv (Cornell University) (Cornell University) · Accepted by INTERSPEECH 2021

openalex publication_date 2021/04/05 · arxiv created 2021/06/18 · arxiv updated 2021/06/22 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Recently, sound-based COVID-19 detection studies have shown great promise to achieve scalable and prompt digital pre-screening. However, there are still two unsolved issues hindering the practice. First, collected datasets for model training are often imbalanced, with a considerably smaller proportion of users tested positive, making it harder to learn representative and robust features. Second, deep learning models are generally overconfident in their predictions. Clinically, false predictions aggravate healthcare costs. Estimation of the uncertainty of screening would aid this. To handle these issues, we propose an ensemble framework where multiple deep learning models for sound-based COVID-19 detection are developed from different but balanced subsets from original data. As such, data are utilized more effectively compared to traditional up-sampling and down-sampling approaches: an AUC of 0.74 with a sensitivity of 0.68 and a specificity of 0.69 is achieved. Simultaneously, we estimate uncertainty from the disagreement across multiple models. It is shown that false predictions often yield higher uncertainty, enabling us to suggest the users with certainty higher than a threshold to repeat the audio test on their phones or to take clinical tests if digital diagnosis still fails. This study paves the way for a more robust sound-based COVID-19 automated screening system.

Citations

Related