mlsanityData to 7 Oct 2026

RecTok: Reconstruction Distillation along Rectified Flow

arXiv 2512.13421 ↗First posted 15 Dec 2025cs.CVcited by 4

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

  1. 2022Flow Matching for Generative Modeling6 Oct
  2. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep

Other references · 64

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  2. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  3. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  4. 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
  5. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
  6. 2023Segment Anythingcited by 8,158
  7. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
  8. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
  9. 2022Scalable Diffusion Models with Transformerscited by 5,931
  10. 2017Neural Discrete Representation Learningcited by 4,650
  11. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  12. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
38 more
  1. 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
  2. 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
  3. 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
  4. 2016Improved Techniques for Training GANscited by 2,938
  5. 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,409
  6. 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,961
  7. 2020GLU Variants Improve Transformercited by 1,724
  8. 2019Root Mean Square Layer Normalizationcited by 1,425
  9. 2019Generating Diverse High-Fidelity Images with VQ-VAE-2cited by 1,376
  10. 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
  11. 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 891
  12. 2021Vector Quantized Diffusion Model for Text-to-Image Synthesiscited by 773
  13. 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 707
  14. 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 666
  15. 2021Vector-quantized Image Modeling with Improved VQGANcited by 656
  16. 2024Autoregressive Image Generation without Vector Quantizationcited by 644
  17. 2023Language Model Beats Diffusion -- Tokenizer is Key to Visual Generationcited by 633
  18. 2021Masked Feature Prediction for Self-Supervised Visual Pre-Trainingcited by 573
  19. 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 498
  20. 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 447
  21. 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Modelscited by 376
  22. 2024Guiding a Diffusion Model with a Bad Version of Itselfcited by 286
  23. 2025Diffusion Transformers with Representation Autoencoderscited by 275
  24. 2023MDTv2: Masked Diffusion Transformer is a Strong Image Synthesizercited by 260
  25. 2025REPA-E: Unlocking VAE for End-to-End Tuning with Latent Diffusion Transformerscited by 212
  26. 2022Scalable Adaptive Computation for Iterative Generationcited by 190
  27. 2023Fast Training of Diffusion Models with Masked Transformerscited by 172
  28. 2023AM-RADIO: Agglomerative Vision Foundation Model -- Reduce All Domains Into Onecited by 165
  29. 2023CLIPSelf: Vision Transformer Distills Itself for Open-Vocabulary Dense Predictioncited by 114
  30. 2025Masked Autoencoders Are Effective Tokenizers for Diffusion Modelscited by 93
  31. 2025PixelFlow: Pixel-Space Generative Models with Flowcited by 91
  32. 2025Lumina-DiMOO: An Omni Diffusion Large Language Model for Multi-Modal Generation and Understandingcited by 83
  33. 2025Latent Diffusion Model without Variational Autoencodercited by 83
  34. 2025PixNerd: Pixel Neural Field Diffusioncited by 75
  35. 2025Improving the Diffusability of Autoencoderscited by 74
  36. 2024A Survey of Multimodal Large Language Model from A Data-centric Perspectivecited by 73
  37. 2024Meissonic: Revitalizing Masked Generative Transformers for Efficient High-Resolution Text-to-Image Synthesiscited by 55
  38. 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

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Variable-Length Tokenization via Learnable Global Merging for Diffusion Transformers
  2. 2026Representation by Design in Generation: Cross-View Class-Token Alignment in Diffusion Transformers
  3. 2025Terminal Velocity Matching
  4. 2025From Navigation to Refinement: Revealing the Two-Stage Nature of Flow-based Diffusion Models through Oracle Velocity
  5. 2026Diffusing in the Right Space: A Systematic Study of Latent Diffusability
  6. 2023Boosting Latent Diffusion with Flow Matching
  7. 2024Rectified Diffusion: Straightness Is Not Your Need in Rectified Flow
  8. 2025Diff2Flow: Training Flow Matching Models via Diffusion Model Alignment