跳到正文
arXiv cs.AI· Felix X. -F. Ye, Yu Chin Fabian Lim, Naigang Wang, Davis Wertheimer·· 1 天前AI 评分22

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

FlashSinkhorn 2: Block-Sparse Entropic Optimal Transport

AI 导读

FlashSinkhorn 2(FS2)是一款面向低维点云平方欧氏代价的熵正则最优传输(EOT)GPU 求解器,通过粗粒度质心求解与块稀疏精修两阶段耦合,在单张 A100 上求解两个 1.34×10^8 粒子测度间的离散 EOT 问题,全粒子边际残差低于 0.01,耗时不到 2.5 小时。

来源:arXiv cs.AI · arxiv.org