CoughSense: Five-Class Respiratory Disease Classification via Whisper Encoder Fine-Tuning and Dual-Encoder Cross-Attention Fusion with Balanced Contrastive Learning
Not on the map. Why?
Abstract (arXiv)
Automated cough analysis offers a path to low-cost respiratory screening, but most existing work stops at binary COVID-19 detection. A practical tool needs to tell apart several respiratory conditions from one cough recording on a consumer smartphone. We present CoughSense, a system that sorts cough recordings into five classes. These are healthy, COVID-19, asthma or respiratory condition, bronchitis, and pneumonia. We aggregated 18,301 recordings from four public datasets (Coswara, CoughVID, Virufy, and the West China Hospital Pediatric Cough Dataset) and used the OpenAI Whisper encoder as a pretrained backbone for cough disease classification. The main contribution is active-frame QKV attention pooling, which restricts attention to the first 200 of 1500 encoder tokens. This avoids the silence-dilution problem that arises because a 3-second cough fills only 150 tokens of Whisper's 30-second input window. Other training parts handle the 19 to 1 class imbalance and the four-dataset domain shift. These include WeightedRandomSampler, SpecAugment, Balanced Mixup with forced minority pairing, a supervised contrastive auxiliary loss, FiLM symptom conditioning, and gradient-reversal domain adaptation. A dual-encoder model fuses Whisper with the OPERA-CT respiratory foundation model through cross-attention. CoughSense (Whisper-tiny, 8.6M parameters) reached 82.3 percent balanced accuracy on five-fold cross-validation (macro-F1 of 0.817, AUC of 0.941). It beat an ImageNet-pretrained EfficientNet-B2 by 11.1 points and a ViT trained from scratch by 29.6 points. All five classes passed 74 percent recall and four of five passed 80 percent. The dual-encoder model reached 85.4 percent balanced accuracy. Active-frame pooling is the largest single contributor across all ablation components at 5.1 points, which should help any short-audio task using Whisper as a backbone.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 21 of its 32 references that are in the map. First posted on arXiv 2 Jun 2026.
Builds on
21 of 32 references.
References · 21
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2022Robust Speech Recognition via Large-Scale Weak Supervisioncited by 4,057
- 2020wav2vec 2.0: A Framework for Self-Supervised Learning of Speech Representationscited by 3,905
- 2015Domain-Adversarial Training of Neural Networkscited by 3,044
- 2020Supervised Contrastive Learningcited by 2,829
- 2021HuBERT: Self-Supervised Speech Representation Learning by Masked Prediction of Hidden Unitscited by 2,342
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
- 2019SpecAugment: A Simple Data Augmentation Method for Automatic Speech Recognitioncited by 1,380
- 2022Large-scale Contrastive Language-Audio Pretraining with Feature Fusion and Keyword-to-Caption Augmentationcited by 719
- 2019PANNs: Large-Scale Pretrained Audio Neural Networks for Audio Pattern Recognitioncited by 574
9 more
- 2022BEATs: Audio Pre-Training with Acoustic Tokenizerscited by 388
- 2022Masked Autoencoders that Listencited by 273
- 2020Exploring Automatic Diagnosis of COVID-19 from Crowdsourced Respiratory Sound Datacited by 60
- 2020The COUGHVID crowdsourcing dataset: A corpus for the study of large-scale cough analysis algorithmscited by 59
- 2020Coswara -- A Database of Breathing, Cough, and Voice Sounds for COVID-19 Diagnosiscited by 58
- 2021Balanced-MixUp for Highly Imbalanced Medical Image Classificationcited by 46
- 2024Towards Open Respiratory Acoustic Foundation Models: Pretraining and Benchmarkingcited by 23
- 2020Virufy: Global Applicability of Crowdsourced and Clinical Datasets for AI Detection of COVID-19 from Coughcited by 15
- 2026Whisper-AuT: Domain-Adapted Audio Encoder for Efficient Audio-LLM Trainingcited by 2
Built on by
Not cited yet.