HeartLLM: Discretized ECG Tokenization for LLM-Based Diagnostic Reasoning
DOI:
https://doi.org/10.1609/aaai.v40i40.40721Abstract
Electrocardiography (ECG) plays a central role in cardiovascular diagnostics, yet existing automated approaches often struggle to generalize across clinical tasks and offer limited support for open-ended reasoning. We present HeartLLM, a novel framework that integrates time-series (TS) and language modeling by enabling large language models (LLMs) to process 12-lead ECG signals for clinical text generation tasks. Our approach discretizes continuous ECG embeddings into quantized codes using a lead-wise encoder and quantization module. These quantized codes are then mapped to an extended ECG vocabulary to form ECG tokens, enabling the model to process both ECG and natural language inputs within a unified framework. To bridge the modality gap, we pretrain the model on an autoregressive ECG token forecasting task, allowing the LLM to capture temporal dynamics through its inherent language modeling capability. Finally, we perform instruction tuning on both ECG question answering and diagnostic report generation. Without modifying the core model, HeartLLM achieves strong performance across tasks while maintaining generalization to out-of-distribution settings. Extensive experiments demonstrate the effectiveness of each component and highlight the potential of integrating discretized ECG tokens into LLMs for medical reasoning.Downloads
Published
2026-03-14
How to Cite
Yang, J., Sun, W., & Shi, W. (2026). HeartLLM: Discretized ECG Tokenization for LLM-Based Diagnostic Reasoning. Proceedings of the AAAI Conference on Artificial Intelligence, 40(40), 34250–34258. https://doi.org/10.1609/aaai.v40i40.40721
Issue
Section
AAAI Technical Track on Natural Language Processing V