Explainable Deep Learning Model for Cardiac Arrhythmia Classification

Abdullah, T.A.A. and Zahid, M.S.B.M. and Tang, T.B. and Ali, W. and Nasser, M. (2022) Explainable Deep Learning Model for Cardiac Arrhythmia Classification. In: UNSPECIFIED.

Full text not available from this repository.
Official URL: https://www.scopus.com/inward/record.uri?eid=2-s2....

Abstract

In this work, we proposed a hybrid deep learning model that (CNN-GRU) combines a One-Dimensional Neural Network (1D CNN) and a Gated Recurrent Unit (GRU) to classify four types of cardiac arrhythmia and applied LIME to provide explanations for its predictions. LIME is a well-known local explanation method that can explain any machine learning model by simulating its behaviours to generate explanations. However, LIME can only explain tabular, text, and image datasets. Therefore, we proposed a visual presentation of LIME on signal dataset by applying a heatmap to highlight important areas on the heartbeat signals. Moreover, we propose an effective method to segment heartbeats from ECG records, ensuring that all key features are extracted correctly, such as QRS Complex, P Wave, and T Wave. The proposed hybrid model was trained using ECG lead II from the MIT-BIH dataset and evaluated based on accuracy, precision, recall, f1 score, and AUC-ROC performance matrix. To highlight the proposed model's validity, we compare it against the standalone CNN and GRU models and prove its superiority in terms of accuracy and ROC. © 2022 IEEE.

Item Type: Conference or Workshop Item (UNSPECIFIED)
Additional Information: cited By 2; Conference of 2022 International Conference on Future Trends in Smart Communities, ICFTSC 2022 ; Conference Date: 1 December 2022 Through 2 December 2022; Conference Code:186671
Uncontrolled Keywords: Diseases; Lead compounds; Learning systems; Lime; Recurrent neural networks; Seismic waves, Arrhythmia; Arrhythmia classification; Cardiac arrhythmia; CNN-gated recurrent unit; Explanation; Heatmaps; Lead II; Learning models; Neural-networks; One-dimensional, Electrocardiograms
Depositing User: Mr Ahmad Suhairi UTP
Date Deposited: 19 Dec 2023 03:23
Last Modified: 19 Dec 2023 03:23
URI: https://khub.utp.edu.my/scholars/id/eprint/17232

Actions (login required)

View Item
View Item