Reconstructing 12-lead ECG from 3-lead ECG using variational autoencoder to improve cardiac disease detection of wearable ECG devices
Twelve-lead electrocardiograms (ECGs) are the clinical gold standard for cardiac diagnosis, offering comprehensive spatial coverage of the heart necessary for detecting conditions such as myocardial infarction (MI). However, their lack of portability limits continuous and large-scale deployment. In contrast, three-lead ECG systems are widely used in wearable devices due to their simplicity and mobility, but they often fail to capture pathologies localized in unmeasured regions. To bridge this gap, we propose WearECG, a Variational Autoencoder (VAE) method that reconstructs 12-lead ECGs from three leads (II, V1, V5). Our model includes architectural improvements to better capture temporal and spatial dependencies in ECG signals. We evaluate generation quality using MSE, MAE, and Fréchet Inception Distance (FID), and assess clinical validity via a Turing test with expert cardiologists. To further validate the diagnostic utility, we fine-tune ECGFounder-a large-scale pretrained ECG model-on a multi-label classification task involving over 40 cardiac conditions, including 6 different myocardial infarction locations using both real and generated signals. Experiments on the MIMIC and PTB-XL datasets 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.
Authors
- Guanyu Mu
- Yongfan Lai
- Deyun Zhang (ORCID: https://orcid.org/0000-0002-4041-3083)
- Yiping Wang
- Rui Wu
- Shenda Hong (ORCID: https://orcid.org/0000-0001-7521-5127)
- Qinghao Zhao
- Jiarui Jin
- Haoyu Wang
- Shijia Geng
- Xinyan Guan
- Li, Jun, 1966 Mar. 15-
Institutions
- University of Science and Technology of China (CN)
- Peking University (CN)
- Beijing Academy of Artificial Intelligence (CN)
- Peking University People's Hospital (CN)
- Peking University First Hospital (CN)
- Second Hospital of Tianjin Medical University (CN)
- Zhengzhou Central Hospital (CN)
- University of Chinese Academy of Sciences (CN)
- Anshan Hospital (CN)
- Tianjin Medical University (CN)
Publication Details
- Journal
- PLOS Digital Health
- Published
- 2026-05-22
- DOI
- https://doi.org/10.1371/journal.pdig.0001335
- Primary Topic
- ECG Monitoring and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China