跳到正文
原文
arXiv · Artificial Intelligence· Felix X. -F. Ye, Yu Chin Fabian Lim, Naigang Wang, Davis Wertheimer·· 16 小时前AI 评分31

FlashSinkhorn 2:块稀疏熵最优传输求解器

FlashSinkhorn 2: Block-Sparse Entropic Optimal Transport

AI 导读

FlashSinkhorn 2(FS2)提出粗粒度质心求解与块稀疏细粒度修正两阶段耦合,在单张 GPU 上求解大规模平方欧氏代价离散 EOT 问题。合成基准中 FS2 在全部 32 个问题上达到目标残差,GeomLoss multiscale 为 10 个。单张 A100 在低于 2.5 小时内对两个 1.34×10^8 粒子测度达到全粒子边际残差 0.01 以下,开源实现已发布。

来源:arXiv · Artificial Intelligence · arxiv.org