Per-Query Gating of LLM Rerankers for Multi-Hop Retrieval
Organizations: Independent researcher.
Abstract
LLM rerankers add of the order of $0.2-0.3 per 1,000 queries and about a second of tail latency on top of a graph-augmented dense pipeline such as HippoRAG2, and on three multi-hop benchmarks they improve final-hop top-K coverage on seven of nine (dataset, K) cells, by up to +34.8 pp. We ask whether a learned per-query gate can skip the reranker where it will not help, using only features available before the LLM call (27 score and lexical statistics of the two retrieval lists plus a PCA of a small query embedding) with an executable fallback. Every choice, including the fallback and the threshold, is made inside the training fold and applied once to held-out queries, and harmful skips (the rerank would have found the target, the fallback did not) are reported next to the aggregate coverage. Across nine cells on 2WikiMultiHopQA, MuSiQue and HotpotQA the gate skips 51% of calls at an average held-out LastHop@K cost of 1.2 pp; four cells meet a pre-registered 1 pp rule, harmful skips occur in eight (190 harmful against 136 beneficial), and a random gate at the same skip rate loses 2 to 11 pp on the high-lift cells. A second rule sets each cell's threshold from a pre-specified budget on the expected harmful-skip rate over Platt-calibrated harm probabilities (ECE 0.025 after calibration, 0.094 before): at a 1 pp budget the gate skips 42% at -0.8 pp with 66 harmful skips and six cells within 1 pp, but realised harm exceeds the promise in six cells (mean 1.45 vs 0.83 pp), a selection optimism we quantify; a 0.5 pp budget realises about 1 pp. The harm probabilities are calibrated but barely discriminative (AUC 0.16 to 0.70). An earlier version reported 73% "lossless" savings; that figure rested on an oracle fallback and a wrong MuSiQue target, and we document both.
Figures & tables
| Block | Count | Features (in code order) |
|---|---|---|
| HippoRAG2 scores | 9 | ; ; ; , mean, std over the top-10 |
| Vector scores | 7 | ; ; , mean, std over the top-10 |
| Cross-list agreement | 5 | Jaccard of the top-1, top-3, top-5 and top-10 title sets of and ; rank of ’s top-1 title in (10 if absent) |
| Query | 6 | characters ; words ; count of WH-words; comparison-pattern flag; aggregation-pattern flag; unique titles in ’s top-10 |
| Dataset | Records | Titles | |
|---|---|---|---|
| 2Wiki Ho et al. (2020) | 491 | 6,119 | 6,119 |
| MuSiQue Trivedi et al. (2022) | 514 | 11,656 | 9,838 |
| HotpotQA Yang et al. (2018) | 521 | 9,811 | 9,811 |
| Dataset | HippoRAG2 | Always-rerank | Paired ( ) | ||
|---|---|---|---|---|---|
| 2Wiki | 1 | 0.1385 | 0.4868 | pp | 221/50 ( , rerank wins) |
| 2Wiki | 5 | 0.6232 | 0.8004 | pp | 87/0 ( , rerank wins) |
| 2Wiki | 10 | 0.7067 | 0.8024 | pp | 47/0 ( , rerank wins) |
| MuSiQue | 1 | 0.1868 | 0.2957 | pp | 125/69 ( , rerank wins) |
| MuSiQue | 5 | 0.6868 | 0.7276 | pp | 42/21 ( , rerank wins) |
| MuSiQue | 10 | 0.7821 | 0.8074 | pp | 13/0 ( , rerank wins) |
| Dataset | Fallback | Skip % | Always-rerank | Gated | pp | Harmful | Beneficial | pp | |
|---|---|---|---|---|---|---|---|---|---|
| 2Wiki | 1 | D/H | 36.9 | 0.4868 | 0.4623 | 48 | 36 | ||
| 2Wiki | 5 | D/H | 21.4 | 0.8004 | 0.7882 | 6 | 0 | ||
| 2Wiki | 10 | D | 36.5 | 0.8024 | 0.7943 | 5 | 1 | ✓ | |
| MuSiQue | 1 | H | 50.6 | 0.2957 | 0.2782 | 59 | 50 | ||
| MuSiQue | 5 | H | 53.5 | 0.7276 | 0.7160 | 17 | 11 | ||
| MuSiQue | 10 | D/H | 45.1 | 0.8074 | 0.8074 | 0 | 0 | ✓ |
| Learned gate | Score-gap gate | Best fixed | Random (matched) | Always | Always | Oracle skip % | ||
|---|---|---|---|---|---|---|---|---|
| Dataset | skip / | skip / | skip / | skip / | / | |||
| 2Wiki | 1 | 36.9 / | 0.2 / | 0 / | 36.9 / | 55 / 69 | ||
| 2Wiki | 5 | 21.4 / | 0.0 / | 0 / | 21.4 / | 82 / 86 | ||
| 2Wiki | 10 | 36.5 / | 0.2 / | 0 / | 36.5 / | 90 / 94 | ||
| MuSiQue | 1 | 50.6 / | 0.2 / | 0 / | 50.6 / | 76 / 79 | ||
| MuSiQue | 5 | 53.5 / | 2.1 / | 0 / | 53.5 / | 92 / 91 |
| Dataset | Inner | Held-out | Gap | Range | Model | Label | |
|---|---|---|---|---|---|---|---|
| 2Wiki | 1 | 35.9 | 36.9 | 24–41 | lr/lr/lr/rf/lr | h/g/h/g/h | |
| 2Wiki | 5 | 22.1 | 21.4 | 13–29 | gbm/rf/lr/gbm/rf | g/g/g/g/g | |
| 2Wiki | 10 | 37.9 | 36.5 | 23–52 | rf/rf/lr/lr/rf | g/g/h/g/g | |
| MuSiQue | 1 | 52.0 | 50.6 | 41–62 | lr/gbm/lr/lr/lr | g/g/g/g/g | |
| MuSiQue | 5 | 56.7 | 53.5 | 39–75 | gbm/gbm/gbm/lr/gbm | g/g/g/h/g | |
| MuSiQue | 10 | 47.3 | 45.1 | 33–70 | lr/lr/lr/lr/rf | g/g/g/g/g |
| Dataset | Cov. rule skip / | Fallback | Skip % | pp | Harmful | Beneficial | Expected harm | Realised harm | ECE cal. | ECE raw | pp | |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 2Wiki | 1 | 36.9 / | H | 6.9 | 11 | 11 | 1.03 | 2.24 | 0.080 | 0.067 | ✓ | |
| 2Wiki | 5 | 21.4 / | D/H | 19.8 | 4 | 0 | 0.90 | 0.81 | 0.021 | 0.139 | ✓ | |
| 2Wiki | 10 | 36.5 / | D | 39.7 | 8 | 2 | 1.03 | 1.63 | 0.014 | 0.135 | ||
| MuSiQue | 1 | 50.6 / | D/H | 8.4 | 9 | 4 | 0.98 | 1.75 | 0.036 | 0.135 | ✓ | |
| MuSiQue | 5 | 53.5 / | D/H | 14.8 | 8 | 2 | 0.67 | 1.56 | 0.008 | 0.102 | ||
| MuSiQue | 10 | 45.1 / | D/H | 94.9 | 12 | 1 | 1.00 | 2.33 | 0.011 | 0.063 |
| Rule | Skip | Harm | Gain | In budget | Worst | |||
|---|---|---|---|---|---|---|---|---|
| coverage rule (v1) | – | 51.4 | 190 | 136 | 4 | – | -3.26 | |
| loss budget | 0.5 | 33.5 | 42 | 19 | 7 | 3 | -1.75 | |
| loss budget | 1.0 | 42.2 | 66 | 31 | 6 | 3 | -2.14 | |
| loss budget | 2.0 | 50.4 | 107 | 48 | 4 | 3 | -3.26 |
| Encoder pool | Paired vs. 2-way | |||||
|---|---|---|---|---|---|---|
| 2-way ( NV-Embed-v2 PhaseGraph ) | 0.487 | 0.782 | 0.800 | 0.800 | 0.802 | — |
| 3-way ( BGE-large ) | 0.511 | 0.780 | 0.804 | 0.804 | 0.807 | |
| 5-way ( Qwen3-8B , e5-mistral ) | 0.481 | 0.776 | 0.802 | 0.802 | 0.804 |
| Pipeline (always on) | 2Wiki | MuSiQue | HotpotQA |
|---|---|---|---|
| HippoRAG2 top-5 | 0.576 | 0.307 | 0.686 |
| PhaseGraph top-5 | 0.561 | 0.263 | 0.659 |
| Always-rerank top-5 | 0.659 | 0.317 | 0.686 |
| Dataset | Covered | HippoRAG2 | Always-rerank | Gated |
|---|---|---|---|---|
| 2Wiki | 416/491 | 0.549 | 0.643 | 0.636 |
| MuSiQue | 514/514 | 0.307 | 0.317 | 0.313 |
| HotpotQA | 521/521 | 0.686 | 0.686 | 0.685 |
| coverage at | smallest reaching | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Dataset | Pipeline | 1 | 3 | 5 | 7 | 10 | ||||
| 2Wiki | always-rerank | 0.487 | 0.782 | 0.800 | 0.800 | 0.802 | 3 | 5 | – | – |
| 2Wiki | gated ( policy) | 0.448 | 0.729 | 0.770 | 0.780 | 0.794 | 4 | – | – | – |
| MuSiQue | always-rerank | 0.296 | 0.613 | 0.728 | 0.774 | 0.807 | 6 | 9 | – | – |
| MuSiQue | gated ( policy) | 0.257 | 0.597 | 0.718 | 0.765 | 0.807 | 7 | 10 | – | – |
| HotpotQA | always-rerank | 0.555 | 0.902 | 0.948 | 0.969 | 0.975 | 2 | 2 | 3 | 6 |
| Dataset | Always $/1k | Gated $/1k | Cost saved % | Skip % | Tokens | |
|---|---|---|---|---|---|---|
| 2Wiki | 1 | 0.287 | 0.181 | 36.9 | 36.9 | per-call average |
| 2Wiki | 5 | 0.287 | 0.225 | 21.4 | 21.4 | per-call average |
| 2Wiki | 10 | 0.287 | 0.182 | 36.5 | 36.5 | per-call average |
| MuSiQue | 1 | 0.218 | 0.116 | 46.6 | 50.6 | measured |
| MuSiQue | 5 | 0.218 | 0.103 | 52.4 | 53.5 | measured |
| MuSiQue | 10 | 0.218 | 0.125 | 42.7 | 45.1 | measured |
LLM Reranking
Large language model reranking uses LLMs to reorder retrieved documents or candidate outputs by relevance, usefulness, or task-specific criteria.
See all 133 papers