HeartLLM: Discretized ECG Tokenization for LLM-Based Diagnostic Reasoning

Authors

  • Jinning Yang Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Chinese Academy of Sciences University of Chinese Academy of Sciences, Beijing, China
  • Wenjie Sun Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Chinese Academy of Sciences University of Chinese Academy of Sciences, Beijing, China
  • Wen Shi Department of Neurology, Massachusetts General Hospital, Boston, MA, USA Department of Neurology, Harvard Medical School, Boston, MA, USA

DOI:

https://doi.org/10.1609/aaai.v40i40.40721

Abstract

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