Beyond the Last Layer: Multi-Layer Representation Fusion for Visual Tokenization
Not on the map. Why?
Abstract (arXiv)
Representation autoencoders that reuse frozen pretrained vision encoders as visual tokenizers have achieved strong reconstruction and generation quality. However, existing methods universally extract features from only the last encoder layer, discarding the rich hierarchical information distributed across intermediate layers. We show that low-level visual details survive in the last layer merely as attenuated residuals after multiple layers of semantic abstraction, and that explicitly fusing multi-layer features can substantially recover this lost information. We propose DRoRAE (Depth-Routed Representation AutoEncoder), a lightweight fusion module that adaptively aggregates all encoder layers via energy-constrained routing and incremental correction, producing an enriched latent compatible with a frozen pretrained decoder. A three-phase decoupled training strategy first learns the fusion under the implicit distributional constraint of the frozen decoder, then fine-tunes the decoder to fully exploit the enriched representation. On ImageNet-256, DRoRAE reduces rFID from 0.57 to 0.29 and improves generation FID from 1.74 to 1.65 (with AutoGuidance), with gains also transferring to text-to-image synthesis. Furthermore, we uncover a log-linear scaling law (R2=0.86) between fusion capacity and reconstruction quality, identifying textitrepresentation richness as a new, predictably scalable dimension for visual tokenizers analogous to vocabulary size in NLP.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 33 of its 34 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
33 of 34 references.
References · 33
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,177
- 2025Qwen3 Technical Reportcited by 8,820
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2016Feature Pyramid Networks for Object Detectioncited by 5,036
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2021Vision Transformers for Dense Predictioncited by 1,803
- 2021Conceptual 12M: Pushing Web-Scale Image-Text Pre-Training To Recognize Long-Tail Visual Conceptscited by 1,216
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
- 2025Emerging Properties in Unified Multimodal Pretrainingcited by 897
21 more
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 737
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 703
- 2021Do Vision Transformers See Like Convolutional Neural Networks?cited by 675
- 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Modelscited by 393
- 2021Deep ViT Features as Dense Visual Descriptorscited by 330
- 2025Diffusion Transformers with Representation Autoencoderscited by 295
- 2014Hypercolumns for Object Segmentation and Fine-grained Localizationcited by 128
- 2025Monet: Reasoning in Latent Visual Space Beyond Images and Languagecited by 82
- 2024Dense Connector for MLLMscited by 49
- 2025Over-Tokenized Transformer: Vocabulary is Generally Worth Scalingcited by 44
- 2024Debiasing Multimodal Large Language Models via Penalization of Language Priorscited by 37
- 2026LongCat-Next: Lexicalizing Modalities as Discrete Tokenscited by 36
- 2025RealUnify: Do Unified Models Truly Benefit from Unification? A Comprehensive Benchmarkcited by 33
- 2025Mavors: Multi-granularity Video Representation for Multimodal Large Language Modelcited by 32
- 2024MMFuser: Multimodal Multi-Layer Feature Fuser for Fine-Grained Vision-Language Understandingcited by 23
- 2025Multimodal Language Models See Better When They Look Shallowercited by 23
- 2025MME-VideoOCR: Evaluating OCR-Based Capabilities of Multimodal LLMs in Video Scenarioscited by 18
- 2026VTC-Bench: Evaluating Agentic Multimodal Models via Compositional Visual Tool Chainingcited by 12
- 2024Instruction-Guided Fusion of Multi-Layer Visual Features in Large Vision-Language Modelscited by 7
- 2026RPiAE: A Representation-Pivoted Autoencoder Enhancing Both Image Generation and Editingcited by 7
- 2026Unveiling Fine-Grained Visual Traces: Evaluating Multimodal Interleaved Reasoning Chains in Multimodal STEM Taskscited by 4
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Edit-Compass & EditReward-Compass: A Unified Benchmark for Image Editing and Reward Modelingcited by 6
- 2026FuseReg: Regularizing Layer Fusion Mitigates the Reconstruction-Generation Gap in Representation Autoencoderscited by 1
- 2026HiRAE: Hierarchical Representation Autoencoding with Residual Budgetscited by 0
- 2026HiRAE: Hierarchical Representation Autoencoding with Residual Budgets29 Sep 2026
- 2026FuseReg: Regularizing Layer Fusion Mitigates the Reconstruction-Generation Gap in Representation Autoencoders25 Sep 2026
- 2026Edit-Compass & EditReward-Compass: A Unified Benchmark for Image Editing and Reward Modeling13 May 2026