实体
FlashSinkhorn
FlashSinkhorn
PulseAugur coverage of FlashSinkhorn — every cluster mentioning FlashSinkhorn across labs, papers, and developer communities, ranked by signal.
总计 · 30天
2
90 天内 2
发布 · 30天
0
90 天内 0
论文 · 30天
2
90 天内 2
层级分布 · 90 天
主题
情绪 · 30 天
1 天有情绪数据
最近 · 第 1/1 页 · 共 2 条
-
熵最优传输应用于大规模模拟和主动学习
两篇新研究论文探讨了熵最优传输（EOT）和Sinkhorn散度在不同AI场景中的应用。第一篇论文介绍了FlashSinkhorn 2 (FS2)，一个基于GPU的大规模离散EOT问题求解器，能够处理来自模拟的海量数据集。第二篇论文利用Sinkhorn散度进行低预算主动学习，证明了其在为模型训练选择关键数据点方面的有效性，特别是在医学成像等领域。
-
FlashSinkhorn求解器加速GPU上的最优输运
研究人员开发了FlashSinkhorn，一种新的GPU加速熵最优输运（EOT）求解器，可显著减少内存输入输出操作。通过重写稳定对数域Sinkhorn更新以模仿Transformer注意力中的归一化过程，FlashSinkhorn实现了融合内核，将数据流经片上SRAM。这种方法在A100 GPU上执行点云OT等任务时，与现有方法相比，前向传播速度提高了32倍，端到端速度提高了161倍。