Tree-Aware Posterior Distillation for Long-Horizon Agentic Reinforcement Learning
Organizations: Tsinghua University · Foundation Model Department, Tencent
Abstract
Reinforcement learning with verifiable rewards has become a scalable post-training paradigm for large language models. However, training agents over long-horizon interactions presents two challenges: using only terminal rewards makes turn-level credit assignment sparse and high-variance, while sequential mini-batch updates on a fixed rollout batch introduce policy staleness as the current policy drifts from the behavior policy. Existing approaches obtain denser credit via learned critics, process reward models, or on-policy distillation, but they rely on a value network, step-level annotations, or a stronger external teacher that increase training complexity. Meanwhile, importance sampling and clipping mitigate policy drift, but they do not specify a finite target that stops sequential updates from overshooting. In this work, we introduce TreeOPSD, a critic-free, tree-aware posterior self-distillation framework for long-horizon agentic reinforcement learning. Treating each interaction turn as a tree node, TreeOPSD branches alternative continuations from shared prefixes and recursively backs up terminal rewards into implicit turn-level credit. It then maps each credit to a closed-form, bounded teacher distribution parameterized by the likelihood ratio relative to the frozen behavior policy. Minimizing the divergence between this teacher and the student policy recovers the policy-gradient direction as the tangent at the behavior policy while bounding updates toward a finite target across sequential mini-batches. Empirically, TreeOPSD improves agentic capabilities across several competitive benchmarks and model scales, especially for long-horizon tasks, e.g., raising the average accuracy of Qwen3-30B-A3B-Thinking-2507 by 4.97 points over GRPO on the Deep Research setting.
Figures & tables
| Method | Multi-Hop QA | Deep Research | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Hotpot | 2Wiki | MuSiQue | Bamboogle | Avg | BC+ | BC | GAIA | xbench | Avg | |
| Qwen3-4B-Thinking-2507 | ||||||||||
| ReAct | 31.61 | 29.63 | 9.68 | 25.60 | 24.13 | 4.70 | 1.97 | 15.53 | 13.00 | 8.80 |
| GRPO | 46.93 | 48.62 | 21.93 | 56.00 | 43.37 | 10.00 | 2.29 | 18.45 | 18.00 | 12.18 |
| GSPO | 46.55 | 49.10 | 21.76 | 55.20 | 43.15 | 9.88 | 2.37 | 18.45 | 18.00 | 12.17 |
| TreePO | 47.25 | 49.90 | 22.05 | 54.40 | 43.40 | 11.08 | 2.76 | 19.42 | 19.00 | 13.07 |
| (a) Rollout allocation | (b) Target radius | ||||
|---|---|---|---|---|---|
| Multi-Hop QA | Deep Research | Multi-Hop QA | Deep Research | ||
| 43.66 | 11.28 | 43.79 | 14.62 | ||
| 42.98 | 10.05 | 44.01 | 15.06 | ||
| 44.13 | 15.45 | 44.13 | 15.45 | ||
| 44.07 | 14.97 | 43.95 | 14.88 | ||
| 43.25 | 11.33 | 43.81 | 14.69 | ||
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Qwen3-1.7B | Llama-3.2-3B-Instruct | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Hotpot | 2Wiki | MuSiQue | Bamboogle | Avg | Hotpot | 2Wiki | MuSiQue | Bamboogle | Avg | |
| ReAct | 23.04 | 26.11 | 6.37 | 19.20 | 18.68 | 12.18 | 7.38 | 4.22 | 20.00 | 10.95 |
| GRPO | 29.10 | 31.48 | 8.61 | 33.60 | 25.70 | 36.80 | 36.35 | 13.78 | 40.00 | 31.73 |
| GSPO | 29.09 | 31.85 | 8.52 | 32.80 | 25.57 | 36.48 | 36.73 | 13.61 | 39.20 | 31.50 |
| TreePO | 29.86 | 32.18 | 8.90 | 33.60 | 26.13 | 37.56 | 37.43 | 14.23 | 40.00 | 32.30 |
| TreeOPSD | 31.30 | 33.81 | 9.60 | 32.80 | 26.88 | 39.59 | 41.15 | 15.97 | 45.60 | 35.58 |
| Method | Multi-Hop QA | Deep Research |
|---|---|---|
| ReAct | ||
| GRPO | ||
| GSPO | ||
| TreePO | ||
| TreeOPSD |
RL for LLM Agents
Reinforcement learning for large language model agents trains models to plan, use tools, and adapt through feedback from interactive tasks.
See all 603 papers