Why ML-based cough models do not generalize: a systematic cross-dataset evaluation for tuberculosis screening
Not on the map. Why?
Abstract (arXiv)
Cough acoustics are promising for non-invasive tuberculosis (TB) screening, yet whether machine learning (ML) models capture disease-related acoustics or artifacts of data collection remains unresolved. We evaluated the cross-dataset generalizability of classical ML and deep learning (DL) cough-based TB classifiers across three independent datasets. Despite moderate within-dataset performance (ROC-AUC up to 0.755 ± 0.056), both pipelines fail to generalize, with external performance frequently below 0.6, indicating a possible limitation of the data. We further observed audio representations are organized by recording device and dataset rather than TB status, predicted TB probability tracks country-level prevalence in CODA, and device mismatch degrades transfer while device-diverse training improves it. Additionally, a clinical-variable baseline generalizes more consistently (ROC-AUC 0.655 - 0.711), indicating acquisition-specific variability is a stronger driver of poor generalizability than population shift. High within-dataset performance is not enough. External validation is essential before cough-based TB models are clinically ready.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 48 references that are in the map. First posted on arXiv 26 Aug 2026.
Builds on
22 of 48 references.
References · 22
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Explaining and Harnessing Adversarial Examplescited by 7,287
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2015Domain-Adversarial Training of Neural Networkscited by 3,023
- 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 762
- 2016CNN Architectures for Large-Scale Audio Classificationcited by 701
- 2013Domain Generalization via Invariant Feature Representationcited by 622
- 2014Deep Domain Confusion: Maximizing for Domain Invariancecited by 597
- 2018Generalizing Across Domains via Cross-Gradient Trainingcited by 268
- 2015Scatter Component Analysis: A Unified Framework for Domain Adaptation and Domain Generalizationcited by 96
- 2021MixStyle Neural Networks for Domain Generalization and Adaptationcited by 64
- 2020The COUGHVID crowdsourcing dataset: A corpus for the study of large-scale cough analysis algorithmscited by 59
10 more
- 2022Detecting Shortcut Learning for Fair Medical AI using Shortcut Testingcited by 52
- 2024Towards Open Respiratory Acoustic Foundation Models: Pretraining and Benchmarkingcited by 23
- 2021Automatic Cough Classification for Tuberculosis Screening in a Real-World Environmentcited by 17
- 2024HeAR -- Health Acoustic Representationscited by 15
- 2022A Semi-Supervised Algorithm for Improving the Consistency of Crowdsourced Datasets: The COVID-19 Case Study on Respiratory Disorder Classificationcited by 7
- 2025Transformation of audio embeddings into interpretable, concept-based representationscited by 5
- 2023Optimizing Audio Augmentations for Contrastive Learning of Health-Related Acoustic Signalscited by 2
- 2024DG-SED: Domain Generalization for Sound Event Detection with Heterogeneous Training Datacited by 2
- 2024Acoustical Features as Knee Health Biomarkers: A Critical Analysiscited by 1
- 2026Tuberculosis Screening from Cough Audio: Baseline Models, Clinical Variables, and Uncertainty Quantificationcited by 1
Built on by
Not cited yet.