arrow
Return

A sequential MAE-clustering self-supervised learning method for arrhythmia detection

delete2025-04-01
delete0
PRE
AI
Y
Yatao Zhang *
X
Xiangyu Li
L
Liting Zhang *
J
Jiang, Shipeng
M
Ma, Yijun
L
Li, Dong
DOI:10.1016/j.eswa.2025.126379delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In supervised learning, obtaining a substantial quantity of accurately labeled electrocardiogram (ECG) data poses significant challenges. In contrast, self-supervised learning, particularly the masked autoencoders (MAE), shows promise. This study proposes an enhanced MAE approach the sequential MAE-clustering (SMC) self-supervised learning method for analyzing various arrhythmias. Our method comprises a sequential MAE (SM) network, which utilizes sequential masking patches, and a clustering learning component consisting of a symbiotic network and Gaussian clustering. The SMC loss function, a combination of the mean squared error loss from the SM network and the ProtoNCE loss from clustering learning, was developed. The process begins by converting 1D ECG heartbeats into 2-D diagrams using the Gramian angular field technique. These diagrams are subsequently processed using the SMC self-supervised learning method to classify heartbeat categories as per the ANSI/AAMI standard namely, N, V, S, Q, and F of the MIT/BIH arrhythmia database. The SMC method demonstrated effective performance, achieving an average accuracy (acc) of 97.84 %, a positive predictive value (ppv) of 84.25 %, a sensitivity (sen) of 65.51 %, and a specificity (spe) of 94.06 % across the five heartbeat categories of the MIT/BIH arrhythmia database.
Keywords:
Self-supervised learning
Clustering
Sequential MAE
Arrhythmia detection

Journal

Expert Systems with Applications cover
Expert Systems with Applications
IF:
7.5
Papers:
3.0W
Citations:
10.2W

Organization

No organization information available
Cited Papers

Cited Papers

Robust detection of atrial fibrillation from short-term electrocardiogram using convolutional neural networks
err2020-12-01
err59
errOAAI
errNurmaini, Siti; Tondas, Alexander Edo; Darmawahyuni, Annisa; Rachmatullah, Muhammad Naufal; Partan, Radiyati Umi; Firdaus, Firdaus; Tutuko, Bambang; Pratiwi, Ferlita; Juliano, Andre Herviant; Khoirani, Rahmi
errShare
errSave
Inferior Myocardial Infarction Detection From Lead II of ECG: A Gramian Angular Field-Based 2D-CNN Approach
err2024-10-01
err0
errOAAI
errAsim Yousuf; Rehan Hafiz; Saqib Riaz; Muhammad Farooq; Kashif Riaz; Muhammad Mahboob Ur Rahman
errShare
errSave
An efficient ECG arrhythmia classification method based on Manta ray foraging optimization
err2021-11-01
err93
PREAI
errHoussein, Essam H.; Ibrahim, Ibrahim E.; Neggaz, Nabil; Hassaballah, M.; Wazery, Yaser M.
errShare
errSave
researcher View more