English

SOUP: Token-level Single-sample Mix-policy Reinforcement Learning for Large Language Models

Computation and Language 2026-01-30 v1

Abstract

On-policy reinforcement learning (RL) methods widely used for language model post-training, like Group Relative Policy Optimization (GRPO), often suffer from limited exploration and early saturation due to low sampling diversity. While off-policy data can help, current approaches that mix entire trajectories cause significant policy mismatch and instability. In this work, we propose the S\textbf{S}ingle-sample Mix-pO\textbf{O}licy U\textbf{U}nified P\textbf{P}aradigm (SOUP), a framework that unifies off- and on-policy learning within individual samples at the token level. It confines off-policy influence to the prefix of a generated sequence sampled from historical policies, while the continuation is generated on-policy. Through token-level importance ratios, SOUP effectively leverages off-policy information while preserving training stability. Extensive experiments demonstrate that SOUP consistently outperforms standard on-policy training and existing off-policy extensions. Our further analysis clarifies how our fine-grained, single-sample mix-policy training can improve both exploration and final performance in LLM RL.

Keywords

Cite

@article{arxiv.2601.21476,
  title  = {SOUP: Token-level Single-sample Mix-policy Reinforcement Learning for Large Language Models},
  author = {Lei Yang and Wei Bi and Chenxi Sun and Renren Jin and Deyi Xiong},
  journal= {arXiv preprint arXiv:2601.21476},
  year   = {2026}
}
R2 v1 2026-07-01T09:25:22.580Z