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

2025-10-13

AI Summary

Portable 3-lead wearable electrocardiograms (ECGs) lack the diagnostic fidelity of clinical 12-lead ECGs, limiting their utility in cardiac screening. Method: We propose WearECG—a novel variational autoencoder (VAE) explicitly modeling spatiotemporal dependencies in ECG signals—to synthesize high-fidelity 12-lead ECGs from 3-lead inputs. The model employs a multi-objective loss combining mean squared error (MSE), mean absolute error (MAE), and Fréchet Inception Distance (FID), and is fine-tuned on ECGFounder for multi-label disease classification. Results: Evaluated on MIMIC-IV, reconstructed ECGs demonstrate physiological plausibility and clinical diagnostic validity, as confirmed by expert blind assessment. Downstream detection of myocardial infarction and other conditions achieves performance comparable to ground-truth 12-lead ECGs. This work pioneers the tight integration of generative modeling with rigorous clinical validation, establishing a scalable, low-cost paradigm for population-level cardiac screening.

Technology Category

• Machine Learning: Deep Generative Models & Autoencoders
• Computer Vision: Generative Adversarial Networks (GANs) for Vision
• Cognitive Modeling & Cognitive Systems: Computational Creativity

Application Category

• Economics, Online Markets and Human Computation: Trust and reliance of crowd workers and data experts on GenAI
• User Modeling, Personalization and Recommendation: On-Device user modeling, personalization, and recommendation
• Web Mining and Content Analysis: Web data generation and simulation

Abstract

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.

Problem

• Reconstructing 12-lead ECG from 3-lead signals
• Improving cardiac disease detection in wearables
• Addressing limited spatial coverage of portable ECGs

Innovation

• VAE reconstructs 12-lead ECG from 3 leads
• Model improves temporal and spatial dependencies
• Validates signals via clinical tests and classification