Reconstructing 12-Lead ECG from 3-Lead ECG using Variational Autoencoder to Improve Cardiac Disease Detection of Wearable ECG Devices

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Guan, Xinyan, Lai, Yongfan, Jin, Jiarui, Li, Jun, Wang, Haoyu, Zhao, Qinghao, Zhang, Deyun, Geng, Shijia, Hong, Shenda
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915550753456128
author Guan, Xinyan
Lai, Yongfan
Jin, Jiarui
Li, Jun
Wang, Haoyu
Zhao, Qinghao
Zhang, Deyun
Geng, Shijia
Hong, Shenda
author_facet Guan, Xinyan
Lai, Yongfan
Jin, Jiarui
Li, Jun
Wang, Haoyu
Zhao, Qinghao
Zhang, Deyun
Geng, Shijia
Hong, Shenda
contents Twelve-lead electrocardiograms (ECGs) are the clinical gold standard for cardiac diagnosis, providing comprehensive spatial coverage of the heart necessary to detect conditions such as myocardial infarction (MI). However, their lack of portability limits continuous and large-scale use. Three-lead ECG systems are widely used in wearable devices due to their simplicity and mobility, but they often fail to capture pathologies in unmeasured regions. To address this, we propose WearECG, a Variational Autoencoder (VAE) method that reconstructs twelve-lead ECGs from three leads: II, V1, and V5. Our model includes architectural improvements to better capture temporal and spatial dependencies in ECG signals. We evaluate generation quality using MSE, MAE, and Frechet Inception Distance (FID), and assess clinical validity via a Turing test with expert cardiologists. To further validate diagnostic utility, we fine-tune ECGFounder, a large-scale pretrained ECG model, on a multi-label classification task involving over 40 cardiac conditions, including six different myocardial infarction locations, using both real and generated signals. Experiments on the MIMIC dataset show that our method produces physiologically realistic and diagnostically informative signals, with robust performance in downstream tasks. This work demonstrates the potential of generative modeling for ECG reconstruction and its implications for scalable, low-cost cardiac screening.
format Preprint
id arxiv_https___arxiv_org_abs_2510_11442
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Reconstructing 12-Lead ECG from 3-Lead ECG using Variational Autoencoder to Improve Cardiac Disease Detection of Wearable ECG Devices
Guan, Xinyan
Lai, Yongfan
Jin, Jiarui
Li, Jun
Wang, Haoyu
Zhao, Qinghao
Zhang, Deyun
Geng, Shijia
Hong, Shenda
Machine Learning
Artificial Intelligence
68T05
I.2.6; I.2.7
Twelve-lead electrocardiograms (ECGs) are the clinical gold standard for cardiac diagnosis, providing comprehensive spatial coverage of the heart necessary to detect conditions such as myocardial infarction (MI). However, their lack of portability limits continuous and large-scale use. Three-lead ECG systems are widely used in wearable devices due to their simplicity and mobility, but they often fail to capture pathologies in unmeasured regions. To address this, we propose WearECG, a Variational Autoencoder (VAE) method that reconstructs twelve-lead ECGs from three leads: II, V1, and V5. Our model includes architectural improvements to better capture temporal and spatial dependencies in ECG signals. We evaluate generation quality using MSE, MAE, and Frechet Inception Distance (FID), and assess clinical validity via a Turing test with expert cardiologists. To further validate diagnostic utility, we fine-tune ECGFounder, a large-scale pretrained ECG model, on a multi-label classification task involving over 40 cardiac conditions, including six different myocardial infarction locations, using both real and generated signals. Experiments on the MIMIC dataset show that our method produces physiologically realistic and diagnostically informative signals, with robust performance in downstream tasks. This work demonstrates the potential of generative modeling for ECG reconstruction and its implications for scalable, low-cost cardiac screening.
title Reconstructing 12-Lead ECG from 3-Lead ECG using Variational Autoencoder to Improve Cardiac Disease Detection of Wearable ECG Devices
topic Machine Learning
Artificial Intelligence
68T05
I.2.6; I.2.7
url https://arxiv.org/abs/2510.11442