Adaptive rollout trees broaden language models’ mathematical reasoning coverage

DATPO organizes training attempts into difficulty-aware trees and rewards diverse branches, improving the chance that at least one of several answers is correct.

PaperIndustrycs.LGarXiv:2609.08650v1
Youngjun Yu · Sanghwan Jang · Hwanjo Yu

Pohang University of Science and Technology (POSTECH)

Research Digest··2 min read
Yu, Jang and Yu study how the structure of reinforcement-learning rollouts affects exploration in mathematical reasoning models. They report that their difficulty-adaptive, sentence-level branching method improves pass@k—the probability of finding a correct answer within k attempts—and consequently benefits test-time scaling.

What they did

The authors analyze three choices in reinforcement learning with verifiable rewards: allocating rollouts according to problem difficulty, organizing samples as a branching tree rather than independent parallel attempts, and selecting branch points using sentence-level entropy, a measure of model uncertainty.

They combine these elements in Difficulty-Adaptive Sentence-entropy-guided Tree-structured Policy Optimization (DATPO). The method also adds a sibling-diversity advantage term that rewards semantically distinct reasoning paths, then evaluates the resulting policy on mathematical reasoning benchmarks.

Key findings

  • Difficulty-adaptive rollout allocation expanded pass@k rather than merely reducing computation on easier problems.
  • Tree-structured rollouts found correct answers more effectively than parallel sampling in the authors’ comparisons.
  • Sentence-entropy-guided branching produced greater semantic diversity than token-level branching, which tended to concentrate branches around localized variations.
  • DATPO outperformed the evaluated baselines particularly on pass@k, yielding stronger performance when additional solutions were sampled at test time.

Why it matters

RLVR often raises single-attempt accuracy without broadening the set of reasoning paths a model can successfully pursue. The authors’ results suggest that rollout topology and branch selection are consequential parts of the learning algorithm: explicitly training for diverse successful trajectories can make inference-time sampling more productive.

Caveats

The supplied abstract does not identify the models, benchmarks, compute budgets, baselines or numerical effect sizes, so the magnitude and generality of the gains cannot be assessed here. The evidence is limited to mathematical reasoning, and it remains unclear whether sentence-level tree search offers similar benefits in domains without straightforward verifiable rewards.

§
newspaper

Research Digest

Articles published under the Zotpaper byline are synthesized from multiple source publications by our AI editor and reviewed by our editorial process. Each story combines reporting from credible outlets to give readers a balanced, comprehensive view.