arXiv · 2410.07063
InAttention: Linear Context Scaling for Transformers
Abstract
VRAM requirements for transformer models scale quadratically with context length due to the self-attention mechanism. In this paper we modify the decoder-only transformer, replacing self-attention with InAttention, which scales linearly with context length during inference by having tokens attend only to initial states. Benchmarking shows that InAttention significantly reduces VRAM usage during inference, enabling handling of long sequences on consumer GPUs. We corroborate that fine-tuning extends context length efficiently, improving performance on long sequences without high training costs. InAttention offers a scalable solution for long-range dependencies in transformer models, paving the way for further optimization.
Explore related subjects
Keep this discovery
Joseph Eisner. 2024-10-09. InAttention: Linear Context Scaling for Transformers. https://arxiv.org/abs/2410.07063
Cite the original work for its findings. Save a collection to share your selection of sources.