Top-K Selection on Array of 262,144 elements Top-1024 selection (test)
99.9RecallGeneralized Two-Stage Approximate Top-K
Evaluation results
| Method | Links | ||||
|---|---|---|---|---|---|
| Generalized Two-Stage Approximate Top-KK'=10, buckets=256, num_elements=2,5602025.06 | 99.9 | 19 | 9 | 28 | |
| Generalized Two-Stage Approximate Top-KK'=16, buckets=128, num_elements=2,0482025.06 | 99.9 | 29 | 8 | 37 | |
| jax.lax.approx_max_kK'=1, buckets=131,072, num_elements=131,0722025.06 | 99.8 | 12 | 649 | 661 | |
| Generalized Two-Stage Approximate Top-KK'=6, buckets=512, num_elements=3,0722025.06 | 99.7 | 14 | 11 | 25 | |
| Generalized Two-Stage Approximate Top-KK'=3, buckets=2,048, num_elements=6,1442025.06 | 99.6 | 16 | 32 | 48 | |
| Generalized Two-Stage Approximate Top-KK'=4, buckets=1,024, num_elements=4,0962025.06 | 99.6 | 13 | 14 | 27 | |
| jax.lax.approx_max_kK'=1, buckets=65,536, num_elements=65,5362025.06 | 99.2 | 13 | 292 | 305 | |
| Generalized Two-Stage Approximate Top-KK'=1, buckets=65,536, num_elements=65,5362025.06 | 99.2 | 13 | 313 | 326 | |
| Generalized Two-Stage Approximate Top-KK'=8, buckets=512, num_elements=4,0962025.06 | 99.2 | 16 | 14 | 30 | |
| Generalized Two-Stage Approximate Top-KK'=2, buckets=4,096, num_elements=8,1922025.06 | 99.1 | 15 | 30 | 45 | |
| Generalized Two-Stage Approximate Top-KK'=5, buckets=512, num_elements=2,5602025.06 | 98.9 | 13 | 9 | 22 | |
| jax.lax.approx_max_kK'=1, buckets=32,768, num_elements=32,7682025.06 | 98.7 | 13 | 131 | 144 | |
| Generalized Two-Stage Approximate Top-KK'=1, buckets=32,768, num_elements=32,7682025.06 | 98.7 | 14 | 141 | 155 | |
| Generalized Two-Stage Approximate Top-KK'=12, buckets=128, num_elements=1,5362025.06 | 98.4 | 23 | 8 | 31 | |
| Generalized Two-Stage Approximate Top-KK'=3, buckets=1,024, num_elements=3,0722025.06 | 97.7 | 12 | 11 | 23 | |
| Generalized Two-Stage Approximate Top-KK'=1, buckets=16,384, num_elements=16,3842025.06 | 97.2 | 13 | 64 | 77 | |
| Generalized Two-Stage Approximate Top-KK'=2, buckets=2,048, num_elements=4,0962025.06 | 96.8 | 13 | 14 | 27 | |
| Generalized Two-Stage Approximate Top-KK'=4, buckets=512, num_elements=2,0482025.06 | 96.3 | 12 | 8 | 20 | |
| Generalized Two-Stage Approximate Top-KK'=6, buckets=256, num_elements=1,5362025.06 | 95.1 | 14 | 8 | 22 | |
| Generalized Two-Stage Approximate Top-KK'=1, buckets=8,192, num_elements=8,1922025.06 | 94.2 | 13 | 30 | 42 |