Radik

Prof-K 算法提供概率性 Top-k 选择，速度显著提升

研究人员开发了 Prof-K，一种用于高效 Top-k 选择的新型算法，提供概率正确性保证。这种单遍方法使用一小部分样本来估计阈值，然后将输入元素处理一次到一个紧凑的缓冲区中，以用户指定的错误概率识别真正的 Top-k 值。Prof-K 在大型场景中，尤其是在 k 值较小到中等的情况下，与现有的 PyTorch 和 RadiK 实现相比，速度显著提升，并在各种输入分布中保持鲁棒性。