跳到正文
arXiv · cs.AI· Jakub Antczak, Joanna Wojciechowicz, Kamil Ksi\k{a}\.zek, Marcin Mazur, {\L}ukasz Struski, Jacek Tabor·· 14 小时前

Fast LapSum:百万级规模下精确可微 Top-k 求解器

Fast LapSum: Exact Differentiable Top-$k$ at Million Scale

AI 导读

Fast LapSum 是面向 LapSum 软 Top-k 公式的可扩展求解器,在保持精确选择质量 k 的同时支持端到端可微。它通过基于二项式顺序统计量的概率括界将排序限制在阈值附近的窄带内,并用一次额外线性遍历的认证过程升级为可验证定位,最坏情况回退到全排序。

来源:arXiv · cs.AI · arxiv.org