S3-R1: Learning to Retrieve and Answer Step-by-Step with Synthetic Data
On the map: LLM agents and tool use → Web, search, and research agents
Abstract (arXiv)
Reinforcement learning (RL) post-training has enabled newer capabilities in models, such as agentic tool-use for search. However, these models struggle primarily due to limitations with sparse outcome-based rewards and a lack of training data that encapsulates questions of differing hardness, which results in models not performing deeper searches with tools to collect evidence for question-answering. To address these limitations, we introduce S^3-R1 (Synthetic data and stabilized Search R1), a framework that couples a data-centric approach with denser learning signals. We first develop a synthetic generation and curation pipeline that programmatically derives diverse, multi-hop questions from existing documents. This pipeline incorporates a retrieval-based verification step to specifically isolate questions of intermediate difficulty. We then pair this expanded training set with a reward structure that evaluates both intermediate search quality and the correctness of the final answer. This setup directly mitigates the credit assignment problems inherent to sparse rewards. Our evaluations show that S^3-R1 outperforms existing baselines by learning more effective search and synthesis strategies, yielding up to a 10% improvement in robust generalization on out-of-domain datasets.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Web, search, and research agents, one of 16 directions in LLM agents and tool use. Builds on 41 of its 53 references that are in the map. First posted on arXiv 2 May 2026.
Builds on
41 of 53 references.
Closest ancestors · 2
Same direction, newest first
- 2025An Empirical Study on Reinforcement Learning for Reasoning-Search Interleaved LLM Agents21 May
- 2024MAIN-RAG: Multi-Agent Filtering Retrieval-Augmented Generation31 Dec
Other references · 39
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2024Gemini 1.5: Unlocking multimodal understanding across millions of tokens of contextcited by 3,175
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
- 2023Retrieval-Augmented Generation for Large Language Models: A Surveycited by 2,185
27 more
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,259
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 951
- 2022Interleaving Retrieval with Chain-of-Thought Reasoning for Knowledge-Intensive Multi-Step Questionscited by 911
- 2022STaR: Bootstrapping Reasoning With Reasoningcited by 895
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
- 2025Group Sequence Policy Optimizationcited by 788
- 2023Siren's Song in the AI Ocean: A Survey on Hallucination in Large Language Modelscited by 695
- 2023Reinforced Self-Training (ReST) for Language Modelingcited by 486
- 2023Scaling Relationship on Learning Mathematical Reasoning with Large Language Modelscited by 417
- 2023Beyond Human Data: Scaling Self-Training for Problem-Solving with Language Modelscited by 325
- 2025R1-Searcher: Incentivizing the Search Capability in LLMs via Reinforcement Learningcited by 322
- 2024Iterative Reasoning Preference Optimizationcited by 244
- 2024Reinforcement learningcited by 236
- 2019Mastering Complex Control in MOBA Games with Deep Reinforcement Learningcited by 150
- 2024Long-Context LLMs Meet RAG: Overcoming Challenges for Long Inputs in RAGcited by 122
- 2020The Importance of Pessimism in Fixed-Dataset Policy Optimizationcited by 99
- 2022Multi-hop Question Answeringcited by 64
- 2025Synthetic Data Generation Using Large Language Models: Advances in Text and Codecited by 61
- 2025Self-Questioning Language Modelscited by 47
- 2025Synthetic Data Generation & Multi-Step RL for Reasoning & Tool Usecited by 45
- 2024PRewrite: Prompt Rewriting with Reinforcement Learningcited by 43
- 2025RLVE: Scaling Up Reinforcement Learning for Language Models with Adaptive Verifiable Environmentscited by 43
- 2024Cofca: A Step-Wise Counterfactual Multi-hop QA benchmarkcited by 26
- 2024Grounding by Trying: LLMs with Reinforcement Learning-Enhanced Retrievalcited by 17
- 2024MaFeRw: Query Rewriting with Multi-Aspect Feedbacks for Retrieval-Augmented Large Language Modelscited by 9
- 2025Contextualizing Search Queries In-Context Learning for Conversational Rewriting with LLMscited by 4
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Synthetic Data RL: Task Definition Is All You Need
- 2025s3: You Don't Need That Much Data to Train a Search Agent via RL
- 2026Search-GRT: Guided Retrieval Training of Search Agents to Optimize for Complex Question Answering
- 2025FrugalRAG: Less is More in RL Finetuning for Multi-Hop Question Answering
- 2025ChatR1: Reinforcement Learning for Conversational Reasoning and Retrieval Augmented Question Answering
- 2025Reinforced Informativeness Optimization for Long-Form Retrieval-Augmented Generation
- 2025StepSearch: Igniting LLMs Search Ability via Step-Wise Proximal Policy Optimization
- 2026Agentic-R: Learning to Retrieve for Agentic Search