Advancing the Pareto Frontier of Test-Time Scaling with Multi-Path Plan Aggregation
Organizations: Georgia Institute of Technology · Cisco Research
Abstract
Test-time scaling seeks to improve reasoning through additional inference-time computation, commonly using solution sampling with majority voting to select the final answer. However, repeated sampling can reproduce correlated errors, limiting the accuracy gains from additional computation. We introduce Multi-Path Plan Aggregation (MPPA), which learns planning and answer selection using separate LoRA adapters on a shared frozen backbone. The planner jointly generates short plans, each guiding a separate solution generated by the frozen executor; the aggregator then scores the resulting solutions and selects the final answer. Joint planning encourages complementary approaches suited to the executor's capabilities, while learned selection uses reasoning evidence to identify correct candidates. Alternating optimization improves planning and aggregation through supervised finetuning and reinforcement learning. Across five mathematics and science benchmarks and three backbones, MPPA improves accuracy--token trade-offs compared with training-free and training-based baselines. With three plans, it achieves higher average accuracy than eight directly sampled solutions with majority voting or the same learned aggregator, while reducing output tokens by 47--54%. Evaluation with task-provided subtasks further demonstrates higher final-task accuracy than direct sampling with majority voting or learned aggregation at either the task or subtask level.
Figures & tables
| Method | Mathematics | Science | Avg | Token | |||
| MATH500 | AIME24 | AIME25 | GPQA | MMLU-Pro | |||
| Qwen2.5-7B-Instruct | |||||||
| Base | 75.6 | 12.6 | 6.7 | 35.1 | 64.6 | 38.9 | |
| Base maj@8 | 81.6 | 14.4 | 13.3 | 34.8 | 69.6 | 42.8 | |
| PS+ maj@8 | 80.6 | 15.6 | 10.0 | 37.9 | 69.6 | 42.7 | |
| L2M maj@8 | 80.4 | 18.9 | 11.1 | 33.8 | 71.8 | 43.2 | |
| Generation | Selection | Mathematics | Science | Avg | |||
|---|---|---|---|---|---|---|---|
| MATH500 | AIME24 | AIME25 | GPQA | MMLU-Pro | |||
| No plan | Majority vote | 77.2 | 13.3 | 7.8 | 35.5 | 67.3 | 40.2 |
| No plan | Aggregator | 83.6 | 17.8 | 13.3 | 54.0 | 78.3 | 49.4 |
| Single plan | Aggregator | 88.0 | 18.9 | 16.7 | 67.2 | 80.1 | 54.2 |
| Independent plans | Aggregator | 87.2 | 20.0 | 13.3 | 69.0 | 81.4 | 54.2 |
| Joint plans | Majority vote | 84.8 | 15.6 | 17.8 | 46.5 | 74.2 | 47.8 |
| Planner | Aggregator | Mathematics | Science | Avg | |||
|---|---|---|---|---|---|---|---|
| MATH500 | AIME24 | AIME25 | GPQA | MMLU-Pro | |||
| SFT | Stage 1 | 80.5 | 14.4 | 10.0 | 39.6 | 70.0 | 42.9 |
| SFT RL | Stage 1 | 83.6 | 16.7 | 12.2 | 43.4 | 73.2 | 45.8 |
| SFT RL | Stage 1 Stage 2 | 86.6 | 18.9 | 15.6 | 46.0 | 76.2 | 48.6 |
| Alternating optimization ( ) | 89.2 | 22.2 | 24.4 | 72.2 | 84.0 | 58.4 | |
Appendix figures & tables28 assets
Supplementary material from the paper’s appendix.
Appendix
| Generation | Mathematics | Science | Avg | |||
|---|---|---|---|---|---|---|
| MATH500 | AIME24 | AIME25 | GPQA | MMLU-Pro | ||
| Base | 83.6 / 84.2 | 17.8 / 18.9 | 13.3 / 14.4 | 54.0 / 54.0 | 78.3 / 78.3 | 49.4 / 50.0 |
| Single plan | 88.0 / 89.6 | 18.9 / 24.4 | 16.7 / 18.9 | 67.2 / 68.2 | 80.1 / 80.2 | 54.2 / 56.3 |
| Independent plans | 87.2 / 89.3 | 20.0 / 23.3 | 13.3 / 15.6 | 69.0 / 70.2 | 81.4 / 83.5 | 54.2 / 56.4 |
| Joint plans | 89.2 / 92.3 | 22.2 / 26.7 | 24.4 / 24.4 | 72.2 / 73.2 | 84.0 / 86.0 | 58.4 / 60.5 |
| Method | MATH500 | AIME24 | AIME25 | GPQA | MMLU-Pro |
|---|---|---|---|---|---|
| Qwen2.5-7B-Instruct | |||||
| Base | 75.6 [71.9, 78.0] | 12.6 [3.6, 20.6] | 6.7 [2.1, 12.2] | 35.1 [30.3, 39.5] | 64.6 [61.1, 67.6] |
| Base maj@8 | 81.6 [77.1, 83.5] | 14.4 [4.9, 28.9] | 13.3 [3.1, 23.1] | 34.8 [31.0, 42.2] | 69.6 [66.0, 73.3] |
| Base agg@8 | 87.9 [85.2, 90.4] | 21.1 [8.8, 35.1] | 20.0 [8.1, 33.5] | 69.4 [64.5, 75.5] | 86.7 [83.8, 89.0] |
| MPPA@1 | 83.6 [80.7, 86.3] | 16.7 [7.4, 26.7] | 14.4 [5.6, 26.3] | 45.1 [39.7, 50.7] | 71.9 [68.5, 75.1] |
| MPPA@3 | 89.2 [86.4, 91.8] | 22.2 [10.0, 36.7] | 24.4 [12.2, 37.8] | 72.2 [65.7, 78.3] | 84.0 [80.8, 87.2] |
Test-Time Scaling
Test-time scaling studies how extra inference-time computation can improve model answers, decisions, and generated outputs.
See all 223 papers