中文

关于鲁棒Wasserstein重心:模型与算法

机器学习 2023-12-27 v1 人工智能

摘要

Wasserstein重心问题旨在计算mm个给定概率测度的平均,已在许多不同领域得到广泛研究;然而,现实世界的数据集通常带有噪声且规模巨大,这阻碍了其实际应用。因此,本文专注于提高两类鲁棒Wasserstein重心问题(RWB)的计算效率:固定支撑RWB(fixed-RWB)和自由支撑RWB(free-RWB);实际上,前者是后者的子程序。首先,我们通过模型简化提高效率;将RWB简化为增广Wasserstein重心问题,这同时适用于fixed-RWB和free-RWB。特别地,fixed-RWB可以在O~(mn2ϵ+)\widetilde{O}(\frac{mn^2}{\epsilon_+})时间内使用现成求解器计算,其中ϵ+\epsilon_+是预先指定的加性误差,nn是输入测度位置的大小。然后,对于free-RWB,我们利用一种质量保证的数据压缩技术——coreset,通过减少数据集大小mm来加速计算。结果表明,在coreset上运行算法就足够了,而不必在原始数据集上运行。接下来,通过结合上述模型简化和coreset技术,我们提出了一种通过交替更新权重和位置的free-RWB算法。最后,我们的实验证明了我们技术的效率。

关键词

引用

@article{arxiv.2312.15762,
  title  = {On Robust Wasserstein Barycenter: The Model and Algorithm},
  author = {Xu Wang and Jiawei Huang and Qingyuan Yang and Jinpeng Zhang},
  journal= {arXiv preprint arXiv:2312.15762},
  year   = {2023}
}

备注

Algorithms for accelerating robust Wasserstein barycenter problem