Mar, 2024

ENOT: 快速准确的神经最优输运训练的 Expectile 正则化

TL;DR通过对共轭势进行特定正则化,我们提出了一种用于神经最优输运(NOT)训练过程的新扩展,能够准确高效地估计最优传输方案。我们通过引入期望损失函数的正则化来解决现有 NOT 求解器的主要瓶颈问题,从而实现对学习双重潜力的限制条件,提供共轭势可能分布的上界估计,使学习过程稳定,消除了额外的广泛微调的需求。我们称之为期望损失正则化神经最优输运(ENOT)的方法在 Wasserstein-2 基准任务上明显优于先前的最新技术方法(质量提高 3 倍,运行时间提高 10 倍)。