Teaching LLMs to Find the Evidence, Not Just the Answer
Open-source language models that score well on standard benchmarks perform near random chance when asked to identify which piece of context actually supports their answer.
- What they did — The researchers added an auxiliary RL objective that trains models to select which of two nearly identical contexts supports a given answer, layered on top of standard GRPO post-training.
- Key result — ContextRL improved over standard GRPO by +2.2% on average across 5 long-horizon benchmarks and +1.8% across 12 visual QA benchmarks, while data-augmentation baselines using the same contrastive data provided little to no improvement.
- Why it matters — The diagnostic probe and training method suggest that strong benchmark scores can mask a failure to ground predictions in available evidence — a gap that a simple, modality-agnostic objective can partially close.
Open-source language models that score well on standard benchmarks perform near random chance when asked to identify which piece of context actually supports their answer.
The Problem
LLMs increasingly operate over rich, complex inputs: long agent trajectories with tool calls, dense images with subtle visual details. Getting the right answer requires more than reasoning ability — it requires grounding decisions in specific, often sparse evidence buried in the context [§1]. A model editing code might ignore a variable definition earlier in the file. A vision-language model might misread a value on a chart despite the information being plainly visible [Figure 1].
The authors call this failure mode "context unawareness": the relevant information is available, but the model's prediction isn't grounded in it [§1]. To measure it, they built a diagnostic probe: 200 contrastive pairs from agentic trajectories and 200 from visual QA images. Each example gives the model a question, an answer, and two nearly identical contexts — only one of which supports that answer. The model must pick the right context [§1].
The results are stark. GPT-5.4 and Claude Opus 4.7 score 98–99% on the agentic version and 83–84% on the multimodal version. Qwen3-VL 8B and Qwen3.5 9B — competitive open-source models — score 52.5–58%, barely above the 50% random baseline [Figure 2]. Strong benchmark performance, it turns out, can coexist with near-total inability to identify supporting evidence.
What They Did
ContextRL adds a single auxiliary training objective to standard GRPO (Group Relative Policy Optimization, a common RL post-training method). Instead of only rewarding correct final answers, it also rewards the model for correctly identifying which context supports a given answer [§2].
Concretely, each training example contains a question, an answer, and two contexts that are superficially almost identical but support different answers. The model sees all three elements and must select the supporting context. Think of it like a reading comprehension test where both passages look nearly the same, and you have to figure out which one actually justifies the conclusion you've been given. This is implemented as a logit-level contrastive loss — the model's confidence score on the correct context option must exceed its score on the incorrect one [§2.3].
Building these contrastive pairs required domain-specific pipelines. For coding agents, the team mined 66k trajectories from SWE-smith, applying cascading filters: same repository, same commit, same file, same function, but different issues. Patch contents were masked to prevent shortcut cues. Only 1k pairs (1.5% of candidates) survived automated verification by GPT-5.4 plus manual review of uncertain cases [§2.1].
For multimodal tasks, they used two strategies across five visual domains (charts, geometry, math, science diagrams, natural images). Natural images were edited using a generative model to change the answer-relevant region while preserving global scene structure, with a ~65% rejection rate during quality verification. Structured images like charts and diagrams were paired via embedding similarity search, requiring cosine similarity ≥ 0.85 and different answers. This produced 7k contrastive image pairs total [§2.2].
The context-awareness loss is trained jointly with the standard GRPO objective on original task data, requiring no architectural changes [§2.3].
The Results
On long-horizon tasks, ContextRL improved over standard GRPO by +3.2% on average across five agentic and long-context benchmarks using Klear-AgentForge-8B, and +1.5% using Qwen3-8B [§1]. On multimodal tasks, it improved by +2.0% on Qwen2.5-VL-7B and +1.6% on Qwen3-VL-8B across 12 visual QA benchmarks [§1].
The most telling result is what happened with the baselines designed to isolate the source of improvement. The researchers took the exact same contrastive data and fed it to models through conventional training: supervised fine-tuning (SFT) or standard outcome-based RL where the contrastive contexts were repurposed as regular query-context-answer examples. These baselines collapsed. Supervised augmentation drove the long-horizon agent's resolve rate to as low as 0%. Outcome-based augmentation added essentially nothing [§1, §2]. Only the context-selection objective — the specific way the signal was integrated — converted the data into gains.
The improvements, while consistent, are modest in absolute terms. The contrastive data is limited: 1k pairs for coding, 7k for multimodal [§1]. The agentic experiments use only two base models (both ~8B parameters), and the multimodal experiments use two more [§1]. Whether these gains hold at larger model scales, with more diverse contrastive data, or on tasks requiring different types of grounding remains untested — scaling studies would likely take another research cycle.
Why It Matters
This work makes two contributions worth tracking separately. First, the diagnostic probe itself is useful: a simple, cheap test that exposes whether a model can identify supporting evidence, independent of whether it gets the final answer right. The 40-point gap between proprietary and open-source models on this probe [Figure 2] suggests that context grounding is a meaningful axis of model quality that standard benchmarks miss.
Second, the training method demonstrates that you can improve context grounding without changing model architecture or collecting large-scale human annotations — just by adding the right auxiliary objective to existing RL pipelines [§1]. The fact that the same data fails when used through conventional training channels makes the case that the objective design, not the data volume, is doing the work [§1].
This is a lab proof-of-concept. The gains are consistent but small, the evaluation covers a limited set of model sizes, and the contrastive data construction pipelines rely on GPT-5.4 for verification. For teams already running GRPO post-training on agentic or multimodal models, the method is lightweight enough to test. For everyone else, the diagnostic probe is the immediately actionable piece: a way to check whether your model actually reads what you give it.