arXiv:2602.02180cs.LG2026-02被引 2

提出STILL框架,让大模型用更少算力实现长文本推理。

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

  • 基于局部全局一致的自显著性评分选关键token,避免位置偏倚。
  • 在长文本任务中比之前方法提升86.2%性能,且保持原模型推理能力。
  • 适合需要高效长文本处理的场景,如文档理解、对话系统。

线性化预训练大语言模型主要依赖层内混合注意力机制以缓解标准softmax注意力的二次复杂度问题。现有方法基于滑动窗口划分进行令牌路由,导致选择结果受位置影响,并难以捕捉令牌的全局重要性。同时,线性注意力因可学习特征映射引发分布偏移,扭曲了预训练特征的幅度。为此,我们提出STILL,一种高效的层内线性化框架。STILL引入具有强局部-全局一致性的自显著性评分,通过滑动窗口计算实现精准令牌选择,保留显著令牌用于稀疏softmax注意力,其余上下文则通过线性注意力汇总。为保留预训练表征,设计了归一化保持特征映射(NP-Map),解耦特征方向与幅度,并重新注入预训练范数。进一步采用统一的训练-推理架构与分块并行化及延迟选择策略,提升硬件效率。实验表明,STILL在常识与通用推理任务上达到或超过原模型性能,在长上下文基准上相较先前线性化注意力方法最高提升86.2%。

原文摘要 · Abstract (English)

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.

大模型压缩线性注意力长文本处理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。