Detecting Structural Heart Disease from Electrocardiograms via a Generalized Additive Model of Interpretable Foundation-Model Predictors
Organizations: Department of Information Center, Fuwai Hospital, Chinese Academy of Medical Sciences and Peking Union Medical College, Beijing, 100037, China · The Tsinghua Shenzhen International Graduate School, Tsinghua University, Shenzhen, 518055, China · National Clinical Research Center for Cardiovascular Diseases, Fuwai Hospital, Chinese Academy of Medical Sciences and Peking Union Medical College, National Center for Cardiovascular Diseases, Beijing, 100037, China
Abstract
Structural heart disease (SHD) is a prevalent condition with many undiagnosed cases, and early detection is often limited by the high cost and accessibility constraints of echocardiography (ECHO). Recent studies show that artificial intelligence (AI)-based analysis of electrocardiograms (ECGs) can detect SHD, offering a scalable alternative. However, existing methods are fully black-box models, limiting interpretability and clinical adoption. To address these challenges, we propose an interpretable and effective framework that integrates clinically meaningful ECG foundation-model predictors within a generalized additive model, enabling transparent risk attribution while maintaining strong predictive performance. Using the EchoNext benchmark of over 80,000 ECG-ECHO pairs, the method demonstrates relative improvements of +0.98% in AUROC, +1.01% in AUPRC, and +1.41% in F1 score over the latest state-of-the-art deep-learning baseline, while achieving slightly better performance even with only 30% of the training data. Subgroup analyses confirm robust performance across heterogeneous populations, and the estimated entry-wise functions provide interpretable insights into the relationships between risks of traditional ECG diagnoses and SHD. This work illustrates a complementary paradigm between classical statistical modeling and modern AI, offering a pathway to interpretable, high-performing, and clinically actionable ECG-based SHD screening.
Figures & tables
| 7 clinical variables | 71 traditional ECG diagnostic categories |
| sex, ventricular rate, atrial rate, PR interval, QRS duration, QT interval, patient age at time of ECG | NORM, LVOLT, SR, SBRAD, ABQRS, IMI, SARRH, AFLT, AFIB, NDT, NST_, DIG, LVH, LPFB, LNGQT, LAFB, RAO/RAE, IRBBB, RVH, IVCD, LMI, ASMI, AMI, 1AVB, ISCAL, STACH, ISC_, PACE, ISCLA, SEHYP, ISCIL, ILMI, PVC, CRBBB, CLBBB, ANEUR, ALMI, ISCAS, TAB_, HVOLT, LOWT, STD_, PAC, EL, NT_, QWAVE, INVT, LPR, VCLVH, LAO/LAE, ILBBB, ISCIN, SVTAC, INJAL, INJAS, IPMI, WPW, ISCAN, INJLA, BIGU, TRIGU, IPLMI, 3AVB, INJIL, 2AVB, PRC(S), PSVT, PMI, STE_, INJIN, SVARR |
| Training set | Validation set | Test set | |
| Patients (n) | 26,218 | 4,626 | 5,442 |
| ECGs (n) | 72,475 | 4,626 | 5,442 |
| Age groups | |||
| 18–59 | 29,783 (41.1%) | 1,787 (38.6%) | 2,124 (39.0%) |
| 60–69 | 18,745 (25.9%) | 1,093 (23.6%) | 1,318 (24.2%) |
| 70–79 | 14,898 (20.6%) | 975 (21.1%) | 1,154 (21.2%) |
| Model | Training data | AUROC | AUPRC | F1 score |
| Logistic regression-1 | 100% | 74.8 (73.6–76.0) | 67.7 (65.6-69.8) | 66.3 (65.0-67.7) |
| Support vector machine-1 | 100% | 75.8 (74.5-77.0) | 67.8 (65.6-69.8) | 66.3 (65.1-67.8) |
| Logistic regression-2 (ours) | 100% | 80.8 (79.6–81.9) | 76.9 (74.9–78.6) | 70.9 (69.7-72.4) |
| Support vector machine-2 (ours) | 100% | 80.8 (79.7-82.0) | 76.7 (74.7-78.4) | 71.0 (69.7-72.4) |
| Columbia mini model | 100% | 82.0 (80.9–83.1) | 78.9 (77.2–80.4) | 70.8 (69.5–72.2) |
| Additive model (ours) | 10% | 81.5 (80.4–82.6) | 77.6 (75.8–79.4) | 70.6 (69.2–72.0) |
| Columbia mini model | Additive model (ours) | ||||
| Subgroup | AUROC | AUPRC | AUROC | AUPRC | |
| Age groups | |||||
| 18–59 | 2124 | 83.0 (81.1–84.8) | 73.7 (70.7–76.8) | 83.5 (81.7–85.4) | 74.7 (71.8–77.7) |
| 60–69 | 1318 | 80.6 (78.2–83.0) | 75.2 (71.6–78.7) | 81.8 (79.3–84.2) | 77.5 (73.8–81.0) |
| 70–79 | 1154 | 77.9 (75.1–80.5) | 80.4 (77.3–83.2) | 79.0 (76.3–81.5) | 81.2 (78.1–84.1) |
| 80+ | 846 | 75.9 (72.7–79.0) | 85.5 (82.7–88.3) | 76.0 (72.9–79.1) | 85.1 (82.3–88.0) |
Electrocardiogram Classification
Electrocardiogram classification uses machine learning to identify heart rhythms and cardiac conditions from ECG waveforms for screening, diagnosis, and monitoring.
See all 104 papers