Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention

Authors: Tsendsuren Munkhdalai, Manaal Faruqui, Siddharth Gopal

9 pages, 4 figures, 4 tables (v2 adds: background, implementation details, recent citations and acknowledgments)
AI-generated keywords: Efficient Infinite Context Transformers Infini-attention Large Language Models Bounded Memory Natural Language Processing

AI-generated Key Points

⚠The license of the paper does not allow us to build upon its content and the key points are generated using the paper metadata rather than the full article.

  • Paper introduces a novel method for scaling Transformer-based Large Language Models (LLMs) to handle infinitely long inputs
  • Key innovation is the development of a new attention mechanism called Infini-attention
  • Technique integrates compressive memory, masked local attention, and long-term linear attention mechanisms within a single Transformer block
  • Demonstrated effectiveness through experiments on various language modeling benchmarks
  • Showcased performance on tasks such as 1M sequence length passkey context block retrieval and 500K length book summarization using LLMs with 1B and 8B parameters
  • Introduces minimal bounded memory parameters for fast streaming inference for LLMs
  • Addresses the challenge of efficiently processing infinitely long inputs while maintaining computational constraints
  • Infini-attention opens up new possibilities for handling complex language tasks without compromising performance or resource limitations
Also access our AI generated: Comprehensive summary, Lay summary, Blog-like article; or ask questions about this paper to our AI assistant.

Abstract: This work introduces an efficient method to scale Transformer-based Large Language Models (LLMs) to infinitely long inputs with bounded memory and computation. A key component in our proposed approach is a new attention technique dubbed Infini-attention. The Infini-attention incorporates a compressive memory into the vanilla attention mechanism and builds in both masked local attention and long-term linear attention mechanisms in a single Transformer block. We demonstrate the effectiveness of our approach on long-context language modeling benchmarks, 1M sequence length passkey context block retrieval and 500K length book summarization tasks with 1B and 8B LLMs. Our approach introduces minimal bounded memory parameters and enables fast streaming inference for LLMs.

Submitted to arXiv on 10 Apr. 2024

Ask questions about this paper to our AI assistant

You can also chat with multiple papers at once here.

⚠The license of the paper does not allow us to build upon its content and the AI assistant only knows about the paper metadata rather than the full article.

AI assistant instructions?

Results of the summarizing process for the arXiv paper: 2404.07143v2

⚠This paper's license doesn't allow us to build upon its content and the summarizing process is here made with the paper's metadata rather than the article.

  • Comprehensive Summary
  • Key points
  • Layman's Summary
  • Blog article
The paper "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" by Tsendsuren Munkhdalai, Manaal Faruqui, and Siddharth Gopal introduces a novel method for scaling Transformer-based Large Language Models (LLMs) to handle infinitely long inputs while maintaining bounded memory and computation requirements. The key innovation in their approach is the development of a new attention mechanism called Infini-attention. This technique integrates compressive memory into the traditional attention mechanism and incorporates both masked local attention and long-term linear attention mechanisms within a single Transformer block. Through experiments on various language modeling benchmarks, the authors demonstrate the effectiveness of their proposed approach. They showcase its performance on tasks such as 1M sequence length passkey context block retrieval and 500K length book summarization using LLMs with 1B and 8B parameters. Importantly, their method introduces minimal bounded memory parameters, enabling fast streaming inference for LLMs. Overall, this work presents a significant advancement in natural language processing by addressing the challenge of efficiently processing infinitely long inputs while maintaining computational constraints. The incorporation of Infini-attention opens up new possibilities for handling complex language tasks that require extensive contextual information without compromising on performance or resource limitations.

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.

⚠The license of this specific paper does not allow us to build upon its content and the summarizing tools will be run using the paper metadata rather than the full article. However, it still does a good job, and you can also try our tools on papers with more open licenses.

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.