Model Capability Dominates: Inference-Time Optimization Lessons from AIMO 3
Abstract
Majority voting over multiple LLM attempts improves mathematical reasoning, but correlated errors limit the effective sample size. A natural fix is to assign different reasoning strategies to different voters. The approach, Diverse Prompt Mixer, is tested on the AIMO 3 competition: 3 models, 23+ experiments, 50 IMO-level problems, one H100 80 GB, 5-hour limit. Every prompt-level intervention fails. High-temperature sampling already decorrelates errors; weaker strategies reduce accuracy more than they reduce correlation. Across an 8-point capability gap at equal N=8 and every optimization tested, model capability dominates. The gap between the best majority-vote score (42/50) and pass@20 (~45.5) is selection loss, not prompt loss. A verifier-based selector could close it. Prompt engineering cannot.
Figures & tables
| Configuration | Orig | Small | Back | Class | LB Score | |
|---|---|---|---|---|---|---|
| Baseline (21 runs) | 8 | 0 | 0 | 0 | 8 | 39.3 mean |
| Conservative | 5 | 1 | 1 | 1 | 8 | 40 |
| Aggressive | 3 | 2 | 2 | 1 | 8 | 40 |
| Equal | 2 | 2 | 2 | 2 | 8 | 38 |
| Isolated strategies (8 each): | ||||||
| 8 Small Cases | 0 | 8 | 0 | 0 | 8 | 37 |
| Temperature | LB Score | from baseline |
| 38 | ||
| 40 | ||
| 39.3 | — (baseline, 21-run mean) | |
| , min_p 0.03 | 37 |
| Model | Total / Active | Quant | Local /10 | LB Score | ||
|---|---|---|---|---|---|---|
| gpt-oss-120b | 116.8B / 5.1B | FP8 | 8 | 0.69 | 10 | 39.3 (21 runs) |
| gpt-oss-20b | 20B / 3.6B | FP8 | 8 | 0.61 | 9 | 31.0 (3-run: 35,28,30 1 ) |
| gpt-oss-20b | 20B / 3.6B | FP8 | 32 | 0.52 | — | 26 |
| gpt-oss-20b | 20B / 3.6B | FP8 | 3 | 0.54 | — | 28.3 (3-run: 26,28,31) |
| Nemotron-Super-120B -NVFP4 | 120B / 12B | NVFP4 | 3 | 0.47 | — | 23 |
| Qwen3.5-35B-A3B | 35B / 3B | BF16 | 16 | 0.46 | 8 | 23 |
| Experiment | Change | Local Score | |
|---|---|---|---|
| Baseline | — | 8 | 8/10 |
| Long prompts | From GPT-OSS | 16 | 7/10 |
| , short | Double attempts | 16 | 8/10 |
| , long | Double + long prompts | 16 | 7/10 |
| Entropy voting | Add logprobs 5 | 16 | 8/10 |
| top_k=20 | Qwen recommended | 16 | 7/10 |
| ID | Change | LB Score | Category | |
|---|---|---|---|---|
| Baseline | — (21-run mean) | 39.3 | — | — |
| #1 | Conservative mix (5+1+1+1) | 40 | +0.7 | Diversity |
| #2 | Aggressive mix (3+2+2+1) | 40 | +0.7 | Diversity |
| #3 | Equal mix (2+2+2+2) | 38 | 1.3 | Diversity |
| E1 | 8 Small Cases | 37 | 2.3 | Strategy |
| E2 | 8 Work Backwards | 39 | 0.3 | Strategy |
| Competition | Winner | This Work | Key Innovation | |
|---|---|---|---|---|
| AIMO-1 (2024) | 29/50 | — | 48 | Brute-force voting ( XTX Investments, 2024 ; Numina Team, 2024 ) |
| AIMO-2 (2025) | 34/50 | — | 64 | Custom training + GenSelect ( Frieder et al., 2024 ; Moshkov et al., 2025 ) |
| AIMO-3 (2026) | 46 * | 42 | 8 | Off-the-shelf model ( Frieder et al., 2025 ) |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Parameter | Value | Notes |
| Model & Serving | ||
| Model | gpt-oss-120b | 116.8B total, 5.1B active (MoE) |
| Model path | /kaggle/input/gpt-oss-120b/ | Kaggle model hub |
| Quantization | FP8 (weights + KV cache) | kv_cache_dtype=fp8_e4m3 |
| Engine | vLLM 0.11.x | pip install vllm==0.11.2 |
| GPU | NVIDIA H100 80 GB | Kaggle competition kernel |
| # | Configuration | LB | Notes |
|---|---|---|---|
| 0 | Baseline 8 original | 42 | Best ever |
| 1 | Conservative (5+1+1+1) | 40 | |
| 2 | Aggressive (3+2+2+1) | 40 | |
| 3 | Equal (2+2+2+2) | 38 | Worst diversity |
| 4 | Nemotron-14B ( =16) | 3 | Format mismatch |
| 5 | Qwen3.5-35B-A3B | 23 | Cross-model |
| Parameter | Qwen3.5-35B-A3B | Nemotron-Super-120B |
|---|---|---|
| Total params | 35B | 120B |
| Active params | 3B (MoE) | 12B (MoE+Mamba-2) |
| Quantization | BF16 | NVFP4 |
| Engine | SGLang | vLLM 0.17.0rc0 |
| (attempts) | 8 / 16 | 3 / 4 |
| Temperature | 1.0 | 1.0 |