流式切片最优传输
机器学习
2026-05-12 v4 统计计算
统计方法学
机器学习
摘要
切片最优传输 (SOT) 或切片 Wasserstein (SW) 距离因其统计与计算可扩展性而广为人知。本工作提出首个从样本流估计 SW 的方法,称为流式切片 Wasserstein (Stream-SW)。为定义 Stream-SW,我们首先引入 1 维 Wasserstein 距离 (1DW) 的流式估计器。由于 1DW 可由比较两个分布分位数函数绝对差的积分给出闭式表达式,我们利用分位数近似技术来定义流式 1DW 估计器。将流式 1DW 应用于所有投影后即可获得 Stream-SW。Stream-SW 的关键优势是低内存复杂度,同时提供近似误差的理论保证。我们展示,Stream-SW 在比较来自流式样本的高斯分布及其混合分布时,比随机抽样更准确地近似 SW,且内存消耗更低。此外,我们在点云分类、点云梯度流和流式变化点检测实验中进一步凸显了 Stream-SW 优异性能。
引用
@article{arxiv.2505.06835,
title = {Streaming Sliced Optimal Transport},
author = {Khai Nguyen},
journal= {arXiv preprint arXiv:2505.06835},
year = {2026}
}
备注
Accepted to ICML 2026, 22 pages, 8 figures, 7 tables