Work

From 128K to 4M: Efficient Training of Ultra-Long Context Large Language Models

Year: 2026
Type: conference-paper
Abstract: Long-context capabilities are essential for a wide range of applications, including document and video understanding, in-context learning, and inference-time scaling, all of which require models to process and reason over long sequences of text and multimodal data.In this work, we introduce an efficient training recipe for building ultra-long context LLMs from aligned instruct model, pushing the boundaries of context lengths from 128K to 1M, 2M, and 4M tokens.Our approach leverages continued pretraining strategies to extend the context window, while employing efficient instruction tuning to maintain short context capabilities.Our UltraLong-8B, built on Llama-3.1-Instruct,achieves state-of-the-art performance across a diverse set of long-context benchmarks.Importantly, UltraLong-8B also maintains competitive performance on standard benchmarks, showing balanced improvements for both long and short context tasks.We provide an in-depth analysis of key design choices, highlighting the impacts of scaling strategies and data composition.Our findings establish a robust framework for efficiently scaling context lengths while preserving general model capabilities.We released all model weights for open research.We release all model weights
Language: English
Open Access status: hybrid
OpenAlex ID: W4416120555
FWCI: 8.547
Cites:
Cited by: 1
Related to: 10