arXiv:2602.02241cs.LG2026-02被引 1

提出新方法提升连续空间熵正则最优传输的计算效率与效果。

Variational Entropic Optimal Transport

  • 通过变分重写对数归一化项,获得可微学习目标。
  • 在合成数据和图像翻译任务中达到更优或相当的生成质量。
  • 无需马尔可夫链蒙特卡洛模拟,适合大规模神经网络训练。

连续空间中基于二次代价的熵正则最优传输(EOT)是解决域转换问题的经典工具。现有方法通常优化依赖单个势函数的弱对偶目标,但因对数归一化项不可计算而效率低下。传统方法要么严重限制传输族以获得闭式归一化(如高斯混合参数化),要么采用通用神经参数化并需基于模拟的训练流程。本文提出变分熵正则最优传输(VarEOT),基于对数期望 $\\(log \\mathbb{E}[\\exp(\\cdot)]$ 的精确变分重构,将其转化为可处理的辅助对数归一化器最小化问题。该方法得到可微学习目标,支持随机梯度优化,训练中无需马尔可夫链蒙特卡洛模拟。理论分析包含有限样本泛化界及通用函数逼近下的近似结果。在合成数据和无配对图像翻译实验中,性能优于或相当现有方法;在使用相同弱对偶目标的求解器间对比,验证了所提优化原则的优势。代码见:https://github.com/DrEternity/VarEOT。

原文摘要 · Abstract (English)

Entropic optimal transport (EOT) in continuous spaces with quadratic cost is a classical tool for solving the domain translation problem. In practice, recent approaches optimize a weak dual EOT objective depending on a single potential, but doing so is computationally not efficient due to the intractable log-partition term. Existing methods typically resolve this obstacle in one of two ways: by significantly restricting the transport family to obtain closed-form normalization (via Gaussian-mixture parameterizations), or by using general neural parameterizations that require simulation-based training procedures. We propose Variational Entropic Optimal Transport (VarEOT), based on an exact variational reformulation of the log-partition $\log \mathbb{E}[\exp(\cdot)]$ as a tractable minimization over an auxiliary log-normalizer. This yields a differentiable learning objective optimized with stochastic gradients and avoids the necessity of MCMC simulations during the training. We provide theoretical guarantees, including finite-sample generalization bounds and approximation results under universal function approximation. Experiments on synthetic data and unpaired image-to-image translation demonstrate competitive or improved translation quality, while comparisons within the solvers that use the same weak dual EOT objective support the benefit of the proposed optimization principle. The code for our solver can be found at https://github.com/DrEternity/VarEOT .

最优传输变分推断图像生成

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