用切片最优传输作先验,提升传统方法精度
Sliced-Regularized Optimal Transport

- 以切片最优传输计划为先验,正则化运输方案
- 相同正则强度下,逼近真实运输方案更精准
- 适合需要高精度传输的图像处理与流形学习
我们提出一种新型正则化最优传输(OT)框架,称为切片-正则化最优传输(SROT)。与熵正则化OT(EOT)不同,SROT将运输计划正则化至平滑后的切片最优传输(SOT)计划。据我们所知,SROT是首个利用SOT计划作为参考来改进经典OT的方法。本文给出了SROT的形式定义,推导其对偶形式,并提供其后贝叶斯解释。进一步开发了类似Sinkhorn的算法,保持与EOT相当的可扩展性。通过引入可扩展的SOT计划作为先验,SROT在相同正则化水平下比EOT更准确地逼近精确OT计划,且优于参考的SOT计划本身。我们还提出了由SROT诱导的相应OT散度,称为SROT散度,并分析其拓扑与计算性质。实验在合成数据集和颜色迁移任务中验证了该方法优于EOT和SOT,梯度流实验进一步凸显SROT散度的优势。
原文摘要 · Abstract (English)
We propose a new regularized optimal transport (OT) formulation, termed sliced-regularized optimal transport (SROT). Unlike entropic OT (EOT), which regularizes the transport plan toward an independent coupling, SROT regularizes it toward a smoothened sliced OT (SOT) plan. To the best of our knowledge, SROT is the first approach to leverage a version of SOT plan as a reference to improve classical OT. We provide a formal definition of SROT, derive its dual formulation, and provide a post-Bayesian interpretation of SROT. We then develop a Sinkhorn-style algorithm for efficient computation, retaining the same scalability advantages as EOT. By incorporating a scalable SOT plan as a prior, SROT yields more accurate approximations of the exact OT plan than EOT under the same level of regularization. Moreover, the resulting transport plan improves upon the reference SOT plan itself. We further introduce the corresponding OT divergence induced by SROT, named SROT divergence, and analyze its topological and computational properties. Finally, we validate our approach through experiments on synthetic datasets and color transfer tasks, demonstrating that SROT is better than both EOT and SOT in approximating exact OT. Additional experiments on gradient flows further highlight the advantages of SROT divergence.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。