An Attention‐Driven Hybrid Deep Neural Network for Enhanced Heart Disease Classification

Lilhore, Umesh Kumar; Simaiya, Sarita; Alhussein, Musaed; Dalal, Surjeet; Aurangzeb, Khursheed; Hussain, Amir

Abstract

Heart disease continues to be a primary cause of mortality globally, highlighting the critical necessity for efficient early prediction and classification techniques. This study presents a new hybrid model attention-based CNN-Bi-LSTM that integrates the SMOTE with an attention-driven improved convolutional neural network-recurrent neural network architecture to improve the classification of heart sounds, especially from imbalanced datasets. Heart sounds are difficult to classify because of their complex acoustic properties and the variability of their characteristics across frequency and temporal domains. The proposed model utilises an advanced CNN to effectively extract global and local features, in conjunction with a bidirectional long short-term memory network to improve the architecture by capturing contextual information from both preceding and subsequent time sequences. The incorporation of spatial attention within the CNN and temporal attention in the RNN enables the model to concentrate on the most pertinent audio segments. To address the challenges presented by imbalanced and noisy datasets that may impede the efficacy of deep learning algorithms, our model employs SMOTE to improve data representation. The hybrid model outperformed popular models such as CNN, LSTM and CNN-LSTM, achieving a classification accuracy of more than 97% on the PCG and PASCAL heart sound datasets. The findings demonstrate the model's reliability as an initial evaluation tool in clinical settings, thereby improving support for cardiovascular disease diagnosis.

Citation

Lilhore, U. K., Simaiya, S., Alhussein, M., Dalal, S., Aurangzeb, K., & Hussain, A. (online). An Attention‐Driven Hybrid Deep Neural Network for Enhanced Heart Disease Classification. Expert Systems, https://doi.org/10.1111/exsy.13791

• Journal Article Type: Article
• Acceptance Date: Oct 31, 2024
• Online Publication Date: Nov 19, 2024
• Deposit Date: Nov 25, 2024
• Journal: Expert Systems
• Print ISSN: 0266-4720
• Electronic ISSN: 1468-0394
• Publisher: Wiley
• Peer Reviewed: Peer Reviewed
• DOI: https://doi.org/10.1111/exsy.13791
• Keywords: attention method, Bi-RNNs, classification, CNN, deep learning, heart disease, hybrid attention-based CNN-Bi-LSTM, RNN, SMOTE