Modelwire
Subscribe

Randomized YaRN Improves Length Generalization for Long-Context Reasoning

Illustration accompanying: Randomized YaRN Improves Length Generalization for Long-Context Reasoning

Researchers propose Randomized YaRN, a training technique that addresses a persistent weakness in LLM deployment: the inability to reliably handle sequences far longer than those seen during pretraining. By injecting positional encodings sampled from an expanded range during short-context training, the method exposes models to out-of-distribution position signals early, effectively teaching them to extrapolate. Combined with a length curriculum, this approach shows measurable gains on challenging long-context reasoning tasks like BABILong and multi-round coreference resolution. For practitioners scaling LLMs to real-world document processing and multi-turn reasoning workloads, this represents a concrete path to better generalization without requiring massive additional compute.

Modelwire context

Explainer

The key detail the summary soft-pedals is that Randomized YaRN does not require any long-context training data at all during the randomization phase. The model learns to handle out-of-distribution positions purely by seeing scrambled positional signals on ordinary short sequences, which is a meaningfully different claim than 'we trained on more context.'

The related AutoDex coverage from the same day (arXiv cs.LG, June 22) sits in embodied AI and does not connect to this work in any direct way. Randomized YaRN belongs to a separate thread: the ongoing effort to make transformer-based models reliable on inputs that exceed their training distribution. That problem shows up repeatedly in document processing, multi-turn agents, and retrieval-augmented pipelines, and the research community has been iterating on positional encoding fixes for roughly two years. This paper is best read as a refinement within that lineage, not a departure from it.

Watch whether teams building production RAG pipelines report consistent gains on sequences beyond 128k tokens using this method. If the BABILong improvements replicate on held-out enterprise document benchmarks within the next two quarters, the no-long-context-data claim becomes a serious practical argument for adoption.

This analysis is generated by Modelwire’s editorial layer from our archive and the summary above. It is not a substitute for the original reporting. How we write it.

MentionsYaRN · Randomized YaRN · BABILong · Multi-Round Coreference Resolution

MW

Modelwire Editorial

This synthesis and analysis was prepared by the Modelwire editorial team. We use advanced language models to read, ground, and connect the day’s most significant AI developments, providing original strategic context that helps practitioners and leaders stay ahead of the frontier.

Modelwire summarizes, we don’t republish. The full content lives on arxiv.org. If you’re a publisher and want a different summarization policy for your work, see our takedown page.

Randomized YaRN Improves Length Generalization for Long-Context Reasoning · Modelwire