Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper investigates the mechanics of hybrid LLM architectures combining full attention with sliding-window or gated linear attention, revealing a "Seesaw Effect" where linear attention hybrids benefit from continual pretraining while sliding-window hybrids excel at length extrapolation. The authors attribute these differing behaviors to the positional inductive biases of each attention mechanism and identify specific limitations in sliding-window hybrids like "Short-Context Learning Trap." They propose "Sliding-Window Linear Attention," achieving significant training-free length extrapolation gains.
Hybrid LLMs exhibit a "Seesaw Effect" in context extension, with linear attention models thriving on continual pretraining and sliding-window models excelling at extrapolation, a distinction rooted in their positional inductive biases.
The architectural design of Large Language Models (LLMs) is shifting from traditional full-attention-only models to hybrid models, which combine different attention modules to improve long-context efficiency and performance in length extrapolation and context extension. To explain why hybrid models work and how to design them better, we propose Mechanics of Long-Context Hybrid Models. As Part 1.1 of this series, we begin with hybrids of full attention and either sliding-window attention (SWA) or gated variants of linear attention (LA), represented by GLA and GDN. We first observe a Seesaw Effect in Context Extension: LA hybrids benefit more from long-context continual pretraining, whereas SWA hybrids perform better under length extrapolation. We attribute this behavior to differences in the positional inductive biases induced by these attention mechanisms. We find that SWA hybrids suffer from a Short-Context Learning Trap, Short-Window Weariness, and Long-Window Laziness, and require extended windows to enhance performance in continual long-context pretraining. For LA hybrids, we summarize the Matthew Effect of Hybrid Position Extrapolation and propose Sliding-Window Linear Attention, achieving 16times training-free length extrapolation while maintaining 100\% accuracy on NIAH-SK1 in 64k context length.