中文

Drift Flow Matching

机器学习 2026-05-19 v1 人工智能

摘要

Iterative generative models such as Flow Matching and Diffusion models have demonstrated strong test-time scaling behavior, where additional inference computation can improve generation quality. In contrast, Drift Models offer efficient one-step generation, but their direct generation paradigm limits such flexibility. In this work, we propose Drift Flow Matching (DFM), a framework that connects drifting generative modeling with flow-based iterative generation. DFM preserves the efficiency of direct transport maps while enabling generation to be refined through multiple inference steps when desired. This bridges the gap between one-step Drift Models and multi-step Flow Matching methods, and provides a novel generative paradigm that can adapt sampling computation to different quality--efficiency requirements. Extensive experiments across different tasks and datasets demonstrate the effectiveness and generality of the proposed framework.

关键词

引用

@article{arxiv.2605.17244,
  title  = {Drift Flow Matching},
  author = {Chenrui Ma and Xi Xiao and Lin Zhao and Tianyang Wang and Ferdinando Fioretto and Yanning Shen},
  journal= {arXiv preprint arXiv:2605.17244},
  year   = {2026}
}