RiT: Vanilla Diffusion Transformers Suffice in Representation Space

Le Zhang, Ning Mang, Aishwarya Agrawal

Flow matching with $x$-prediction -- regressing the clean data point rather than the ambient velocity -- is known to exploit low-dimensional manifold structure effectively in pixel…