推荐星级:
  • 1
  • 2
  • 3
  • 4
  • 5

DeepSeek上下文扩展技术从YaRN到百万token.docx

更新时间:2026-10-04 12:02:59 大小:38K 上传用户:潇潇江南查看TA发布的资源 标签:DeepSeek上下文扩展YaRN百万tokenCSA/HCA混合注意力 下载积分:2分 评价赚积分 (如何评价?) 打赏 收藏 评论(0) 举报

资料介绍

DeepSeek的上下文扩展技术:从YaRN到百万token

一、引言

DeepSeek系列模型的上下文窗口经历了从4K到128K(V2/V3)再到1M(V4)的跨越式发展。短上下文训练、长上下文推理的关键在于高效的上下文扩展技术。DeepSeek在V2/V3中使用YaRN实现128K上下文,在V4中通过全新的CSA/HCA混合注意力机制达到百万token级别。本文深入解析DeepSeek的上下文扩展技术体系。

二、短上下文训练的动机

2.1 计算效率

训练超长序列的计算复杂度呈二次方增长。DeepSeek选择在4K序列长度上进行预训练,将有限的计算资源用于处理更多的token总量,而非更长的序列。预训练完成后,再通过上下文扩展技术将窗口延长至128K或1M。

2.2 扩展策略

DeepSeek的上下文扩展采用"先短后长"的策略:先用短序列高效训练模型的基础能力,再通过少量额外训练(约1000步)在长序列上微调,将上下文窗口扩展到目标长度。这种策略在计算效率和上下文能力之间取得了最佳平衡。

三、YaRN技术原理

3.1 RoPE基础

旋转位置编码(RoPE)通过旋转矩阵编码位置信息,具有良好的相对位置编码特性和长度外推能力。但RoPE的外推能力有限,超过训练长度后性能会快速下降。


部分文件列表

文件名 大小
DeepSeek上下文扩展技术从YaRN到百万token.docx 38K

全部评论(0)

暂无评论

上传资源 上传优质资源有赏金

  • 打赏
  • 30日榜单

推荐下载