Conference Paper
Large Language Models

How to train long-context language models (effectively)

Tianyu Gao, Alexander Wettig, Howard Yen, Danqi Chen
January 1, 2025Annual Meeting of the Association for Computational Linguistics147 citations

147

Citations

22

Influential Citations

Annual Meeting of the Association for Computational Linguistics

Venue

2025

Year

Abstract

… In this section, we study how to best enable longcontext language models to follow instructions. We focus on supervised fine-tuning on instruction datasets (Ouyang et al.…

Analysis

Why This Paper Matters

Long-context language models are increasingly important for tasks like processing entire documents, codebases, or lengthy conversations. However, enabling these models to effectively follow instructions over long contexts remains a challenge. This paper addresses a critical gap by systematically studying how supervised fine-tuning on instruction datasets can be optimized for long-context scenarios. As models grow to handle millions of tokens, understanding the right training methodology becomes essential for practical deployment.

Technical Contributions

  • Focuses on supervised fine-tuning (SFT) as the primary method for instruction following in long-context models
  • Investigates the impact of training data composition and length on instruction-following performance
  • Provides empirical comparisons of different fine-tuning strategies for extended context windows
  • Likely explores trade-offs between context length and instruction adherence

Results

While the abstract does not provide specific numerical results, the paper's findings are expected to include comparisons of fine-tuning approaches on benchmarks for long-context instruction following. The 147 citations suggest the work has already influenced subsequent research in this area.

Significance

This research has direct implications for AI practitioners building long-context applications, such as legal document analysis, code review, or conversational agents with long-term memory. By clarifying effective training strategies, it helps reduce trial-and-error in model development. The work also contributes to the broader understanding of how to balance context length with instruction-following fidelity, a key challenge for next-generation language models.