arXiv cs.AI· Jakub Antczak, Joanna Wojciechowicz, Kamil Ksi\k{a}\.zek, Marcin Mazur, {\L}ukasz Struski, Jacek Tabor·· 6 小时前AI 评分29
Fast LapSum：百万级规模下精确可微 Top-k 求解器
Fast LapSum: Exact Differentiable Top-$k$ at Million Scale
AI 导读
Fast LapSum 是一种可扩展的 LapSum 软 Top-k 求解器，在保持精确选择质量 k 的同时支持端到端可微。其认证 GPU 实现处理 10^6、10^7、10^8 个分数中位耗时分别为 0.92、1.39、7.24 ms。在百万像素稀疏对抗样本上较 SOTA 方法实现一个数量级加速，并用于 3D Gaussian splatting 减少高斯数量。
来源：arXiv cs.AI · arxiv.org