Improving the Training of Rectified Flows - Lee, Sangyun; Lin, Zinan; Fanti, Giulia (Advances in Neural Information Processing Systems, 2024)