Equivalent Flows, Unequal Learning: Clean-Latent Prediction in Transformers

Funing Fu, Tenghui Wang, Guanyu Zhou +2

Flow samplers consume velocity, but the neural network can predict the clean endpoint and convert it to velocity through a fixed affine readout. We study this choice with JLT, a la…