Observation-Aware Stabilization of Reinforcement Learning for Retrieval-Augmented Generation

Naixin Zhang, Tong Lin


Abstract
Reinforcement learning (RL) for retrieval-augmented language models enables policies to interleave generated reasoning with retrieved observations. However, in Search-R1-style PPO, training may become unstable after the model has already reached strong validation performance. This paper studies late-stage collapse as a training-stability problem in search-based RL. Four metrics are proposed to quantify validation drawdown (VDD), advantage-tail expansion (ATE), actor update stress (AUS), and observation-to-actor contamination in generalized advantage estimation (GAE), denoted as OAGC. The analysis further shows how observation tokens can influence actor updates through value targets, value-model learning, and GAE. To address this issue, Observation-aware Stabilization (OAS) combines observation residual decay, old-value anchoring, and boundary smoothing. On three benchmark datasets (NQ, TriviaQA, and HotpotQA), OAS improves Search-R1 PPO under multiple retrieval budgets. In the main Top-3 setting, our method improves the exact match (EM) measure from 31.8/49.6/18.7 to 37.6/56.8/23.9 and reduces OAGC from 0.331 to 0.088, with lower search-budget saturation and no increase in the no-search rate.
Anthology ID:
2026.findings-emnlp.1242
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
26287–26307
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1242/
DOI:
Bibkey:
Cite (ACL):
Naixin Zhang and Tong Lin. 2026. Observation-Aware Stabilization of Reinforcement Learning for Retrieval-Augmented Generation. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 26287–26307, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Observation-Aware Stabilization of Reinforcement Learning for Retrieval-Augmented Generation (Zhang & Lin, Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1242.pdf