Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

BranPO: Scalable Contrastive Branch Sampling for Long-Horizon Agentic Reinforcement Learning

About

Agentic reinforcement learning enables large language models to perform multi-turn planning and tool use, but long-horizon training remains challenging under sparse trajectory-level rewards, where a single outcome is uniformly assigned to all decisions. Prior methods introduce finer-grained supervision via tree-based exploration or process-level evaluation, but often incur high cost or produce noisy credit signals. In agentic trajectories, early mistakes may still be corrected by later actions, while seemingly promising intermediate states can fail due to poor subsequent decisions. We call this property non-monotonic correctness, which makes outcome rewards or state values insufficient for guiding what actions should be taken from each state. To address this, we propose Branching Relative Policy Optimization (\textbf{BranPO}), a value-free method that constructs localized contrastive supervision without dense rewards. BranPO truncates trajectories at intermediate prefixes and resamples continuations to form contrastive branches that share the same prefix but diverge in final outcomes, thereby isolating decisions that drive success or failure. We further introduce difficulty-aware branch sampling and Redundant Step Masking to improve sampling efficiency and suppress redundant updates. Experiments show that BranPO consistently outperforms diverse baseline categories across multiple multi-hop QA benchmarks without additional training cost, and generalizes to broader long-horizon agentic tasks with consistent improvements. Our code is available at https://github.com/YubaoZhao/BranPO.

Yubao Zhao, Weiquan Huang, Sudong Wang, Ruochen Zhao, Chen Chen, Yao Shu, Chengwei Qin• 2026

Related benchmarks

TaskDatasetResultRank
Multi-hop Question AnsweringMuSiQue--
209
Single-hop Question AnsweringPopQA--
186
Multi-hop Question Answering2WikiMQA
F1 Score75.6
161
Single-hop Question AnsweringTriviaQA--
133
Multi-hop Question AnsweringHotpotQA
F1 Score64.5
31
Multi-hop Question AnsweringBamboogle
F157.8
25
Web Search AgentGAIA Avg@4
F1 (Lv.1)57.9
4
Showing 7 of 7 rows

Other info

Follow for update