实现线性化的关键:基于分析驱动的Transformer线性化方法 / The Key to Going Linear: Analysis-Driven Transformer Linearization
1️⃣ 一句话总结
本文针对Transformer模型在处理超长文本时自注意力机制计算成本过高的问题,通过严格分析软注意力机制的核心原理,提出了一系列结构改进(如插入“沉没标记”、短卷积和固定预算缓存路由),成功在保持模型性能的同时实现了高效的线性化推理,并在大模型上验证了其有效性。
The quadratic cost of causal self-attention severely bottlenecks long-context transformer inference. While numerous post hoc linearization pipelines exist, it is difficult to identify which components preserve model quality. This work isolates the effect of state update design in a strict frozen-backbone regime. We show that softmax relies on key-dependent, rank-1 orthogonal projections, elucidating why delta-style networks outperform purely gated accumulation. We identify a potential source of approximation errors and introduce structural interventions, specifically sink tokens, short convolutions, and fixed-budget cache routing, which reduces the remaining gap. We scale this linearization approach across LLaMA and Qwen models up to 32B parameters, outperforming prior post hoc baselines on MMLU and matching the long-context retrieval of complex adaptive-caching frameworks.
实现线性化的关键:基于分析驱动的Transformer线性化方法 / The Key to Going Linear: Analysis-Driven Transformer Linearization
本文针对Transformer模型在处理超长文本时自注意力机制计算成本过高的问题,通过严格分析软注意力机制的核心原理,提出了一系列结构改进(如插入“沉没标记”、短卷积和固定预算缓存路由),成功在保持模型性能的同时实现了高效的线性化推理,并在大模型上验证了其有效性。
源自 arXiv: 2607.07706