[2512.15657] SoFlow: Solution Flow Models for One-Step Generative Modeling

Computer Science > Machine Learning arXiv:2512.15657 (cs) [Submitted on 17 Dec 2025 (v1), last revised 1 Mar 2026 (this version, v2)] Title:SoFlow: Solution Flow Models for One-Step Generative Modeling Authors:Tianze Luo, Haotian Yuan, Zhuang Liu View a PDF of the paper titled SoFlow: Solution Flow Models for One-Step Generative Modeling, by Tianze Luo and 2 other authors View PDF HTML (experimental) Abstract:The multi-step denoising process in diffusion and Flow Matching models causes major efficiency issues, which motivates research on few-step generation. We present Solution Flow Models (SoFlow), a framework for one-step generation from scratch. By analyzing the relationship between the velocity function and the solution function of the velocity ordinary differential equation (ODE), we propose a Flow Matching loss and a solution consistency loss to train our models. The Flow Matching loss allows our models to provide estimated velocity fields for Classifier-Free Guidance (CFG) during training, which improves generation performance. Notably, our consistency loss does not require the calculation of the Jacobian-vector product (JVP), a common requirement in recent works that is not well-optimized in deep learning frameworks like PyTorch. Experimental results indicate that, when trained from scratch using the same Diffusion Transformer (DiT) architecture and an equal number of training epochs, our models achieve better FID-50K scores than MeanFlow models on the ImageNet 256...