Difficulty-Adaptive Tree-Structured Policy Optimization for Expanding Reasoning Coverage in RLVR

Youngjun Yu, Sanghwan Jang, Hwanjo Yu
9/8/2026
cs.LGcs.AIcs.CL

No official code found. Best implementation: colin31472/DATPO (0 stars, MIT, last commit 2026-08-22).

Abstract

Reinforcement Learning with Verifiable Rewards (RLVR) has been central to the recent success of Large Reasoning Models. However, while RLVR significantly improves single-sample accuracy, it often fails to expand the model's intrinsic reasoning coverage (pass@k) due to limited exploration during training. To address this, we optimize the structural design of train-time rollouts to enhance pass@k. Our analysis identifies three key design principles: (1) difficulty-adaptive rollout can play an important role in expanding pass@k, beyond serving as an efficiency heuristic; (2) tree-based rollout outperforms parallel sampling in discovering correct answers; and (3) sentence-entropy-guided forking overcomes the localization phenomenon of token-level branching to maximize semantic diversity. Building on these insights, we propose DATPO (Difficulty-Adaptive Sentence-entropy-guided Tree-structured Policy Optimization). DATPO integrates difficulty-adaptive tree search with a sibling-diversity advantage term, explicitly promoting semantic diversity to expand reasoning coverage during training. Experiments on mathematical reasoning benchmarks demonstrate that DATPO outperforms baselines especially in pass@k, which directly translates to superior test-time scaling performance.

View on arXivView PDF

Code Implementations(1)

colin31472/DATPOCommunity71%
MIT

[EMNLP 2026 Findings] Difficulty-Adaptive Tree-Structured Policy Optimization for Expanding Reasoning Coverage in RLVR

00Aug 22, 20262 months agoMIT

Similar Papers

Cite this paper

@article{yu2026difficultyadaptive,
  title  = {Difficulty-Adaptive Tree-Structured Policy Optimization for Expanding Reasoning Coverage in RLVR},
  author = {Youngjun Yu and Sanghwan Jang and Hwanjo Yu},
  year   = {2026},
  eprint = {2609.08650},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2609.08650v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2609.08650.svg)](https://codeofpaper.com/paper/2609.08650)

Discussion