Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Unified Medical Image Tokenizer for Autoregressive Synthesis and Understanding

About

Autoregressive modeling has driven major advances in multimodal AI, yet its application to medical imaging remains constrained by the absence of a unified image tokenizer that simultaneously preserves fine-grained anatomical structures and rich clinical semantics across heterogeneous modalities. Existing approaches jointly optimize image reconstruction and textual semantic objectives, relying on large-scale image-caption pairs and are prone to gradient interference. This is ill-suited for the medical domain where paired data are scarce and abundant unpaired images remain unexploited. This work identifies these issues in building unified medical image tokenizers, and introduces a principled two-stage training framework using visual representation as a bridge to address them. The propose visual representation alignment stage enables the utilization of large-scale unpaired medical images to ensure reconstruction fidelity and establish foundational semantics, alleviating the interference and better preparing for the second stage where fine-grained textual semantics are injected using image-text pairs. The resulting tokenizer, MedITok, is trained on over 33 million medical images spanning 9 modalities and 2 million image-text pairs. MedITok achieves state-of-the-art performance on 30+ benchmarks spanning 9 imaging modalities and 4 task families. It further enables autoregressive modeling for diagnostic and generative applications, serving as a scalable component for future multimodal models with unified synthesis and understanding capabilities in the medical domain. Project page: https://github.com/Masaaki-75/meditok

Chenglong Ma, Yuanfeng Ji, Jin Ye, Zilong Li, Chenhui Wang, Junzhi Ning, Wei Li, Lihao Liu, Qiushan Guo, Tianbin Li, Junjun He, Hongming Shan• 2025

Related benchmarks

TaskDatasetResultRank
Visual Question AnsweringPMC-VQA (test)
Accuracy25.55
15
Image TokenizationMedical Images (inference)
Memory Usage4.69
14
Image ClassificationDermoscopy (test)
mAP71.52
8
Image ClassificationFundus (test)
mAP56.41
8
Image ClassificationPathology (test)
mAP96.88
8
Image ClassificationX-ray (test)
mAP99.08
8
Image ClassificationMedical Modalities Average (test)
mAP82.27
8
Medical Image ReconstructionCT
rFID7.88
8
Medical Image ReconstructionDermatology
rFID22.27
8
Medical Image ReconstructionEndo
rFID10.66
8
Showing 10 of 25 rows

Other info

Follow for update