arXiv:2508.14807cs.LG2025-08被引 10

通过调整源分布实现生成模型引导,保持原向量场不变

Source-Guided Flow Matching

  • 直接修改源分布而非向量场,保留预训练模型结构
  • 理论证明可精确恢复目标分布,误差有界
  • 灵活适配不同采样方法,适合逆问题与物理建模

生成模型的引导通常通过添加引导场来修改概率流向量场实现。本文提出源引导流匹配(SGFM)框架,不改变预训练向量场,而是直接调整源分布,将引导问题转化为从源分布采样的明确问题。理论上,SGFM 可精确恢复目标分布;当使用近似源分布采样器和近似向量场时,我们给出了生成分布的 Wasserstein 误差上界。该方法的优势在于用户可根据具体任务灵活选择采样策略。我们系统比较了多种采样方法,并讨论渐近精确引导的条件。此外,本框架与最优流匹配模型兼容,因向量场生成的直线传输映射得以保留。在二维合成基准、物理信息生成任务及成像逆问题上的实验验证了该框架的有效性与灵活性。

原文摘要 · Abstract (English)

Guidance of generative models is typically achieved by modifying the probability flow vector field through the addition of a guidance field. In this paper, we instead propose the Source-Guided Flow Matching (SGFM) framework, which modifies the source distribution directly while keeping the pre-trained vector field intact. This reduces the guidance problem to a well-defined problem of sampling from the source distribution. We theoretically show that SGFM recovers the desired target distribution exactly. Furthermore, we provide bounds on the Wasserstein error for the generated distribution when using an approximate sampler of the source distribution and an approximate vector field. The key benefit of our approach is that it allows the user to flexibly choose the sampling method depending on their specific problem. To illustrate this, we systematically compare different sampling methods and discuss conditions for asymptotically exact guidance. Moreover, our framework integrates well with optimal flow matching models since the straight transport map generated by the vector field is preserved. Experimental results on synthetic 2D benchmarks, physics-informed generative tasks, and imaging inverse problems demonstrate the effectiveness and flexibility of the proposed framework.

生成模型流匹配源引导逆问题

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。