Related Experiment Video
Updated: Jun 24, 2025
Author Spotlight: Advancing Alzheimer's Research – Exploring Early Detection and Multi-Omics Approaches
Published on: December 15, 2023
ECG autoencoder based on low-rank attention
Shilin Zhang1, Yixian Fang2, Yuwei Ren1
1School of Information Science and Engineering (Institute of Data Science and Technology), Shandong Normal University, Jinan, 250014, China.
Insights
This study introduces a novel ECG autoencoder with low-rank attention to capture spatial features, significantly improving cardiovascular disease detection accuracy. The method enhances machine learning models for better arrhythmia classification.
Area of Science:
- Cardiology
- Machine Learning
- Signal Processing
Background:
- Cardiovascular disease (CVD) is a leading cause of mortality globally.
- Electrocardiogram (ECG) analysis is crucial for diagnosing CVD.
- Current machine learning models often overlook spatial features in ECG signals.
Purpose of the Study:
- To propose a novel ECG autoencoder network incorporating low-rank attention (LRA-autoencoder).
- To capture and leverage spatial dimension features in ECG signals for improved diagnostic accuracy.
- To enhance the differentiation of features among different cardiovascular disease categories.
Main Methods:
- Developed an LRA-autoencoder architecture to interpret ECG signals spatially.
- Utilized a low-rank attention block (LRA-block) with singular value decomposition to extract and weight spatial features.
- Employed a ResNet-18 network classifier for performance evaluation on benchmark datasets.
Main Results:
- The LRA-autoencoder achieved a mean accuracy of 0.997 on the MIT-BIH Arrhythmia dataset.
- On the PhysioNet Challenge 2017 dataset, the method obtained a mean accuracy of 0.850 and a mean F1-score of 0.843.
- Experimental results demonstrate superior classification performance compared to existing methods.
Conclusions:
- The proposed LRA-autoencoder effectively captures spatial features in ECG signals.
- This approach significantly enhances the accuracy of cardiovascular disease classification using machine learning.
- The method shows promise for improving automated ECG diagnostic tools.
Abstract:
The prevalence of cardiovascular disease (CVD) has surged in recent years, making it the foremost cause of mortality among humans. The Electrocardiogram (ECG), being one of the pivotal diagnostic tools for cardiovascular diseases, is increasingly gaining prominence in the field of machine learning. However, prevailing neural network models frequently disregard the spatial dimension features inherent in ECG signals. In this paper, we propose an ECG autoencoder network architecture incorporating low-rank attention (LRA-autoencoder). It is designed to capture potential spatial features of ECG signals by interpreting the signals from a spatial perspective and extracting correlations between different signal points. Additionally, the low-rank attention block (LRA-block) obtains spatial features of electrocardiogram signals through singular value decomposition, and then assigns these spatial features as weights to the electrocardiogram signals, thereby enhancing the differentiation of features among different categories. Finally, we utilize the ResNet-18 network classifier to assess the performance of the LRA-autoencoder on both the MIT-BIH Arrhythmia and PhysioNet Challenge 2017 datasets. The experimental results reveal that the proposed method demonstrates superior classification performance. The mean accuracy on the MIT-BIH Arrhythmia dataset is as high as 0.997, and the mean accuracy and -score on the PhysioNet Challenge 2017 dataset are 0.850 and 0.843.
More Related Videos
06:37Author Spotlight: Addressing Technical and Subjective Challenges in Measuring Classroom Attention
Published on: December 15, 2023
08:45Mapping Cortical Dynamics Using Simultaneous MEG/EEG and Anatomically-constrained Minimum-norm Estimates: an Auditory Attention Example
Published on: October 24, 2012
