English

FlowLM: Few-Step Language Modeling via Diffusion-to-Flow Adaptation

Computation and Language 2026-05-21 v1 Artificial Intelligence

Abstract

We present FlowLM, a flow matching language model transformed from pre-trained diffusion language models via efficient fine-tuning. By re-aligning the curved sampling trajectories of diffusion models into straight-line flows, FlowLM enables high quality few-step generation that rivals or even outperforms the quality of 2,000-step diffusion sampling with very few training epochs. Remarkably, finetuned FlowLM reaches performance saturation with only half as many training epochs as training from scratch, both approaches greatly outperforming the original diffusion model, thereby validating our method. Furthermore, we validate a more effective training objective for flow matching: predicting clean data to consistently guide the sampling process towards the true data distribution. Empirical results demonstrate that our approach is highly effective for high-quality, few-step text generation.

Keywords

Cite

@article{arxiv.2605.20199,
  title  = {FlowLM: Few-Step Language Modeling via Diffusion-to-Flow Adaptation},
  author = {Runzhe Zhang and Letian Chen and Wenpeng Zhang and Zhouhan Lin and Peilin Zhao},
  journal= {arXiv preprint arXiv:2605.20199},
  year   = {2026}
}

Comments

26 pages, 11 figures