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
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Visual Question Answering | PMC-VQA (test) | Accuracy25.55 | 15 | |
| Image Tokenization | Medical Images (inference) | Memory Usage4.69 | 14 | |
| Image Classification | Dermoscopy (test) | mAP71.52 | 8 | |
| Image Classification | Fundus (test) | mAP56.41 | 8 | |
| Image Classification | Pathology (test) | mAP96.88 | 8 | |
| Image Classification | X-ray (test) | mAP99.08 | 8 | |
| Image Classification | Medical Modalities Average (test) | mAP82.27 | 8 | |
| Medical Image Reconstruction | CT | rFID7.88 | 8 | |
| Medical Image Reconstruction | Dermatology | rFID22.27 | 8 | |
| Medical Image Reconstruction | Endo | rFID10.66 | 8 |