arXiv:2603.06972cs.LGcs.CV2026-03

提出抗异常值的条件生成模型,提升数据分布匹配稳定性。

Conditional Unbalanced Optimal Transport Maps: An Outlier-Robust Framework for Conditional Generative Modeling

  • 用散度惩罚放松条件分布硬匹配,保留条件边缘
  • 在合成与图像数据上显著提升对异常值的鲁棒性
  • 适合小样本或含噪声的条件生成任务

条件最优传输(COT)旨在寻找条件源分布与目标分布间的传输映射,以最小化传输代价。近年来,该映射被用于条件生成建模,实现分布间高效映射。然而,经典COT继承了最优传输对异常值敏感的固有缺陷,源于严格的分布匹配约束。这一问题在条件设置下更为突出,因每个条件分布均由有限数据子集估计得出。为此,本文提出条件非平衡最优传输(CUOT)框架,通过引入Csiszár散度惩罚松弛条件分布匹配约束,同时严格保持条件边缘分布。我们建立了CUOT问题的严谨形式,并推导其对偶与半对偶形式。基于半对偶形式,提出条件非平衡最优传输映射(CUOTM),一种基于三角c-变换参数化的抗异常值条件生成模型。理论上证明了该参数化有效性,即最优三角映射满足c-变换关系。在二维合成数据与图像尺度数据集上的实验表明,CUOTM相比现有基于COT的基线方法,在异常值鲁棒性和分布匹配性能上表现更优,同时保持高采样效率。

原文摘要 · Abstract (English)

Conditional Optimal Transport (COT) problem aims to find a transport map between conditional source and target distributions while minimizing the transport cost. Recently, these transport maps have been utilized in conditional generative modeling tasks to establish efficient mappings between the distributions. However, classical COT inherits a fundamental limitation of optimal transport, i.e., sensitivity to outliers, which arises from the hard distribution matching constraints. This limitation becomes more pronounced in a conditional setting, where each conditional distribution is estimated from a limited subset of data. To address this, we introduce the Conditional Unbalanced Optimal Transport (CUOT) framework, which relaxes conditional distribution-matching constraints through Csiszár divergence penalties while strictly preserving the conditioning marginals. We establish a rigorous formulation of the CUOT problem and derive its dual and semi-dual formulations. Based on the semi-dual form, we propose Conditional Unbalanced Optimal Transport Maps (CUOTM), an outlier-robust conditional generative model built upon a triangular $c$-transform parameterization. We theoretically justify the validity of this parameterization by proving that the optimal triangular map satisfies the $c$-transform relationships. Our experiments on 2D synthetic and image-scale datasets demonstrate that CUOTM achieves superior outlier robustness and competitive distribution-matching performance compared to existing COT-based baselines, while maintaining high sampling efficiency.

生成模型最优传输异常值鲁棒

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