EdgeInfinite: A Memory-Efficient Infinite-Context Transformer for Edge Devices
Authors: Jiyu Chen, Shuang Peng, Daxiong Luo, Fan Yang, Renshou Wu, Fangyuan Li, Xiaoxin Chen
AI-generated Key Points
- EdgeInfinite is a groundbreaking solution designed to address challenges faced by Transformer-based large language models (LLMs) processing long sequences on edge devices.
- Challenges stem from quadratic complexity of attention mechanisms and increasing memory demands from Key-Value (KV) cache.
- Existing KV cache optimizations struggle with irreversible token eviction in tasks requiring long outputs.
- EdgeInfinite introduces innovative solution for handling infinite contexts within Transformer-based LLMs, incorporating compressed memory into models through a .
- Maintains full compatibility with standard Transformer architectures and requires minimal parameter fine-tuning.
- Enables selective activation of for routing tasks involving both long and short contexts.
- Experimental results show EdgeInfinite achieves performance levels comparable to baseline Transformer-based LLMs on benchmarks with long contexts, optimizing memory consumption and reducing processing time for the first token.
Abstract: Transformer-based large language models (LLMs) encounter challenges in processing long sequences on edge devices due to the quadratic complexity of attention mechanisms and growing memory demands from Key-Value (KV) cache. Existing KV cache optimizations struggle with irreversible token eviction in long-output tasks, while alternative sequence modeling architectures prove costly to adopt within established Transformer infrastructure. We present EdgeInfinite, a memory-efficient solution for infinite contexts that integrates compressed memory into Transformer-based LLMs through a trainable memory-gating module. This approach maintains full compatibility with standard Transformer architectures, requiring fine-tuning only a small part of parameters, and enables selective activation of the memory-gating module for long and short context task routing. The experimental result shows that EdgeInfinite achieves comparable performance to baseline Transformer-based LLM on long context benchmarks while optimizing memory consumption and time to first token.
Tools for this paper
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.
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 representationLook for similar papers (in beta version)
We compare the title and abstract with papers indexed in our database. For arXiv papers, we search within the same category; for DOI and other sources, we search across sources and subjects. You can rerun the search as our database grows.
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.