English

CaTok: Taming Mean Flows for One-Dimensional Causal Image Tokenization

Computer Vision and Pattern Recognition 2026-03-09 v1

Abstract

Autoregressive (AR) language models rely on causal tokenization, but extending this paradigm to vision remains non-trivial. Current visual tokenizers either flatten 2D patches into non-causal sequences or enforce heuristic orderings that misalign with the "next-token prediction" pattern. Recent diffusion autoencoders similarly fall short: conditioning the decoder on all tokens lacks causality, while applying nested dropout mechanism introduces imbalance. To address these challenges, we present CaTok, a 1D causal image tokenizer with a MeanFlow decoder. By selecting tokens over time intervals and binding them to the MeanFlow objective, as illustrated in Fig. 1, CaTok learns causal 1D representations that support both fast one-step generation and high-fidelity multi-step sampling, while naturally capturing diverse visual concepts across token intervals. To further stabilize and accelerate training, we propose a straightforward regularization REPA-A, which aligns encoder features with Vision Foundation Models (VFMs). Experiments demonstrate that CaTok achieves state-of-the-art results on ImageNet reconstruction, reaching 0.75 FID, 22.53 PSNR and 0.674 SSIM with fewer training epochs, and the AR model attains performance comparable to leading approaches.

Keywords

Cite

@article{arxiv.2603.06449,
  title  = {CaTok: Taming Mean Flows for One-Dimensional Causal Image Tokenization},
  author = {Yitong Chen and Zuxuan Wu and Xipeng Qiu and Yu-Gang Jiang},
  journal= {arXiv preprint arXiv:2603.06449},
  year   = {2026}
}

Comments

Project website is available in https://sharelab-sii.github.io/catok-web

R2 v1 2026-07-01T11:07:15.338Z