Files
myWiki/concepts/length-extrapolation.md

2.6 KiB
Raw Permalink Blame History

title, created, updated, type, tags, sources
title created updated type tags sources
长度外推 (Length Extrapolation) 2026-05-14 2026-05-14 concept
llm
attention
position-encoding
generalization
https://arxiv.org/abs/2309.17453
https://arxiv.org/abs/2104.09864
https://arxiv.org/abs/2108.12409

长度外推 (Length Extrapolation)

定义

长度外推Length Extrapolation是 LLM 研究中的一个核心问题:让在较短文本上训练的模型能够在测试时处理远超训练窗口长度的输入序列。

与上下文窗口扩展Context Window Extension不同长度外推的目标不是扩大一次前向传播能处理的 token 数量,而是让模型在序列长度超出预训练窗口后仍保持性能稳定。

与其他方向的关系

方向 目标 代表工作
长度外推 处理超出训练长度的输入 [[streaming-llm
上下文窗口扩展 扩大单次前向的窗口 RoPE 插值, FlashAttention
长文本利用优化 更好地理解和利用长上下文 LongBench, Lost-in-the-Middle

这三个方向互相正交——一个方向的进展不自动解决其他方向的问题。

主要方法

相对位置编码

  • RoPE (Su et al., 2021):通过旋转 query 和 key 编码相对位置。理论上有外推能力,但实证表明在超出训练窗口后性能下降
  • ALiBi (Press et al., 2022):基于距离偏置注意力分数。外推能力优于 RoPE但仍会在序列极长时崩溃

基于注意力机制的外推

  • streaming-llm (Xiao et al., 2024):保留 attention-sinks + 最近 Token 的滑动窗口 → 首次实现无限长外推(验证至 4M Token
  • Window Attention:仅保留最近 Token → 初始 Token 被逐出后崩溃

位置插值

  • 通过缩放或插值 RoPE 频率来扩展上下文窗口,但仍是有限扩展

关键挑战

  1. SoftMax 归一化约束:注意力分数必须和为 1迫使模型将多余注意力"倾倒"至某些 token
  2. Attention Sink 依赖性:初始 token 承担了注意力汇的角色,一旦被逐出即崩溃
  3. 位置编码泛化:训练中未见过的位置索引在推理时表现不可预测

相关概念