Reinforced Efficient Reasoning via Semantically Diverse Exploration

Ziqi Zhao, Zhaochun Ren, Jiahong Zou, Liu Yang, Zhiwei Xu +7 more
1/8/2026
cs.AIcs.CL

Official code: ZiqiZhao1/ROSE-rl (13 stars, Apache-2.0, last commit 2026-01-09).

Abstract

Reinforcement learning with verifiable rewards (RLVR) has proven effective in enhancing the reasoning of large language models (LLMs). Monte Carlo Tree Search (MCTS)-based extensions improve upon vanilla RLVR (e.g., GRPO) by providing tree-based reasoning rollouts that enable fine-grained and segment-level credit assignment. However, existing methods still suffer from limited exploration diversity and inefficient reasoning. To address the above challenges, we propose reinforced efficient reasoning via semantically diverse explorations, i.e., ROSE, for LLMs. To encourage more diverse reasoning exploration, our method incorporates a semantic-entropy-based branching strategy and an $\varepsilon$-exploration mechanism. The former operates on already sampled reasoning rollouts to capture semantic uncertainty and select branching points with high semantic divergence to generate new successive reasoning paths, whereas the latter stochastically initiates reasoning rollouts from the root, preventing the search process from becoming overly local. To improve efficiency, we design a length-aware segment-level advantage estimator that rewards concise and correct reasoning while penalizing unnecessarily long reasoning chains. Extensive experiments on various mathematical reasoning benchmarks with Qwen and Llama models validate the effectiveness and efficiency of ROSE. Codes are available at https://github.com/ZiqiZhao1/ROSE-rl.

View on arXivView PDF

Code Implementations(1)

ZiqiZhao1/ROSE-rlOfficial100%
Apache-2.0
131Roff, ShellJan 8, 20269 months agoApache-2.0

Similar Papers

Cite this paper

@article{zhao2026reinforced,
  title  = {Reinforced Efficient Reasoning via Semantically Diverse Exploration},
  author = {Ziqi Zhao and Zhaochun Ren and Jiahong Zou and Liu Yang and Zhiwei Xu and Xuri Ge and Zhumin Chen and Xinyu Ma and Daiting Shi and Shuaiqiang Wang and Dawei Yin and Xin Xin},
  year   = {2026},
  eprint = {2601.05053},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.05053v1}
}

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/2601.05053.svg)](https://codeofpaper.com/paper/2601.05053)

Discussion