📖 AI 总结

FlashSinkhorn 2（FS2）是一种面向低维点云上平方欧氏代价的熵最优传输（EOT）求解器，旨在单张 GPU 上以规定的边际残差求解大规模离散 EOT 问题。它采用两阶段耦合策略：粗阶段在单元质心上求解并将势函数提升到每个点，当抽样边际检验拒绝该提升时继续在质心上迭代，从而替代大部分点级更新；块稀疏的精细阶段则消除粗更新无法处理的质心误差，其 Morton 排序块支持筛选与融合张量核执行，并由块质量设定阈值以约束被省略瓦片对每行每列的贡献。在合成基准上，FS2 在全部 32 个问题上达到目标残差，而 GeomLoss 多尺度仅 10 个。在单张 A100 上，FS2 在不到 2.5 小时内求解了两个来自宇宙学 N 体模拟的 1.34×10^8 粒子测度之间的离散 EOT，熵模糊等于平均粒子间距，全粒子边际残差低于 0.01，据作者所知是数小时内以该精度求解的最大离散 EOT 问题。

🔑 关键词速览