LeadWeaver: Topology-Aware Masked-Lead Modeling for Robust 12-Lead ECG Diagnosis
- 1German Research Center for Artificial Intelligence (DFKI)
Abstract
Clinical electrocardiography is organized as a structured set of correlated leads, yet most neural ECG classifiers assume that all twelve leads are present, correctly ordered, and acquired under the same device conditions at training and deployment. We propose LeadWeaver, a topology-aware ECG representation model designed for diagnosis under missing, corrupted, or device-shifted leads. LeadWeaver represents each lead as a temporal token stream, couples leads through a fixed-plus-learned physiological lead graph, and performs masked-lead latent completion rather than waveform synthesis. A reliability router estimates which observed leads should dominate each diagnostic decision, while a consistency objective aligns complete-lead and incomplete-lead predictions. We evaluate the model on PTB-XL and CPSC-style settings using complete-input diagnosis, arbitrary missing-lead subsets, lead corruption, cross-dataset transfer, calibration, and selective prediction. The central hypothesis is that robust ECG diagnosis requires explicitly modeling both inter-lead topology and uncertainty about unavailable evidence, rather than treating missing channels as zero-valued inputs.
Related
- STEM-LM: Spatio-Temporal Ecological Modeling via Masked Language Model for Multi-Species Distribution
- Routine serum biomarkers and lightgbm modeling for the diagnosis of abomasal ulcers in culled dairy cows.
- Empirical Geometry-Guided Modeling Enables Robust, High-Throughput Collagen Structure Prediction
- VenusGenome: Genome-Aware Masked Autoencoding for Microbial Genome Representation Learning
- VenusGenome: Genome-Aware Masked Autoencoding for Microbial Genome Representation Learning
BibTeX
@misc{schneider2026leadweaver,
title = {{LeadWeaver: Topology-Aware Masked-Lead Modeling for Robust 12-Lead ECG Diagnosis}},
author = {F. Schneider and N. Collins and L. A. Weber},
howpublished = {bioRxiv},
year = {2026},
month = nov,
doi = {10.64898/2026.11.05.764472},
url = {https://www.biorxiv.org/content/10.64898/2026.11.05.764472v1},
note = {CC BY},
}
RIS
TY - UNPB
TI - LeadWeaver: Topology-Aware Masked-Lead Modeling for Robust 12-Lead ECG Diagnosis
AU - F. Schneider
AU - N. Collins
AU - L. A. Weber
PY - 2026
DA - 2026/11/05
PB - bioRxiv
DO - 10.64898/2026.11.05.764472
UR - https://www.biorxiv.org/content/10.64898/2026.11.05.764472v1
AB - Clinical electrocardiography is organized as a structured set of correlated leads, yet most neural ECG classifiers assume that all twelve leads are present, correctly ordered, and acquired under the same device conditions at training and deployment. We propose LeadWeaver, a topology-aware ECG representation model designed for diagnosis under missing, corrupted, or device-shifted leads. LeadWeaver represents each lead as a temporal token stream, couples leads through a fixed-plus-learned physiological lead graph, and performs masked-lead latent completion rather than waveform synthesis. A reliability router estimates which observed leads should dominate each diagnostic decision, while a consistency objective aligns complete-lead and incomplete-lead predictions. We evaluate the model on PTB-XL and CPSC-style settings using complete-input diagnosis, arbitrary missing-lead subsets, lead corruption, cross-dataset transfer, calibration, and selective prediction. The central hypothesis is that robust ECG diagnosis requires explicitly modeling both inter-lead topology and uncertainty about unavailable evidence, rather than treating missing channels as zero-valued inputs.
ER -