CaTok: Taming Mean Flows for One-Dimensional Causal Image Tokenization
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 63 of its 75 references that are in the map. First posted on arXiv 6 Mar 2026.
Builds on
63 of 75 references.
References · 63
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2023GPT-4 Technical Reportcited by 18,132
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2025Qwen3 Technical Reportcited by 8,195
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2023BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Modelscited by 6,069
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2022Generative Adversarial Networkscited by 5,603
38 more
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2017Neural Discrete Representation Learningcited by 4,650
- 2021Zero-Shot Text-to-Image Generationcited by 4,276
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
- 2016Improved Techniques for Training GANscited by 2,938
- 2019Generating Diverse High-Fidelity Images with VQ-VAE-2cited by 1,376
- 2025DINOv3cited by 1,297
- 2025SigLIP 2: Multilingual Vision-Language Encoders with Improved Semantic Understanding, Localization, and Dense Featurescited by 1,127
- 2022MaskGIT: Masked Generative Image Transformercited by 1,082
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
- 2025Qwen-Image Technical Reportcited by 973
- 2023Vision Transformers Need Registerscited by 844
- 2024Emu3: Next-Token Prediction is All You Needcited by 761
- 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 760
- 2022Autoregressive Image Generation using Residual Quantizationcited by 706
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 666
- 2024Autoregressive Image Generation without Vector Quantizationcited by 644
- 2023Muse: Text-To-Image Generation via Masked Generative Transformerscited by 640
- 2023Language Model Beats Diffusion -- Tokenizer is Key to Visual Generationcited by 633
- 2025Mean Flows for One-step Generative Modelingcited by 524
- 2021Diffusion Autoencoders: Toward a Meaningful and Decodable Representationcited by 447
- 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Modelscited by 376
- 2024Infinity: Scaling Bitwise AutoRegressive Modeling for High-Resolution Image Synthesiscited by 297
- 2024An Image is Worth 32 Tokens for Reconstruction and Generationcited by 295
- 2023Sequential Modeling Enables Scalable Learning for Large Vision Modelscited by 212
- 2025REPA-E: Unlocking VAE for End-to-End Tuning with Latent Diffusion Transformerscited by 212
- 2022Lossy Image Compression with Conditional Diffusion Modelscited by 165
- 2024Open-MAGVIT2: An Open-Source Project Toward Democratizing Auto-regressive Visual Generationcited by 153
- 2024Randomized Autoregressive Visual Generationcited by 125
- 2024OmniTokenizer: A Joint Image-Video Tokenizer for Visual Generationcited by 117
- 2025FlexTok: Resampling Images into 1D Token Sequences of Flexible Lengthcited by 105
- 2024MaskBit: Embedding-free Image Generation via Bit Tokenscited by 96
- 2023Return of Unconditional Generation: A Self-supervised Representation Generation Methodcited by 95
- 2025SimpleAR: Pushing the Frontier of Autoregressive Visual Generation through Pretraining, SFT, and RLcited by 85
- 2014Learning Ordered Representations with Nested Dropoutcited by 81
50 most cited of 63 All 63 (JSON)
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026Unified Multimodal Autoregressive Modeling with Shared Context-Visual Tokenizer is Key to Unificationcited by 3
- 2026DecQ: Detail-Condensing Queries for Enhanced Reconstruction and Generation in Representation Autoencoderscited by 2
- 2026Channel-wise Vector Quantizationcited by 0
- 2026ZipTok3D: High-Fidelity 3D Tokenization with Compact Token Prefixescited by 0
- 2026NesTok: Nested Self-Aligned 1D Tokenizer for Autoregressive Image Generationcited by 0
- 2026NesTok: Nested Self-Aligned 1D Tokenizer for Autoregressive Image Generation29 Sep 2026
- 2026ZipTok3D: High-Fidelity 3D Tokenization with Compact Token Prefixes1 Sep 2026
- 2026Unified Multimodal Autoregressive Modeling with Shared Context-Visual Tokenizer is Key to Unification16 Jun 2026
- 2026Channel-wise Vector Quantization25 May 2026
- 2026DecQ: Detail-Condensing Queries for Enhanced Reconstruction and Generation in Representation Autoencoders21 May 2026