Multi-Task Learning for Lung sound & Lung disease classification

Suma K V,Deepali Koppad,Preethi Kumar,Neha A Kantikar,Surabhi Ramesh
2024-04-05
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.
Machine Learning,Artificial Intelligence,Sound
What problem does this paper attempt to address?
The paper proposes a novel approach using multi-task learning (MTL) to simultaneously classify lung sounds and lung diseases. In this study, the authors utilized the ICBHI 2017 respiratory sound database and employed deep learning models including 2D CNN, ResNet50, MobileNet, and DenseNet to extract relevant features from lung sound recordings. Experimental results demonstrated that the MTL method employing the MobileNet model achieved an accuracy of 74% in lung sound analysis and a high accuracy of 91% in lung disease classification. Additionally, the paper utilized demographic data of patients to calculate the risk level of chronic obstructive pulmonary disease (COPD), with the random forest classifier performing the best with an accuracy of 92%. This approach helps alleviate the workload of doctors by not only diagnosing pathologies but also effectively explaining possible causes or outcomes to patients. The advantages of multi-task learning are introduced in the paper, such as improved generalization ability, data efficiency, and regularization effect. By sharing information, the model can simultaneously optimize multiple related tasks, resulting in better performance in lung sound and disease classification. In summary, the main contributions of this paper are: 1. Modification of existing deep learning models to enable multi-task learning. 2. Simultaneous classification of lung sounds and lung diseases. 3. Prediction of risk levels using demographic data. Future research directions may involve the development of remote monitoring technology, utilizing intelligent or artificial intelligence algorithms to remotely analyze lung sounds and provide more convenient and efficient healthcare services.