Beyond Reward Engineering: A Data Recipe for Long-Context Reinforcement Learning
About
Long-context reasoning is an essential capability for large language models, particularly when they are deployed as autonomous agents that must reason over lengthy trajectories. Reinforcement learning (RL) has recently emerged as a dominant paradigm for improving this ability, yet existing work largely focuses on reward engineering while diverse training data remains scarce. We revisit this problem from a data-centric perspective and show that a simple yet effective data recipe alone, paired with a minimal outcome-based GRPO setup, suffices to substantially improve long-context reasoning. Our recipe targets three complementary task families -- retrieval, multi-evidence synthesis, and reasoning -- for which we construct and curate eight datasets totaling ~14K examples. Experiments on three models (Qwen3-4B/8B/30B-A3B) yield average gains of +7.2/+3.2/+6.4 points across seven long-context benchmarks, surpassing prior RL training sets. We further demonstrate that these gains transfer to agentic tasks, where continuing RL training on an agent-tuned model with our data recipe improves GAIA by +4.8 and BrowseComp by +7.0 points. We will release our datasets to facilitate future research.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Long-context Reasoning | LongBench v2 | -- | 113 | |
| Multi-hop Question Answering | FRAMES | Accuracy85.8 | 46 | |
| Holistic Long-context Reasoning | DocFinQA | Accuracy51.63 | 12 | |
| Multi-hop Question Answering | LB QA v1 | Accuracy77.3 | 12 | |
| Synthetic Long-context Reasoning | LongReason | Accuracy84.89 | 12 | |
| Synthetic Long-context Reasoning | GraphWalks | Accuracy67.75 | 12 | |
| Long-context Reasoning | AA-LCR | -- | 12 | |
| Agentic Interaction | GAIA (test) | Pass@354.9 | 3 | |
| Agentic Interaction | Browsecomp | Pass@319.7 | 3 | |
| Long-context Reasoning | Long-context Reasoning Suite Aggregate | Average LC Score54.64 | 3 |