Evaluation of Deep Learning Methods for Pulmonary Disease Classification.

Abstract

Introduction: Driven by environmental pollution and the rise in infectious diseases, the increasing prevalence of lung conditions demands advancements in diagnostic techniques.

Materials and methods: This study explores the use of various features, such as spectrograms, chromograms, and Mel Frequency Cepstral Coefficients (MFCC), to extract crucial information from auscultation recordings. It addresses challenges through filter-based audio enhancement methods. The primary goal is to improve disease detection accuracy by leveraging convolutional neural networks (CNNs) for feature extraction and dense neural networks for classification.

Results: While deep learning models like CNNs and Recurrent Neural Network (RNN) outperform traditional machine learning models such as Sequence Vector Machine, K-Nearest Neighbours (KNN) and random forest with accuracies ranging from 70% to 85%. The combination of CNN, RNN, and long short-term memory achieved an accuracy of 88%. By integrating MFCC, Chroma Short-Term Fourier Transform (STFT), and spectrogram features with a CNN-based classifier, the proposed multi-feature deep learning model achieved the highest accuracy of 92%, surpassing all other methods.

Discussion: The study effectively addresses key issues, including the overrepresentation of Chronic Obstructive Pulmonary Disease (COPD) samples over Lower Respiratory Tract Infections (LRTI) and Upper Respiratory Tract Infections (URTI) which hampers generalization across test audio samples.

Conclusion: The proposed methodology caters common challenges like background noise in recordings, and the limited and imbalanced nature of datasets. These findings pave the way for enhanced clinical applications, showcasing the transformative potential of multi-feature deep learning methods in the classification of pulmonary diseases.