Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

STILL: Selecting Tokens for Intra-Layer Hybrid Attention to Linearize LLMs

About

Linearizing pretrained large language models (LLMs) primarily relies on intra-layer hybrid attention mechanisms to alleviate the quadratic complexity of standard softmax attention. Existing methods perform token routing based on sliding-window partitions, resulting in position-based selection and fails to capture token-specific global importance. Meanwhile, linear attention further suffers from distribution shift caused by learnable feature maps that distort pretrained feature magnitudes. Motivated by these limitations, we propose STILL, an intra-layer hybrid linearization framework for efficiently linearizing LLMs. STILL introduces a Self-Saliency Score with strong local-global consistency, enabling accurate token selection using sliding-window computation, and retains salient tokens for sparse softmax attention while summarizing the remaining context via linear attention. To preserve pretrained representations, we design a Norm-Preserved Feature Map (NP-Map) that decouples feature direction from magnitude and reinjects pretrained norms. We further adopt a unified training-inference architecture with chunk-wise parallelization and delayed selection to improve hardware efficiency. Experiments show that STILL matches or surpasses the original pretrained model on commonsense and general reasoning tasks, and achieves up to a 86.2% relative improvement over prior linearized attention methods on long-context benchmarks.

Weikang Meng, Liangyu Huo, Yadan Luo, Jiawen Guan, Jingyi Zhang, Yingjian Li, Zheng Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningWinoGrande
Accuracy73.4
1581
Commonsense ReasoningPIQA
Accuracy81.3
400
ReasoningARC Easy
Accuracy83
242
Common Sense ReasoningHellaSwag
Accuracy79
85
Multi-task Language UnderstandingMMLU
Top-1 Accuracy61.3
46
ReasoningARC Challenge
Accuracy56.7
44
Long-context evaluationRULER 4k context length (test)
MK21.5
7
Showing 7 of 7 rows

Other info

Follow for update