RPiAE: A Representation-Pivoted Autoencoder Enhancing Both Image Generation and Editing
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Diffusion models have become the dominant paradigm for image generation and editing, with latent diffusion models shifting denoising to a compact latent space for efficiency and scalability. Recent attempts to leverage pretrained visual representation models as tokenizer priors either align diffusion features to representation features or directly reuse representation encoders as frozen tokenizers. Although such approaches can improve generation metrics, they often suffer from limited reconstruction fidelity due to frozen encoders, which in turn degrades editing quality, as well as overly high-dimensional latents that make diffusion modeling difficult. To address these limitations, We propose Representation-Pivoted AutoEncoder, a representation-based tokenizer that improves both generation and editing. We introduce Representation-Pivot Regularization, a training strategy that enables a representation-initialized encoder to be fine-tuned for reconstruction while preserving the semantic structure of the pretrained representation space, followed by a variational bridge which compress latent space into a compact one for better diffusion modeling. We adopt an objective-decoupled stage-wise training strategy that sequentially optimizes generative tractability and reconstruction-fidelity objectives. Together, these components yield a tokenizer that preserves strong semantics, reconstructs faithfully, and produces latents with reduced diffusion modeling complexity. Experiments demonstrate that RPiAE outperforms other visual tokenizers on text-to-image generation and image editing, while delivering the best reconstruction fidelity among representation-based tokenizers.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 39 of its 43 references that are in the map. First posted on arXiv 19 Mar 2026.
Builds on
39 of 43 references.
Closest ancestors · 14
Same direction, newest first
- 2025Both Semantics and Reconstruction Matter: Making Representation Encoders Ready for Text-to-Image Generation and Editing19 Dec
- 2025One Layer Is Enough: Adapting Pretrained Visual Encoders for Image Generation8 Dec
- 2025Latent Diffusion Model without Variational Autoencoder17 Oct
- 2025Diffusion Transformers with Representation Autoencoders13 Oct
- 2025REPA-E: Unlocking VAE for End-to-End Tuning with Latent Diffusion Transformers14 Apr
- 2025Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models2 Jan
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think9 Oct
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers16 Jan
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis4 Jul
- 2023Fast Training of Diffusion Models with Masked Transformers15 Jun
- 2022Scalable Diffusion Models with Transformers19 Dec
2 more
Other references · 25
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2025Qwen3 Technical Reportcited by 8,195
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2022Generative Adversarial Networkscited by 5,603
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2023Sigmoid Loss for Language Image Pre-Trainingcited by 3,048
- 2022MaskGIT: Masked Generative Image Transformercited by 1,082
- 2025Qwen-Image Technical Reportcited by 973
- 2025FLUX.1 Kontext: Flow Matching for In-Context Image Generation and Editing in Latent Spacecited by 960
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
13 more
- 2025Emerging Properties in Unified Multimodal Pretrainingcited by 855
- 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 760
- 2021Vector-quantized Image Modeling with Improved VQGANcited by 656
- 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignmentcited by 455
- 2025Step1X-Edit: A Practical Framework for General Image Editingcited by 417
- 2025OmniGen2: Towards Instruction-Aligned Multimodal Generationcited by 241
- 2023VIEScore: Towards Explainable Metrics for Conditional Image Synthesis Evaluationcited by 196
- 2024OmniEdit: Building Image Editing Generalist Models Through Specialist Supervisioncited by 145
- 2025Emu3.5: Native Multimodal Models are World Learnerscited by 119
- 2025LongCat-Image Technical Reportcited by 92
- 2025EditScore: Unlocking Online RL for Image Editing via High-Fidelity Reward Modelingcited by 61
- 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapsecited by 29
- 2020Autoencoding Variational Autoencodercited by 24
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026Improved Baselines with Representation Autoencoderscited by 24
- 2026What Matters for Diffusion-Friendly Latent Manifold? Prior-Aligned Autoencoders for Latent Diffusioncited by 7
- 2026WavCube: Unifying Speech Representation for Understanding and Generation via Semantic-Acoustic Joint Modelingcited by 3
- 2026Beyond the Last Layer: Multi-Layer Representation Fusion for Visual Tokenizationcited by 3
- 2026DecQ: Detail-Condensing Queries for Enhanced Reconstruction and Generation in Representation Autoencoderscited by 2
- 2026PixelDiT2: Representation-Grounded Pixel Diffusion Transformerscited by 0
- 2026HiRAE: Hierarchical Representation Autoencoding with Residual Budgetscited by 0
- 2026HiRAE: Hierarchical Representation Autoencoding with Residual Budgets29 Sep 2026
- 2026PixelDiT2: Representation-Grounded Pixel Diffusion Transformers21 Sep 2026
- 2026DecQ: Detail-Condensing Queries for Enhanced Reconstruction and Generation in Representation Autoencoders21 May 2026
- 2026Improved Baselines with Representation Autoencoders18 May 2026
- 2026Beyond the Last Layer: Multi-Layer Representation Fusion for Visual Tokenization11 May 2026
- 2026What Matters for Diffusion-Friendly Latent Manifold? Prior-Aligned Autoencoders for Latent Diffusion8 May 2026
- 2026WavCube: Unifying Speech Representation for Understanding and Generation via Semantic-Acoustic Joint Modeling7 May 2026
Similar, not cited
Close in wording, no citation either way
- 2025Boosting Generative Image Modeling via Joint Image-Feature Synthesis
- 2025AlignTok: Aligning Visual Foundation Encoders to Tokenizers for Diffusion Models
- 2025Diffuse and Disperse: Image Generation with Representation Regularization
- 2025Masked Autoencoders Are Effective Tokenizers for Diffusion Models
- 2025Generalization of Diffusion Models Arises with a Balanced Representation Space
- 2022Representation Learning with Diffusion Models
- 2025MADI: Masking-Augmented Diffusion with Inference-Time Scaling for Visual Editing
- 2024Epsilon-VAE: Denoising as Visual Decoding