paper-with-me

홈 › Papers

On The Hidden Biases of Flow Matching Samplers

2025-12-18 · Soon Hoe Lim arxiv

Flow matching (FM) constructs continuous-time ODE samplers by prescribing probability paths between a base distribution and a target distribution. In this note, we study FM through the lens of finite-sample plug-in estimation. In addition to replacing population expectations by sample averages, one may replace the target distribution itself by a finite-sample surrogate, ranging from the empirical measure to a smoothed estimator. This viewpoint yields a natural hierarchy of empirical FM models. For affine conditional flows, we derive the exact empirical minimizer and identify a smoothed plug-in regime in which the terminal law is exactly a kernel-mixture estimator. This plug-in perspective clarifies several coupled finite-sample biases of empirical FM. First, replacing the target law by a finite-sample surrogate changes the statistical target. Second, the empirical minimizer is generally not a gradient field, even when each conditional flow is. Third, a fixed empirical marginal path does not determine a unique particle dynamics: one may add extra vector fields whose probability flux has zero divergence without changing the marginal path. For Gaussian affine conditional paths, we give explicit families of such flux-null corrections. Finally, the source distribution provides a primary mechanism controlling upper tails of kinetic energy. In particular, Gaussian bases yield exponential upper-tail bounds for instantaneous and integrated kinetic energies, whereas polynomially tailed bases yield corresponding polynomial upper-tail bounds.

📄 PDF Abstract BibTeX arXiv:2512.16768

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Asymptotic-Preserving A Posteriori Analysis of Diffusion and Flow-Matching Samplers

2026-07-05 · Shiheng Zhang arxiv

Diffusion and flow-matching samplers integrate a learned probability-flow ODE from a large noise scale down to a small terminal floor $σ_{\min}$, at which the score is stiff and the flow develops a boundary layer. We tre…

Fast Samplers for Inverse Problems in Iterative Refinement Models

2024-05-27 · Kushagra Pandey, Ruihan Yang, Stephan Mandt

Constructing fast samplers for unconditional diffusion and flow-matching models has received much attention recently; however, existing methods for solving inverse problems, such as super-resolution, inpainting, or deblu…

DeblurringImage RestorationSuper-Resolution

Efficient Identification of Critical Transitions via Flow Matching: A Scalable Generative Approach for Many-Body Systems

2025-08-21 · Qian-Rui Lee, Daw-Wei Wang arxiv

We propose a machine learning framework based on Flow Matching (FM) to identify critical properties in many-body systems efficiently. Using the 2D XY model as a benchmark, we demonstrate that a single network, trained on…

Convergence Analysis for General Probability Flow ODEs of Diffusion Models in Wasserstein Distances

2024-01-31 · Xuefeng Gao, Lingjiong Zhu

Score-based generative modeling with probability flow ordinary differential equations (ODEs) has achieved remarkable success in a variety of applications. While various fast ODE-based samplers have been proposed in the l…

Precise: SDE-Consistent Stochastic Sampling for RL Post-Training of Flow-Matching Models

2026-05-22 · Jade Zou, Tao Huang, Weijie Kong, Junzhe Li 외 arxiv

Reinforcement learning (RL) has become an effective way to improve prompt alignment and perceptual quality in diffusion and flow-matching generators. A critical step for applying online RL to flow matching is turning the…

Reinforcement Learning