研究提出QAT-FM:用分位数对齐树构建流匹配耦合,降低路径交叉与构建成本

arXiv新论文提出Quantile AlignTree Flow Matching,在O(Nd log N)时间内构建高斯先验与目标分布的层次化耦合,支持O(d)复杂度逐对采样,理论上证明可诱导非交叉线性插值路径。

AI解读:这篇论文要解决的是流匹配(Flow Matching)生成模型中的一个核心矛盾:模型表现高度依赖源分布与目标分布之间的“耦合”质量,但现有方案两头不讨好——独立耦合计算便宜却容易让多条生成路径交叉,导致局部速度方向混乱、训练不稳定;基于最优传输(OT)的耦合路径干净,但构建成本高,不适合大规模高维数据。研究者提出的QAT-FM采用分位数对齐的树结构,把高斯先验到目标分布的映射组织成层次化耦合,将构建复杂度降到O(Nd log N),且支持对每个样本对以O(d)复杂度独立采样,理论上还证明了这种耦合满足边际一致性、能诱导非交叉的线性插值路径,从而缓解速度歧义。对普通读者来说,这项研究的意义在于为高分辨率图像、三维数据等大规模生成任务提供一种更省算力且更稳定的训练方案;对研究者和工程师而言,如果当前正被流匹配训练中的路径交叉或耦合构建开销困扰,QAT-FM给出了一个可直接实现的理论框架——但论文目前只报告了基准测试上的性能,没有公开与既有方法在主流数据集上的对比细节,实际收益仍需复现验证,现阶段不必急于切换技术栈。

arXiv预印本网站9月3日发布论文,提出Quantile AlignTree Flow Matching(QAT-FM),一种用于流匹配生成模型的耦合构建策略。论文由Junyi Lin、Mengyu Li、Jingxuan Hu、Kejun He和Cheng Meng撰写,称QAT-FM可在O(Nd log N)时间内构建耦合,并以O(d)复杂度支持逐对源采样。

方法:分位数对齐树构建层次耦合

论文摘要指出,流匹配的性能很大程度上取决于源分布与目标分布之间的耦合质量,但独立耦合常导致路径交叉和局部速度歧义,而基于最优传输(OT)的耦合构建成本高。QAT-FM通过分位数对齐的树结构,在高斯先验与目标数据分布之间构建层次化耦合,以降低构建开销。

理论部分,作者证明QAT耦合满足边际一致性,能诱导非交叉的线性插值路径,并在中间时间点相比独立耦合持续改善路径分离,从而缓解局部速度歧义。论文还称QAT-FM可自然扩展到条件生成,在保持全局高斯对齐的同时实现结构化条件耦合。

  • 耦合构建时间复杂度:O(Nd log N)
  • 逐对源采样复杂度:O(d)
  • 实验在多个基准数据集上进行,称取得有竞争力的生成性能,同时大幅降低耦合构建成本

信息来源