YaRN: Efficient Context Window Extension of Large Language Models

Authors: Bowen Peng, Jeffrey Quesnelle, Honglu Fan, Enrico Shippole

License: CC BY 4.0
AI-generated keywords: YaRN LLMs Context Window Extension Generalization Abilities NLP Tasks

AI-generated Key Points

  • YaRN is a compute-efficient approach to extend the context window of transformer-based language models.
  • Large Language Models (LLMs) have strong performance in in-context learning tasks but struggle to generalize beyond their trained sequence length.
  • YaRN allows LLMs to effectively utilize and extrapolate to longer context lengths than their original pre-training would allow.
  • YaRN surpasses state-of-the-art methods for context window extension, using 10x fewer tokens and 2.5x fewer training steps.
  • The authors demonstrate the effectiveness of YaRN by fine-tuning Llama 2 7B/13B models with extended context windows of 64k and 128k using YaRN.
  • Checkpoints for these models are provided on GitHub.
  • YaRN can extrapolate beyond the limited context of a fine-tuning dataset, enhancing its generalization abilities.
  • It enables leveraging longer-context data for superior performance in various NLP tasks such as summarization.
  • Overall, YaRN presents an efficient solution for extending the context window of LLMs.
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Abstract: Rotary Position Embeddings (RoPE) have been shown to effectively encode positional information in transformer-based language models. However, these models fail to generalize past the sequence length they were trained on. We present YaRN (Yet another RoPE extensioN method), a compute-efficient method to extend the context window of such models, requiring 10x less tokens and 2.5x less training steps than previous methods. Using YaRN, we show that LLaMA models can effectively utilize and extrapolate to context lengths much longer than their original pre-training would allow, while also surpassing previous the state-of-the-art at context window extension. In addition, we demonstrate that YaRN exhibits the capability to extrapolate beyond the limited context of a fine-tuning dataset. We publish the checkpoints of Llama 2 7B/13B fine-tuned using YaRN with 64k and 128k context windows at https://github.com/jquesnelle/yarn

Submitted to arXiv on 31 Aug. 2023

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2309.00071v1

  • Comprehensive Summary
  • Key points
  • Layman's Summary
  • Blog article
The paper introduces YaRN (Yet another RoPE extensioN method), a compute-efficient approach to extend the context window of transformer-based language models. These models, known as Large Language Models (LLMs), have shown strong performance in in-context learning (ICL) tasks. However, their ability to generalize beyond the sequence length they were trained on is limited. YaRN addresses this limitation by allowing LLMs to effectively utilize and extrapolate to longer context lengths than their original pre-training would allow. It achieves this while also surpassing the state-of-the-art methods for context window extension, using 10x fewer tokens and 2.5x fewer training steps than previous approaches. The authors demonstrate the effectiveness of YaRN by fine-tuning Llama 2 7B/13B models with extended context windows of 64k and 128k using YaRN and provide the checkpoints for these models on GitHub. In addition, the paper highlights that YaRN exhibits the capability to extrapolate beyond the limited context of a fine-tuning dataset, further enhancing its generalization abilities and enabling it to leverage longer-context data for superior performance in various NLP tasks such as summarization. Overall, YaRN presents an efficient solution for extending the context window of LLMs.

Assess the quality of the AI-generated content by voting

Score: 0

Why do we need votes?

Votes are used to determine whether we need to re-run our summarizing tools. If the count reaches -10, our tools can be restarted.

The previous summary was created more than a year ago and can be re-run (if necessary) by clicking on the Run button below.

Similar papers summarized with our AI tools

Navigate through even more similar papers through a

tree representation

Disclaimer: The AI-based summarization tool and virtual assistant provided on this website may not always provide accurate and complete summaries or responses. We encourage you to carefully review and evaluate the generated content to ensure its quality and relevance to your needs.