论文

RangeSWA无需训练,以注意力和位置调整改善扩散LM长度外推

RangeSWA: Training-Free Length Extrapolation for Continuous Diffusion Language Models

模型架构Transformer

摘要

连续扩散语言模型在文本生成任务中显示出广阔的潜力,但当生成长度超过训练长度时,容易出现文本重复、语义漂移和生成质量下降等问题。为了应对这些挑战,我们提出了 RangeSWA,一种免训练的长度外推方法。该方法结合了局部滑动窗口注意力、远距离块摘要和动态内容检索,以保留局部细节并支持远距离上下文访问。此外,通过位置重新映射、条件注意校正和去噪阶段自适应窗口调度,可以减轻由长度外推引起的生成退化。我们对多种连续扩散语言模型进行了实验,结果表明 RangeSWA 可以有效降低长文本生成中的生成困惑度和文本重复率。

2610.42084v1:论文原图
图1:长度外推时,不同注意力机制的上下文访问范围。(a)全注意力可访问全部token,但可能超出训练时的位置范围。(b)滑动窗口注意力仅关注附近token,忽略远处信息。(c)RangeSWA同时使用局部与远处信息,并将远处位置映射回训练范围内。