RecTok: Reconstruction Distillation along Rectified Flow
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
Visual tokenizers play a crucial role in diffusion models. The dimensionality of latent space governs both reconstruction fidelity and the semantic expressiveness of the latent feature. However, a fundamental trade-off is inherent between dimensionality and generation quality, constraining existing methods to low-dimensional latent spaces. Although recent works have leveraged vision foundation models to enrich the semantics of visual tokenizers and accelerate convergence, high-dimensional tokenizers still underperform their low-dimensional counterparts. In this work, we propose RecTok, which overcomes the limitations of high-dimensional visual tokenizers through two key innovations: flow semantic distillation and reconstruction--alignment distillation. Our key insight is to make the forward flow in flow matching semantically rich, which serves as the training space of diffusion transformers, rather than focusing on the latent space as in previous works. Specifically, our method distills the semantic information in VFMs into the forward flow trajectories in flow matching. And we further enhance the semantics by introducing a masked feature reconstruction loss. Our RecTok achieves superior image reconstruction, generation quality, and discriminative performance. It achieves state-of-the-art results on the gFID-50K under both with and without classifier-free guidance settings, while maintaining a semantically rich latent space structure. Furthermore, as the latent dimensionality increases, we observe consistent improvements. Code and model are available at https://shi-qingyu.github.io/rectok.github.io.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 66 of its 76 references that are in the map. First posted on arXiv 15 Dec 2025.
Builds on
66 of 76 references.
Closest ancestors · 2
Same direction, newest first
- 2022Flow Matching for Generative Modeling6 Oct
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep
Other references · 64
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2023Segment Anythingcited by 8,158
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2017Neural Discrete Representation Learningcited by 4,650
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
38 more
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
- 2016Improved Techniques for Training GANscited by 2,938
- 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,409
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,961
- 2020GLU Variants Improve Transformercited by 1,724
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2019Generating Diverse High-Fidelity Images with VQ-VAE-2cited by 1,376
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 891
- 2021Vector Quantized Diffusion Model for Text-to-Image Synthesiscited by 773
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 707
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 666
- 2021Vector-quantized Image Modeling with Improved VQGANcited by 656
- 2024Autoregressive Image Generation without Vector Quantizationcited by 644
- 2023Language Model Beats Diffusion -- Tokenizer is Key to Visual Generationcited by 633
- 2021Masked Feature Prediction for Self-Supervised Visual Pre-Trainingcited by 573
- 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 498
- 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 447
- 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Modelscited by 376
- 2024Guiding a Diffusion Model with a Bad Version of Itselfcited by 286
- 2025Diffusion Transformers with Representation Autoencoderscited by 275
- 2023MDTv2: Masked Diffusion Transformer is a Strong Image Synthesizercited by 260
- 2025REPA-E: Unlocking VAE for End-to-End Tuning with Latent Diffusion Transformerscited by 212
- 2022Scalable Adaptive Computation for Iterative Generationcited by 190
- 2023Fast Training of Diffusion Models with Masked Transformerscited by 172
- 2023AM-RADIO: Agglomerative Vision Foundation Model -- Reduce All Domains Into Onecited by 165
- 2023CLIPSelf: Vision Transformer Distills Itself for Open-Vocabulary Dense Predictioncited by 114
- 2025Masked Autoencoders Are Effective Tokenizers for Diffusion Modelscited by 93
- 2025PixelFlow: Pixel-Space Generative Models with Flowcited by 91
- 2025Lumina-DiMOO: An Omni Diffusion Large Language Model for Multi-Modal Generation and Understandingcited by 83
- 2025Latent Diffusion Model without Variational Autoencodercited by 83
- 2025PixNerd: Pixel Neural Field Diffusioncited by 75
- 2025Improving the Diffusability of Autoencoderscited by 74
- 2024A Survey of Multimodal Large Language Model from A Data-centric Perspectivecited by 73
- 2024Meissonic: Revitalizing Masked Generative Transformers for Efficient High-Resolution Text-to-Image Synthesiscited by 55
- 2025GigaTok: Scaling Visual Tokenizers to 3 Billion Parameters for Autoregressive Image Generationcited by 55
50 most cited of 64 All 64 (JSON)
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026MUSE: Resolving Manifold Misalignment in Visual Tokenization via Topological Orthogonalitycited by 3
- 2026Video Generation with Predictive Latentscited by 0
- 2026STREAM: Stochastic Riemannian Flow Matching with Anisotropic Decoder for Digital Histopathology Image Generationcited by 0
- 2026AffineTok: Semantic Affine Consistency for Diffusion-Friendly Visual Tokenizercited by 0
- 2026AffineTok: Semantic Affine Consistency for Diffusion-Friendly Visual Tokenizer24 Aug 2026
- 2026STREAM: Stochastic Riemannian Flow Matching with Anisotropic Decoder for Digital Histopathology Image Generation5 Jun 2026
- 2026MUSE: Resolving Manifold Misalignment in Visual Tokenization via Topological Orthogonality7 May 2026
- 2026Video Generation with Predictive Latents4 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026Variable-Length Tokenization via Learnable Global Merging for Diffusion Transformers
- 2026Representation by Design in Generation: Cross-View Class-Token Alignment in Diffusion Transformers
- 2025Terminal Velocity Matching
- 2025From Navigation to Refinement: Revealing the Two-Stage Nature of Flow-based Diffusion Models through Oracle Velocity
- 2026Diffusing in the Right Space: A Systematic Study of Latent Diffusability
- 2023Boosting Latent Diffusion with Flow Matching
- 2024Rectified Diffusion: Straightness Is Not Your Need in Rectified Flow
- 2025Diff2Flow: Training Flow Matching Models via Diffusion Model Alignment