Copy Less, Ground More: Overcoming Repetitive Copying in Long-Context Reasoning via Evidence-Aware Reinforcement Learning

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

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

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

Topics

Best for: Research Scientist, AI Engineer, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.