arXiv cs.AI· Huynh Dang Nguyen, Trong-Thang Pham, Anh-Thuan Do, Phuong-Nhi Dinh, Brijesh Patel, Ngan Le, Van Nguyen·· 12 小时前AI 评分26
IECG:面向心电图分析的弱监督可解释性增强基础模型
IECG: Weakly-Supervised Interpretability-Enhancing Foundation Model for Electrocardiogram
AI 导读
IECG 是一个弱监督心电图基础模型,无需人工标注异常区域即可同时完成诊断分类与异常定位。它从既有心电图临床知识中推导异常热图伪标签,通过 KL 散度损失引导 CLS-token 注意力聚焦诊断相关波形区域,并在预训练中联合优化临床引导注意力监督、ECG-文本对比对齐与报告生成目标。实验显示其在保持分类性能的同时实现 SOTA 异常区域定位,论文已被 WACV 2027 接收。
正文 · 原文
Abstract:Electrocardiography (ECG) is a cornerstone of cardiac diagnosis, yet most machine learning methods for ECG analysis lack interpretability, limiting their clinical adoption. We present IECG, a weakly-supervised ECG foundation model that jointly performs diagnostic classification and abnormality localization without requiring manual abnormal region annotations, overcoming a long-standing bottleneck in interpretable ECG learning. Central to IECG is a weakly-supervised attention strategy that explicitly incorporates clinical priors by deriving abnormality heatmap pseudo-labels from established ECG clinical knowledge. These pseudo-labels guide the model's CLS-token attention through a KL-divergence loss, encouraging it to focus on diagnostically relevant waveform regions. During pretraining, IECG is jointly optimized with clinically guided attention supervision, ECG-text contrastive alignment, and report captioning objectives, enabling it to learn rich multimodal representations while producing localized waveform evidence. By explicitly aggregating features from clinically guided attention regions, IECG grounds its diagnostic predictions in interpretable waveform evidence without sacrificing classification performance. Extensive experiments demonstrate that IECG achieves state-of-the-art abnormal region localization while maintaining strong classification performance, advancing trustworthy foundation models for ECG analysis.
| Comments: | Accepted in WACV 2027 |
| Subjects: | Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2610.28083 [cs.AI] |
| (or arXiv:2610.28083v1 [cs.AI] for this version) | |
| https://doi.org/10.48550/arXiv.2610.28083 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Huynh Nguyen [view email]
[v1]
Tue, 20 Oct 2026 12:03:09 UTC (607 KB)
来源:arXiv cs.AI · arxiv.org