LLM2D
基于半不平衡神经最优传输的鲁棒重心估计
Robust Barycenter Estimation using Semi-Unbalanced Neural Optimal Transport
作者: Milena Gazdieva, Jaemoo Choi, Alexander Kolesov, Jaewoong Choi, Petr Mokrov, Alexander Korotin
发布日期: 10/8/2024
arXiv ID: oai:arXiv.org:2410.03974v1

摘要

从多个数据源聚合数据的一个常见挑战可以形式化为一个 *最优传输* (OT) 重心问题,该问题旨在计算概率分布关于 OT 差异的平均值。然而,数据度量中异常值和噪声的存在会严重阻碍传统统计方法估计 OT 重心的性能。为了解决这个问题,我们提出了一种新颖的可扩展方法来估计 *鲁棒* 连续重心,利用 *(半)不平衡* OT 问题的对偶公式。据我们所知,本文首次尝试在连续分布设置下开发鲁棒重心算法。我们的方法被构建为一个 $\min$-$\max$ 优化问题,并且适应于 *一般* 成本函数。我们严格建立了所提出方法的理论基础,并通过大量说明性实验证明了其对异常值和类不平衡的鲁棒性。