mlsanityData to 14 Oct 2026

Efficient Generative Model Training via Embedded Representation Warmup

arXiv 2504.10188 ↗First posted 14 Apr 2025cs.LG cs.AIcited by 6

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

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  3. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  4. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  5. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  6. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  7. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
  8. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  9. 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
  10. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  11. 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
  12. 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
38 more
  1. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
  2. 2022Scalable Diffusion Models with Transformerscited by 6,054
  3. 2022Flow Matching for Generative Modelingcited by 5,702
  4. 2015Rethinking the Inception Architecture for Computer Visioncited by 5,475
  5. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  6. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
  7. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
  8. 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,988
  9. 2016Improved Techniques for Training GANscited by 2,970
  10. 2020Improved Baselines with Momentum Contrastive Learningcited by 2,287
  11. 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,979
  12. 2023Consistency Modelscited by 1,822
  13. 2019Similarity of Neural Network Representations Revisitedcited by 1,755
  14. 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
  15. 2018On the Spectral Bias of Neural Networkscited by 1,341
  16. 2021Cascaded Diffusion Models for High Fidelity Image Generationcited by 1,163
  17. 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 950
  18. 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 944
  19. 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 939
  20. 2023Adversarial Diffusion Distillationcited by 759
  21. 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 737
  22. 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 703
  23. 2017The Perception-Distortion Tradeoffcited by 587
  24. 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 502
  25. 2023Simple diffusion: End-to-end diffusion for high resolution imagescited by 422
  26. 2023Analyzing and Improving the Training Dynamics of Diffusion Modelscited by 402
  27. 2024The Platonic Representation Hypothesiscited by 382
  28. 2023Your Diffusion Model is Secretly a Zero-Shot Classifiercited by 306
  29. 2023Understanding Diffusion Objectives as the ELBO with Simple Data Augmentationcited by 304
  30. 2021Generating Images with Sparse Representationscited by 266
  31. 2023MDTv2: Masked Diffusion Transformer is a Strong Image Synthesizercited by 262
  32. 2024Applying Guidance in a Limited Interval Improves Sample and Distribution Quality in Diffusion Modelscited by 254
  33. 2022MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesiscited by 227
  34. 2021Projected GANs Converge Fastercited by 208
  35. 2023Generalization in diffusion models arises from geometry-adaptive harmonic representationscited by 185
  36. 2023Fast Training of Diffusion Models with Masked Transformerscited by 176
  37. 2023Diffusion Schr"odinger Bridge Matchingcited by 153
  38. 2023Denoising Diffusion Autoencoders are Unified Self-supervised Learnerscited by 130

50 most cited of 61 All 61 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 6.

6 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Repulsor: Accelerating Generative Modeling with a Contrastive Memory Bank
  2. 2025Boosting Generative Image Modeling via Joint Image-Feature Synthesis
  3. 2025Diffusion Transformers with Representation Autoencoders
  4. 2026Representation-Conditioned Diffusion Models for Guided Training Data Generation
  5. 2026Training-Free Representation Guidance for Diffusion Models with a Representation Alignment Projector
  6. 2025Generalization of Diffusion Models Arises with a Balanced Representation Space
  7. 2026Representation Learning in Diffusion and Flow-based Model: An Application Aspect
  8. 2024On Improved Conditioning Mechanisms and Pre-training Strategies for Diffusion Models