arXiv 人工智能· Felix X. -F. Ye, Yu Chin Fabian Lim, Naigang Wang, Davis Wertheimer·· 5 小时前AI 评分36
FlashSinkhorn 2:块稀疏熵最优传输求解器
FlashSinkhorn 2: Block-Sparse Entropic Optimal Transport
AI 导读
FlashSinkhorn 2(FS2)通过粗粒度质心与块稀疏细粒度两阶段设计,实现单 GPU 求解大规模离散熵最优传输(EOT)并已开源。在合成基准中,FS2 在全部 32 个问题上达到目标残差,而 GeomLoss multiscale 仅完成 10 个。在单张 A100 上,FS2 在 2.5 小时内完成了两个 1.34×10^8 粒子测度间的 EOT 计算,全粒子边际残差低于 0.01。
来源:arXiv 人工智能 · arxiv.org