Copy Less, Ground More: Overcoming Repetitive Copying in Long-Context Reasoning via Evidence-Aware Reinforcement Learning
Summary
Repetitive copying is identified as a critical failure mode in large language models operating in long-context settings, where models extensively duplicate input text into their reasoning traces instead of productively solving problems. This behavior is pervasive across frontier long-context LLMs and intensifies with increasing context length. Research shows that insufficient grounding is the root cause, as models indiscriminately copy from prompts, leading to incorrect answers when they fail to focus on key evidence. To address this, a method called GEAR (Grounding Evidence-Aware Reward) is proposed. GEAR is a reward shaping technique that enhances the accuracy signal with a grounding reward for overlap with key evidence and a distractor penalty for irrelevant context. An automated pipeline generates evidence-annotated training data for GEAR. Validated across multiple model scales and benchmarks, GEAR consistently improves performance by up to +4.6 average points over standard reinforcement learning, with greater gains observed in longer contexts, while also reducing repetitive copying and thinking length.
Key takeaway
For machine learning engineers developing or fine-tuning long-context LLMs, you should prioritize explicit grounding mechanisms to combat repetitive copying. Implement reward shaping techniques like GEAR, which penalize irrelevant context and reward key evidence overlap, to significantly improve reasoning accuracy. This approach can yield up to +4.6 average point gains, especially in longer contexts, by ensuring your models focus on relevant information.
Key insights
Insufficient grounding causes repetitive copying in long-context LLMs, addressable by evidence-aware reward shaping.
Principles
- Repetitive copying intensifies with context length.
- Insufficient grounding leads to incorrect answers.
- Reward shaping can improve evidence grounding.
Method
GEAR augments accuracy rewards with grounding rewards for key evidence overlap and penalties for irrelevant context overlap, using an automated annotation pipeline.
In practice
- Implement evidence-aware reward shaping.
- Automate evidence annotation for training data.
- Prioritize grounding in long-context LLM development.
Topics
- Large Language Models
- Long-Context Reasoning
- Repetitive Copying
- Evidence Grounding
- Reinforcement Learning
- Reward Shaping
- GEAR Method
Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.