Top-K Selection on Array of 262,144 elements Top-1024 selection (test) benchmark leaderboard · SOTA2 Research

Top-K Selection on Array of 262,144 elements Top-1024 selection (test)

99.9RecallGeneralized Two-Stage Approximate Top-K

Evaluation Results

Method
Generalized Two-Stage Approximate Top-K2025.06	99.9	19	9	28
Generalized Two-Stage Approximate Top-K2025.06	99.9	29	8	37
jax.lax.approx_max_k2025.06	99.8	12	649	661
Generalized Two-Stage Approximate Top-K2025.06	99.7	14	11	25
Generalized Two-Stage Approximate Top-K2025.06	99.6	16	32	48
Generalized Two-Stage Approximate Top-K2025.06	99.6	13	14	27
jax.lax.approx_max_k2025.06	99.2	13	292	305
Generalized Two-Stage Approximate Top-K2025.06	99.2	13	313	326
Generalized Two-Stage Approximate Top-K2025.06	99.2	16	14	30
Generalized Two-Stage Approximate Top-K2025.06	99.1	15	30	45
Generalized Two-Stage Approximate Top-K2025.06	98.9	13	9	22
jax.lax.approx_max_k2025.06	98.7	13	131	144
Generalized Two-Stage Approximate Top-K2025.06	98.7	14	141	155
Generalized Two-Stage Approximate Top-K2025.06	98.4	23	8	31
Generalized Two-Stage Approximate Top-K2025.06	97.7	12	11	23
Generalized Two-Stage Approximate Top-K2025.06	97.2	13	64	77
Generalized Two-Stage Approximate Top-K2025.06	96.8	13	14	27
Generalized Two-Stage Approximate Top-K2025.06	96.3	12	8	20
Generalized Two-Stage Approximate Top-K2025.06	95.1	14	8	22
Generalized Two-Stage Approximate Top-K2025.06	94.2	13	30	42