Learning from the Gap Between Pass@K and Pass@1
Organizations: Shanghai Jiao Tong University · William Marsh Rice University · East China Normal University
Abstract
Sampling many responses and keeping one that passes a verifier lets large language models solve problems beyond their single-response ability, but this search must be paid again for every query, while many deployments answer with a single response. Post-training on verified responses can transfer the benefit of search into the model. With a fixed budget, selecting by correctness alone spends slots on problems the model already answers correctly, leaving fewer to correct its failures. To address this imbalance, we propose GapFT, which trains on the gap between Pass@K and Pass@1: problems that the source model fails with one response but solves within K samples. GapFT keeps the objective and training budget fixed and changes only which verified responses enter training; an exact decomposition splits the resulting Pass@1 change into corrected failures and regressions on problems the source model already solved. On LogiQA 2.0 and ReClor with three model families, GapFT is above budget-matched uniform rejection-sampling fine-tuning (RFT) in every setting, with a positive pooled effect, and on Llama-3.1-8B and Mistral-7B it recovers about two thirds to four fifths of the gain of fine-tuning on the entire verified pool with 11-34% of its problems. Further analyses reveal that the gain comes from failures that the first few search samples recover, while failures found only by deeper search displace replay and add no net gain, that filling the same budget with gold-labeled failures search cannot reach lowers accuracy, and that the gain is bounded by how many transferable failures search exposes.
Figures & tables
| Problems | Pass@1 (%) | GapFT Source | |||||||
| Model | Task | All | Source | Replay | Uniform | GapFT | All | (pp) | |
| Llama-3.1-8B | LogiQA 2.0 | 3,190 | 9,429 | 50.33 | 55.84 | 60.33 | 62.59 | 65.89 | [ , ] |
| ReClor | 913 | 3,050 | 55.40 | 61.27 | 69.13 | 70.67 | 73.80 | [ , ] | |
| Qwen2.5-7B | LogiQA 2.0 | 275 | 8,797 | 68.54 | 68.50 | 68.20 | 68.81 | 70.06 | [ , ] |
| ReClor | 63 | 2,977 | 72.20 | 72.00 | 71.53 | 72.47 | 73.20 | [ , ] | |
| Mistral-7B | LogiQA 2.0 | 887 | 6,554 | 47.26 | 49.74 | 49.57 | 51.87 | 53.21 | [ , ] |
| Dataset | Contrast | Pass@1 (pp) | 95% CI | |||
| A. Search depth, replay replaced by recovered failures, main recipe | ||||||
| LogiQA 2.0 | [3.78, 6.95] | |||||
| LogiQA 2.0 | [4.49, 8.80] | |||||
| ReClor | [3.33, 8.13] | |||||
| ReClor | [6.47, 12.53] | |||||
| B. Randomized inclusion of failures first recovered in samples 5–8, main recipe | ||||||
| Condition | Post Pass@1 (%) | vs source (pp) | |||
|---|---|---|---|---|---|
| deployed successes | 63.39 | 41.35 | 3.08 | 11.49 | |
| search-recovered failures | 66.11 | 40.38 | 4.48 | 7.34 | |
| uniform RFT | 64.11 | 33.65 | 5.60 | 9.02 | |
| positive SFT | 66.17 | 41.35 | 5.04 | 7.69 | |
| verified correction | 64.83 | 34.62 | 2.52 | 7.16 | |
| 188 distinct problems, 4 exposures each | 65.78 | 40.06 | 3.92 | 7.60 |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Setting | Value |
|---|---|
| Models | Llama-3.1-8B-Instruct, Qwen2.5-7B-Instruct, and Mistral-7B-Instruct-v0.3 in the main comparison; Llama-3.1-8B-Instruct in the controls |
| Search | for the main comparison; for the search-depth control; temperature , top- , at most 64 new tokens; shortest verified response selected, with response index as tie-breaker |
| Verifier / deployment | Exact option matching, on the whole response for Llama and Qwen on LogiQA 2.0 and on the first explicit answer otherwise / greedy choice decoding |
| Update | RFT with loss on response tokens only; AdamW; bfloat16; one epoch over the recipe repeated eight times |
| LoRA | Rank 8, alpha 16, dropout 0.05; seven attention and MLP projection modules listed above |
| Optimization | Learning rate ; batch size 2; gradient accumulation 8; weight decay 0.01; no scheduler; gradient norm clipped to 1.0; maximum sequence length 2048 |
| Comparison | Task | ||||
|---|---|---|---|---|---|
| Main, Llama | LogiQA 2.0 | 3,190 | 25,520 | 6,049,464 | 1,595 |
| ReClor | 913 | 7,304 | 1,793,896 | 457 | |
| Main, Qwen | LogiQA 2.0 | 275 | 2,200 | 512,408 | 138 |
| ReClor | 63 | 504 | 120,064 | 32 | |
| Main, Mistral | LogiQA 2.0 | 887 | 7,096 | 1,901,832 | 444 |
| ReClor | 252 | 2,016 | 555,896 | 126 |
| Setting | Value |
|---|---|
| Model | Qwen2.5-7B-Instruct |
| Evidence | 188 distinct failures recovered by search or deployed successes, each repeated four times; subject and difficulty strata differ by at most one problem |
| Update | 752 steps using sequence-normalized cross-entropy on response tokens only; bfloat16; AdamW; constant learning rate ; weight decay 0.01; batch size 1; maximum sequence length 2048 |
| Budget matching | 752 training examples and model positions; 270,912 versus 269,980 supervised response tokens, below the prespecified 2% tolerance |
| LoRA | Rank 8, alpha 16, dropout 0.05; the seven attention and MLP projection modules listed above |
| Audit | 600 disjoint MATH problems ( Hendrycks et al., 2021 ) ; deterministic decoding with at most 1,024 new tokens |
| Setting | Value |
|---|---|
| Models | Qwen2.5-Coder-7B-Instruct ( Hui et al., 2024 ) and DeepSeek-Coder-6.7B-Instruct ( Guo et al., 2024 ) |
| Search | independent source episodes; temperature , top- ; at most three attempts and 1,536 new tokens per attempt; shortest successful response from the final turn selected, with sample index and seed as tie-breakers |
| Verifier / deployment | All official public and hidden tests / one episode with at most three internal attempts and public test feedback |
| Update | RFT with loss on response tokens only; AdamW; bfloat16; one epoch; gradient checkpointing |
| LoRA | Rank 16, alpha 32, dropout 0.05; the same seven projection modules as the logic experiments |
| Optimization | Learning rate ; one sequence per microbatch; training examples distributed across exactly 64 update groups; weight decay 0.01; no scheduler; gradient norm clipped to 1.0; maximum sequence length 8192 |
| observed rates (%) | redistribution null (%) | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Dataset | Model | (%) | Net gain (pp) | |||||||
| LogiQA 2.0 | Llama-3.1-8B | 4 | 770 / 306 / 496 | 19.47 | 47.17 | 20.50 | 8.23 | – | – | |
| LogiQA 2.0 | Llama-3.1-8B | 8 | 770 / 416 / 386 | 26.46 | 50.72 | 16.93 | 6.88 | 21.90 | 11.11 | |
| LogiQA 2.0 | Qwen2.5-7B | 4 | 1,050 / 28 / 494 | 1.78 | 51.19 | 8.37 | 2.79 | – | – | |
| LogiQA 2.0 | Qwen2.5-7B | 8 | 1,050 / 37 / 485 | 2.35 | 54.95 | 7.56 | 2.73 | 27.83 | 4.92 | |
| ReClor | Llama-3.1-8B | 4 | 277 / 96 / 127 | 19.20 | 46.88 | 38.58 | 6.38 | – | – | |
| Method | Question | Assumption on verified evidence | Mechanism | Regime | Difference from GapFT |
|---|---|---|---|---|---|
| RFT ( Shao et al., 2024 ) | Improve reasoning from self-generated data | All verified responses are useful | Sample, keep verifier-accepted responses, fine-tune | Offline | Ignores state; budget not matched across evidence sets |
| Distinct paths ( Yuan et al., 2023 ) | What drives RFT gains | Distinct correct paths matter more than samples | Augment with diverse verified paths | Offline | Diversity of responses, not state of the problem |
| DART-Math ( Tong et al., 2024 ) | Correct the easy-query bias of rejection sampling | Difficult queries are under-covered | Give more synthesis trials to queries with a high fail rate under a separate synthesis model | Offline | Selects by query difficulty for the synthesizer, not the deployed state of the trained policy; budget not matched |
| DAPO dynamic sampling ( Yu et al., 2025 ) | Keep gradient signal in group RL | Prompts with uniform group outcome carry no signal | Drop all-correct and all-incorrect prompts online | Online | Filters on current-policy reward variance, which Appendix B.5 shows is not the source state |
| BOND ( Sessa et al., 2025 ) | Amortize Best-of- | The Best-of- distribution is the target | Distill Best-of- via Jeffreys divergence | Online | Targets a distribution, not the failure set; no state distinction |
| SCoRe ( Kumar et al., 2025 ) | Self-correction at test time | Off-policy correction traces mismatch the policy | Two-stage on-policy RL with reward shaping | Online | Repairs mismatch by RL; GapFT selects evidence by state offline |
| LogiQA 2.0 | ReClor | |||
|---|---|---|---|---|
| Training condition | Pass@1 (%) | vs replay (pp) | Pass@1 (%) | vs replay (pp) |
| no post-training (source) | 50.33 | – | 55.40 | – |
| solved replay, | 54.18 | 0 (ref.) | 59.27 | 0 (ref.) |
| search evidence, | 56.91 | [1.51, 3.93] | 62.40 | [0.47, 5.60] |
| search evidence, | 62.33 | [6.78, 9.61] | 67.93 | [6.27, 10.93] |
| failures selected with gold labels | 63.37 | [7.14, 11.09] | 73.20 | [11.33, 16.67] |
| Rates (%) | Term difference vs uniform (pp) | ||||||||
|---|---|---|---|---|---|---|---|---|---|
| Model | Task | (%) | Condition | ||||||
| Llama-3.1-8B | LogiQA 2.0 | 26.6 | replay | 30.7 | 4.4 | 7.3 | |||
| uniform | 46.5 | 16.9 | 12.5 | ref. | ref. | ref. | |||
| GapFT | 54.7 | 31.0 | 18.8 | ||||||
| all verified | 56.1 | 26.7 | 11.0 | ||||||
| Llama-3.1-8B | ReClor | 26.0 | replay | 27.4 | 15.8 | 7.6 | |||
| Source (%) | ||||||||
|---|---|---|---|---|---|---|---|---|
| Model | Task | Greedy | Pass@4 | Pass@8 | Replay | Uniform | GapFT | All |
| Llama-3.1-8B | LogiQA 2.0 | 50.33 | 68.40 | 76.81 | 1 | 2 | 2 | 3 |
| ReClor | 55.40 | 73.32 | 81.40 | 1 | 2 | 3 | 4 | |
| Qwen2.5-7B | LogiQA 2.0 | 68.54 | 70.17 | 70.67 | 1 | 3 | ||
| ReClor | 72.20 | 73.69 | 74.40 | 1 | 2 | |||
| Mistral-7B-v0.3 | LogiQA 2.0 | 47.26 | 52.53 | 54.94 | 1 | 1 | 3 | 4 |
| Answer only | Permissive | |||
| Condition | Pass@1 (%) | vs GapFT | Pass@1 (%) | vs GapFT |
| source | 48.98 | – | 49.36 | – |
| GapFT | 63.40 | ref. | 63.40 | ref. |
| GRPO, all prompts | 42.56 | [ , ] | 51.55 | [ , ] |
| GRPO, flat groups masked | 19.02 | [ , ] | 47.43 | [ , ] |
| Quantity | LogiQA 2.0 | ReClor |
|---|---|---|
| A. Partition and flows for each item | ||
| Unstable share among items the source gets wrong (%) | 35.91 | 44.39 |
| Mean agreement across partitions (%) | 79.90 | 74.89 |
| Jaccard across partitions (%) | 67.67 | 63.87 |
| Mean flow (pp) | 15.07 | 14.87 |
| Mean flow (pp) | 6.39 | 7.53 |