Multi-Task Learning for Lung sound & Lung disease classification

TL;DR

The paper tackles automatic, simultaneous classification of lung sounds and lung diseases using a multi-task learning framework. It evaluates four deep CNN backbones (2D CNN, ResNet50, MobileNet, DenseNet) on MFCC features extracted from the ICBHI 2017 respiratory sound dataset, with MobileNet achieving the best joint performance (74% LS, 91% LD). In addition, the authors compute COPD risk levels from patient demographics using LR, SVM, and RF, with Random Forest achieving 92% accuracy. The results demonstrate that MTL can jointly predict sound and disease classes with improved training efficiency, and the work envisions real-time deployment in intelligent stethoscopes and multi-modal clinical workflows.

Abstract

In recent years, advancements in deep learning techniques have considerably enhanced the efficiency and accuracy of medical diagnostics. In this work, a novel approach using multi-task learning (MTL) for the simultaneous classification of lung sounds and lung diseases is proposed. Our proposed model leverages MTL with four different deep learning models such as 2D CNN, ResNet50, MobileNet and Densenet to extract relevant features from the lung sound recordings. The ICBHI 2017 Respiratory Sound Database was employed in the current study. The MTL for MobileNet model performed better than the other models considered, with an accuracy of74\% for lung sound analysis and 91\% for lung diseases classification. Results of the experimentation demonstrate the efficacy of our approach in classifying both lung sounds and lung diseases concurrently. In this study,using the demographic data of the patients from the database, risk level computation for Chronic Obstructive Pulmonary Disease is also carried out. For this computation, three machine learning algorithms namely Logistic Regression, SVM and Random Forest classifierswere employed. Among these ML algorithms, the Random Forest classifier had the highest accuracy of 92\%.This work helps in considerably reducing the physician's burden of not just diagnosing the pathology but also effectively communicating to the patient about the possible causes or outcomes.

Figures (5)

• Figure 1: General Methodology of Multitask Learning
• Figure 6: Classification report for lung sounds
• Figure 7: ROC Curve for Lung Sound Classification
• Figure 8: ROC Curve for Lung Sound Classification
• Figure 9: Accuracy and Loss Curve