LoopREG: Representation Supervision Enables Test-Time Recurrent Depth in Diffusion Transformers

Zipeng Wang, Anping Xiong, Jing Zhang +17

Reusing Transformer blocks offers a way to increase diffusion-model computation at test time, but naive recurrence can degrade generation beyond its training depth. We introduce Lo…