cs.LGSep 3, 2026

Beyond Straightness: Non-Crossing Flow Matching via Quantile AlignTree Coupling

Authors: Junyi LinMengyu LiJingxuan HuKejun HeCheng Meng

Organizations: Institute of Statistics and Big Data, Renmin University of China, Beijing, China · Department of Statistics and Data Science, Tsinghua University, Beijing, China

Abstract

The performance of Flow Matching largely depends on the quality of the coupling between the source and target distributions. However, independent coupling often leads to path crossings and local velocity ambiguity, while OT-based couplings typically incur high construction costs. To address this challenge, we propose Quantile AlignTree Flow Matching (QAT-FM), an efficient structured coupling strategy that constructs a hierarchical coupling between a Gaussian prior and the target data distribution via a quantile-aligned tree structure. QAT-FM constructs the coupling in O(NdlogN)\mathcal{O}(Nd\log N) time and supports per-pair source sampling with O(d)\mathcal{O}(d) complexity, enabling scalable training for large-scale high-dimensional generative tasks. Theoretically, we prove that the QAT coupling satisfies marginal consistency, induces non-crossing linear interpolation paths, and consistently improves path separation at intermediate times compared with independent coupling, thereby alleviating local velocity ambiguity. QAT-FM further extends naturally to conditional generation, enabling structured conditional coupling while preserving global Gaussian alignment. Experiments across diverse benchmark datasets demonstrate that QAT-FM achieves competitive generative performance while substantially reducing coupling construction cost.

Explore similar work

CardsList
  1. Structured Coupling for Flow Matching

    May 8, 2026Xavier Sumba, Carles Balsells-Rodas, Yingzhen LiLatent Variable

  2. Optimal Transport Flow Matching by Design

    Jun 2, 2026Shimon Malnick, Matan Rusanovsky, Ohad Fried +1Hierarchical Flow MatchingGenerative Priors