Efficient Generative Model Training via Embedded Representation Warmup
Not on the map. Why?
Abstract (arXiv)
Generative models face a fundamental challenge: they must simultaneously learn high-level semantic concepts (what to generate) and low-level synthesis details (how to generate it). Conventional end-to-end training entangles these distinct, and often conflicting objectives, leading to a complex and inefficient optimization process. We argue that explicitly decoupling these tasks is key to unlocking more effective and efficient generative modeling. To this end, we propose Embedded Representation Warmup (ERW), a principled two-phase training framework. The first phase is dedicated to building a robust semantic foundation by aligning the early layers of a diffusion model with a powerful pretrained encoder. This provides a strong representational prior, allowing the second phase -- generative full training with alignment loss to refine the representation -- to focus its resources on high-fidelity synthesis. Our analysis confirms that this efficacy stems from functionally specializing the model's early layers for representation. Empirically, our framework achieves a 11.5× speedup in 350 epochs to reach FID=1.41 compared to single-phase methods like REPA. Code is available at https://github.com/LINs-lab/ERW.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 61 of its 72 references that are in the map. First posted on arXiv 14 Apr 2025.
Builds on
61 of 72 references.
References · 61
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
38 more
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,475
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,988
- 2016Improved Techniques for Training GANscited by 2,970
- 2020Improved Baselines with Momentum Contrastive Learningcited by 2,287
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,979
- 2023Consistency Modelscited by 1,822
- 2019Similarity of Neural Network Representations Revisitedcited by 1,755
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
- 2018On the Spectral Bias of Neural Networkscited by 1,341
- 2021Cascaded Diffusion Models for High Fidelity Image Generationcited by 1,163
- 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 950
- 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 944
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 939
- 2023Adversarial Diffusion Distillationcited by 759
- 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
- 2017The Perception-Distortion Tradeoffcited by 587
- 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 502
- 2023Simple diffusion: End-to-end diffusion for high resolution imagescited by 422
- 2023Analyzing and Improving the Training Dynamics of Diffusion Modelscited by 402
- 2024The Platonic Representation Hypothesiscited by 382
- 2023Your Diffusion Model is Secretly a Zero-Shot Classifiercited by 306
- 2023Understanding Diffusion Objectives as the ELBO with Simple Data Augmentationcited by 304
- 2021Generating Images with Sparse Representationscited by 266
- 2023MDTv2: Masked Diffusion Transformer is a Strong Image Synthesizercited by 262
- 2024Applying Guidance in a Limited Interval Improves Sample and Distribution Quality in Diffusion Modelscited by 254
- 2022MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesiscited by 227
- 2021Projected GANs Converge Fastercited by 208
- 2023Generalization in diffusion models arises from geometry-adaptive harmonic representationscited by 185
- 2023Fast Training of Diffusion Models with Masked Transformerscited by 176
- 2023Diffusion Schr"odinger Bridge Matchingcited by 153
- 2023Denoising Diffusion Autoencoders are Unified Self-supervised Learnerscited by 130
50 most cited of 61 All 61 (JSON)
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026Self-Adversarial One Step Generation via Condition Shiftingcited by 2
- 2026Data Warmup: Complexity-Aware Curricula for Efficient Diffusion Trainingcited by 0
- 2026PixelDense: Dense Prediction as Representation Alignment for Pixel Diffusioncited by 0
- 2026Just Align bmx: Aligning Predictions, Not Representationscited by 0
- 2026From Pixels, Without Pre-training: Joint Generative and Self-Supervised Representation Learning in One Modelcited by 0
- 2026UltraText Bench: A Comprehensive Bilingual Benchmark for Evaluating Visual Text Rendering in Image Generationcited by 0
- 2026UltraText Bench: A Comprehensive Bilingual Benchmark for Evaluating Visual Text Rendering in Image Generation7 Oct 2026
- 2026From Pixels, Without Pre-training: Joint Generative and Self-Supervised Representation Learning in One Model5 Oct 2026
- 2026Just Align bmx: Aligning Predictions, Not Representations30 Sep 2026
- 2026PixelDense: Dense Prediction as Representation Alignment for Pixel Diffusion30 Sep 2026
- 2026Self-Adversarial One Step Generation via Condition Shifting14 Apr 2026
- 2026Data Warmup: Complexity-Aware Curricula for Efficient Diffusion Training8 Apr 2026
Similar, not cited
Close in wording, no citation either way
- 2025Repulsor: Accelerating Generative Modeling with a Contrastive Memory Bank
- 2025Boosting Generative Image Modeling via Joint Image-Feature Synthesis
- 2025Diffusion Transformers with Representation Autoencoders
- 2026Representation-Conditioned Diffusion Models for Guided Training Data Generation
- 2026Training-Free Representation Guidance for Diffusion Models with a Representation Alignment Projector
- 2025Generalization of Diffusion Models Arises with a Balanced Representation Space
- 2026Representation Learning in Diffusion and Flow-based Model: An Application Aspect
- 2024On Improved Conditioning Mechanisms and Pre-training Strategies for Diffusion Models