Shuo Jiang

What Visual Generators Need from Teachers: Rethinking Representation Alignment

Representation alignment speeds up diffusion transformer training by pulling an intermediate block of the model (student) toward features of a frozen pretrained encoder (teacher). Which teacher layer to align, and for how long, is still set by convention, and each alternative costs a training run. We find that alignmen...