Lightweight Ensemble Network for Detecting Heart Disease Using ECG Signals
Siho Shin, Mingu Kang, Gengjia Zhang, Jaehyo Jung, Youn Tae Kim
Abstract
Heart disease should be treated quickly when symptoms appear. Machine-learning methods for detecting heart disease require desktop computers, an obstacle that can have fatal consequences for patients who must check their health periodically. Herein, we propose a MobileNet-based ensemble algorithm for arrhythmia diagnosis that can be easily and quickly operated in a mobile environment. The electrocardiogram (ECG) signal measured over a short period of time was augmented using the matching pursuit algorithm to achieve a high accuracy. The arrhythmia data were classified through an ensemble classifier combining MobileNetV2 and BiLSTM. By classifying the data using this algorithm, an accuracy of 91.7% was achieved. The performance of the algorithm was evaluated using a confusion matrix and a receiver operating characteristic curve. The sensitivity, specificity, precision, and F1 score were 0.92, 0.91, 0.92, and 0.92, respectively. Because the proposed algorithm does not require long-term ECG signal measurement, it facilitates health management for busy people. Moreover, parameters are exchanged when learning data, enhancing the security of the system. In addition, owing to the lightweight deep-learning model, the proposed algorithm can be applied to mobile healthcare, object detection, text recognition, and authentication.