ECG-Byte: A Tokenizer for End-to-End Generative Electrocardiogram Language Modeling

William Han, Chaojing Duan, Michael A. Rosenberg, Emerson Liu, Ding Zhao
12/18/2024
cs.CLeess.SP

No official code found. Best implementation: willxxy/ECG-Bench (50 stars, MIT, last commit 2026-02-23). 1 more below.

Abstract

Large Language Models (LLMs) have demonstrated exceptional versatility across domains, including applications to electrocardiograms (ECGs). A growing body of work focuses on generating text from multi-channeled ECG signals and corresponding textual prompts. Existing approaches often involve a two-stage process: pretraining an ECG-specific encoder with a self-supervised learning (SSL) objective, followed by finetuning an LLM for natural language generation (NLG) using encoder-derived features. However, these methods face two key limitations: inefficiency due to multi-stage training and challenges in interpreting encoder-generated features. To overcome these issues, we propose ECG-Byte, an adapted byte pair encoding (BPE) tokenizer pipeline for autoregressive language modeling of ECGs. ECG-Byte compresses and encodes ECG signals into tokens, enabling direct end-to-end LLM training by combining ECG and text tokens. This approach enhances interpretability, as ECG tokens can be directly mapped back to the original signals. Leveraging ECG-Byte, we achieve competitive NLG performance while training 3 times faster and using just 48\% of the data required by traditional two-stage methods.

View on arXivView PDF

Code Implementations(2)

willxxy/ECG-ByteCommunity100%
MIT

[MLHC 2025] ECG-Byte: A Tokenizer for End-to-End Generative Electrocardiogram Language Modeling

331C, C++Dec 18, 20241 years agoMIT
biosignalsdeep-learningecgelectrocardiogramgenerative-ai+8 more
willxxy/ECG-BenchCommunity72%
MIT

A Unified Framework for Benchmarking Generative Electrocardiogram-Language Models (ELMs)

5010Rust, ShellDec 23, 20247 months agoMIT
biosignalsdeep-learningecgecg-signalelectrocardiogram+10 more

Similar Papers

Cite this paper

@article{han2024ecgbyte,
  title  = {ECG-Byte: A Tokenizer for End-to-End Generative Electrocardiogram Language Modeling},
  author = {William Han and Chaojing Duan and Michael A. Rosenberg and Emerson Liu and Ding Zhao},
  year   = {2024},
  eprint = {2412.14373},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.14373v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2412.14373.svg)](https://codeofpaper.com/paper/2412.14373)

Discussion