Previously submitted to: JMIR Formative Research (no longer under consideration since Apr 25, 2024)
Date Submitted: Nov 22, 2023
Warning: This is an author submission that is not peer-reviewed or edited. Preprints - unless they show as "accepted" - should not be relied on to guide clinical practice or health-related behavior and should not be reported in news media as established information.
Research on the Cough-Related Disease Classification and Recognition Method Combining Attention Mechanism with Residual Neural Networks
ABSTRACT
Background:
Coughing, a common clinical symptom, is a protective respiratory reflex associated with various respiratory disorders. Due to its diverse acoustic characteristics resulting from different underlying causes, the analysis of a patient's cough sounds provides valuable insights for physicians in diagnosing associated diseases.
Objective:
The aim of this study is to present an improved method for the detection of cough-related diseases based on deep learning, aimed at assisting in the diagnosis of relevant diseases.
Methods:
In this study, we conducted extensive data collection, which included voluntary cough audio recordings from patients with well-diagnosed respiratory diseases (such as chronic obstructive pulmonary disease, lung cancer, COVID-19, and pneumonia), as well as healthy participants,We collected a total of 2,610 audio samples.We added the Channel Attention Mechanism CAM to the last convolutional block of each residual block in the ResNet18 neural network, and later constructed the CAM-ResNet18 neural network model. We transformed the collected cough audio recordings into spectrograms and established the corresponding dataset, utilizing these spectrograms as input data for the CAM-ResNet18 neural network model. Subsequently, we trained the CAM-ResNet18 neural network model using the training set within this dataset,adjust the model parameters and train the model until it converges. Finally, we input the spectrograms from the test set into the pre-trained neural network model for precise classification.
Results:
Through experiments conducted on the audio dataset we collected, our proposed improvement method achieved an accuracy of 83.9% in classifying these five types of cough audio, while the traditional ResNet18 model had an accuracy of 78.16%.
Conclusions:
The experimental results of this study validate the effectiveness of the proposed method. The uniqueness of this model lies in its enhanced ability to focus on critical features, effectively reducing overfitting issues. As we continue to expand the cough audio dataset in future work, we anticipate further improvements in classification accuracy. The findings of this study hold significant potential application value in clinical diagnosis.
Citation
Request queued. Please wait while the file is being generated. It may take some time.
Copyright
© The authors. All rights reserved. This is a privileged document currently under peer-review/community review (or an accepted/rejected manuscript). Authors have provided JMIR Publications with an exclusive license to publish this preprint on it's website for review and ahead-of-print citation purposes only. While the final peer-reviewed paper may be licensed under a cc-by license on publication, at this stage authors and publisher expressively prohibit redistribution of this draft paper other than for review purposes.