mlsanityData to 14 Oct 2026

Training-Free Representation Guidance for Diffusion Models with a Representation Alignment Projector

arXiv 2601.22468 ↗First posted 30 Jan 2026cs.CV cs.AIcited by 2

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Recent progress in generative modeling has enabled high-quality visual synthesis with diffusion-based frameworks, supporting controllable sampling and large-scale training. Inference-time guidance methods such as classifier-free and representative guidance enhance semantic alignment by modifying sampling dynamics; however, they do not fully exploit unsupervised feature representations. Although such visual representations contain rich semantic structure, their integration during generation is constrained by the absence of ground-truth reference images at inference. This work reveals semantic drift in the early denoising stages of diffusion transformers, where stochasticity results in inconsistent alignment even under identical conditioning. To mitigate this issue, we introduce a guidance scheme using a representation alignment projector that injects representations predicted by a projector into intermediate sampling steps, providing an effective semantic anchor without modifying the model architecture. Experiments on SiTs and REPAs show notable improvements in class-conditional ImageNet synthesis, achieving substantially lower FID scores; for example, REPA-XL/2 improves from 5.9 to 3.3, and the proposed method outperforms representative guidance when applied to SiT models. The approach further yields complementary gains when combined with classifier-free guidance, demonstrating enhanced semantic coherence and visual fidelity. These results establish representation-informed diffusion sampling as a practical strategy for reinforcing semantic preservation and image consistency.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 23 of its 27 references that are in the map. First posted on arXiv 30 Jan 2026.

Builds on

23 of 27 references.

Closest ancestors · 14

Same direction, newest first

  1. 2025Towards a Golden Classifier-Free Guidance Path via Foresight Fixed Point Iterations24 Oct
  2. 2025Diffusion Transformers with Representation Autoencoders13 Oct
  3. 2024Golden Noise for Diffusion Models: A Learning Framework14 Nov
  4. 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think9 Oct
  5. 2024Guiding a Diffusion Model with a Bad Version of Itself4 Jun
  6. 2024Applying Guidance in a Limited Interval Improves Sample and Distribution Quality in Diffusion Models11 Apr
  7. 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers16 Jan
  8. 2023Universal Guidance for Diffusion Models14 Feb
  9. 2022Scalable Diffusion Models with Transformers19 Dec
  10. 2022Classifier-Free Diffusion Guidance26 Jul
  11. 2022Entropy-driven Sampling and Training Scheme for Conditional Diffusion Generation23 Jun
  12. 2021Diffusion Models Beat GANs on Image Synthesis11 May
2 more
  1. 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
  2. 2020Denoising Diffusion Probabilistic Models19 Jun

Other references · 9

  1. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  2. 2022Flow Matching for Generative Modelingcited by 5,702
  3. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
  4. 2022Building Normalizing Flows with Stochastic Interpolantscited by 995
  5. 2023Attend-and-Excite: Attention-Based Semantic Guidance for Text-to-Image Diffusion Modelscited by 702
  6. 2024InitNO: Boosting Text-to-Image Diffusion Models via Initial Noise Optimizationcited by 123
  7. 2023Return of Unconditional Generation: A Self-supervised Representation Generation Methodcited by 98
  8. 2024Zigzag Diffusion Sampling: Diffusion Models Can Self-Improve via Self-Reflectioncited by 41
  9. 2025SAGA: Learning Signal-Aligned Distributions for Improved Text-to-Image Generationcited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Learning Diffusion Models with Flexible Representation Guidance
  2. 2025AlignTok: Aligning Visual Foundation Encoders to Tokenizers for Diffusion Models
  3. 2024Diffusion Models Need Visual Priors for Image Generation
  4. 2026Representation by Design in Generation: Cross-View Class-Token Alignment in Diffusion Transformers
  5. 2026Test-Time Conditioning with Representation-Aligned Visual Features
  6. 2025Boosting Generative Image Modeling via Joint Image-Feature Synthesis
  7. 2025SARA: Structural and Adversarial Representation Alignment for Training-efficient Diffusion Models
  8. 2025REPA Works Until It Doesn't: Early-Stopped, Holistic Alignment Supercharges Diffusion Training