複数の推論経路を計画して数学問題の正答率を上げる
Planned Test-Time Scaling with Coordinated Reasoning Paths
この論文をやさしく読む
ひとことで言うと
数学問題を複数回解く際、各試行が似た解き方にならないよう先に経路を計画する方法です。
何に役立つ?
複数の解答候補を生成する推論システムで、追加の計算量から得られる正答率の改善を高める用途が考えられる。
この研究の面白いところ
計画器が全枝の解法の概略を協調して作り、実行器は固定したまま経路の多様性を高める。
どこまで分かった?
数値結果はQwen3の1.7Bと4B、五つの数学ベンチマークに基づく。理論上の改善は単純化した設定での証明である。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
難しい推論課題で性能を高めるため、推論時に複数の枝を並列に試す方法が広く使われている。主流の反復サンプリングでは一つの方策から枝を独立に生成するため、似た試行が重複し、計算量を増やしても利得が限られる可能性がある。著者らは、この制約に対し、独立サンプリングを協調した共同方策に置き換えるPlanned Test-Time Scaling(PTTS)を提案する。計画器が枝ごとに解法の概略を作り、異なる推論経路へ導き、実行器が各概略に基づいて完全な解答を生成する。形式的には、PTTSが反復サンプリングを厳密に一般化することを示し、単純化した設定では、相補的な推論様式をより広く覆い、pass@kの伸びを改善することを証明する。強い推論モデルを実行器として固定し、反復サンプリングをPTTSの推論に置き換えて実装した。二つの変種を開発した。PTTS-ZSは一回の自己回帰生成で全枝の概略をまとめて作るようモデルに指示し、PTTS-RLは効率的な学習とより明確な報酬信号のため、途中で打ち切る実行ロールアウトを用い、pass@k報酬に対して計画器を直接最適化する。Qwen3の1.7Bと4Bを使った五つの数学的推論ベンチマークでは、PTTS-ZSが反復サンプリングに比べpass@64を最大6.7ポイント改善し、PTTS-RLでは改善幅が最大13.4ポイントになった。追加の分析は、異なる推論経路をより広く覆うことが改善に寄与することを示している。PTTSは、ゼロショット型と学習型の実装を通じて推論経路を協調させ、推論時の計算量増加をより有効に使う一般的な枠組みを与える。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-23(UTC)
- 最新改訂
- 2026-09-23 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-23 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Test-time scaling with parallel branches is widely adopted to improve performance on challenging reasoning tasks. The predominant approach, repeated sampling, draws branches independently from a single policy, which can produce redundant attempts and thereby limit the gains from additional inference compute. To address this limitation, we propose Planned Test-Time Scaling (PTTS), which replaces independent sampling with a coordinated joint policy: a planner generates a solution outline for each branch, steering the branches toward distinct reasoning paths, and an executor produces a full solution conditioned on each outline. Formally, we show that PTTS strictly generalizes repeated sampling and, in a stylized setting, provably promotes coverage of complementary reasoning modes and yields better pass@k scaling. We instantiate PTTS on top of strong reasoning models, keeping them fixed as executors while replacing repeated sampling with PTTS inference to further enhance test-time scaling. Concretely, we develop two variants: PTTS-ZS prompts a model to jointly generate outlines for all branches in a single autoregressive pass, while PTTS-RL directly optimizes the planner against the pass@k reward using truncated execution rollouts for efficient training and a sharper reward signal. Across five mathematical reasoning benchmarks with Qwen3-1.7B and 4B, PTTS-ZS improves pass@64 over repeated sampling by up to 6.7 points, while PTTS-RL further increases the gain to up to 13.4 points. Further analysis indicates that broader coverage of distinct reasoning paths contributes to these gains. Overall, PTTS provides a general framework for improving test-time scaling by coordinating reasoning branches, with zero-shot and trainable instantiations that yield substantial performance gains.
arXiv ID: 2609.27374 / 要約の誤りについて