← TW Hub / 173 · long-context-viz-lab
v1.0
LONG CONTEXT LAB

长上下文大模型可视化实验室

探索长上下文大模型核心技术:从 RoPE 旋转位置编码到位置插值,从 Ring Attention 分布式计算到大海捞针评估。通过交互式可视化理解模型如何突破上下文窗口限制,处理百万级 token 序列。

范围 4K → 10M tokens技术 RoPE / YaRN / Ring-Attn 评估 ∞Bench模块 8

上下文窗口与位置编码

不同上下文长度规模对比,及 RoPE 旋转位置编码在外推时的角度变化。

RoPEContext WindowExtrapolation
Token 数
131,072
≈ 97K 词
等效文本
1 本
长篇小说
最大角度
8192π
旋转圈数
上下文长度规模对比
历史基线
当前选择
训练范围
RoPE 旋转向量
RoPE 对位置 m 每个维度施加旋转 m·θᵢ,θᵢ=10000^(-2i/d)。序列超过训练长度时,低频维度角度外推导致位置编码进入未见区域,引发性能下降。

RoPE 与位置插值

对比原始 RoPE、线性插值(PI)、NTK-aware、YaRN 四种位置编码扩展方法的频率缩放。

Position InterpolationNTK-awareYaRNFrequency Scaling
方法
原始 RoPE
无缩放
高频维度
保留
局部注意力
低频维度
保留
长程依赖
复平面旋转向量对比
频率缩放曲线
原始 RoPE:直接外推,所有频率维度角度线性增长,低频维度快速进入未训练区域。

Ring Attention 分布式长上下文

Ring Attention 将长序列分片到多 GPU 组成环形拓扑,KV 块在环上循环传递,实现序列并行注意力。

Ring AttentionSequence ParallelismStriped Attention
当前步
0
/ 4 轮
序列分片
256K
每 GPU
总上下文
1M
token
GPU 环形拓扑与数据流
每个 GPU 持有 Q 和一块 KV。每步:① 本地计算 Q×KV 部分注意力;② 将 KV 块发给环上下一个 GPU。N 步后每个 GPU 的 query 见过全部 KV,注意力完成。

注意力机制与复杂度

标准注意力 O(n²) 与稀疏、线性注意力的复杂度与内存消耗对比。

O(n²)Sparse AttentionLinear AttentionFlashAttention
对数滑块
标准
67M
O(n²) FLOPs
稀疏
2.1M
O(n·√n)
线性
24K
O(n)
注意力矩阵热力图
标准稠密
稀疏带状
线性核
复杂度曲线
标准注意力 n×n 矩阵在 128K 序列下需 64GB 内存。FlashAttention 不改变复杂度但分块 IO 优化降内存至 O(n);稀疏/线性注意力从算法层面降复杂度,但可能损失长程精度。

Needle-in-Haystack 测试(大海捞针)

在长文档中藏入事实("针"),测试模型在不同文档长度和针深度下的检索成功率。

Needle-in-HaystackRetrievalLost in the Middle
成功率
—
点击测试
针深度
中段
50%
文档长度
32K
token
检索成功率热力图
失败 0%
部分 50%
成功 100%
经典发现:Lost in the Middle — 当"针"位于文档中段时检索成功率显著下降。优秀模型应使热力图呈现均匀绿色。针在首尾成功率最高。

KV-Cache 与内存优化

长上下文下 KV-Cache 内存爆炸问题,及量化、驱逐、压缩等优化策略对比。

KV-CacheINT4 QuantizationPagedAttentionEviction
KV 内存
40.0 GB
FP16 / 128K
压缩比
1.0×
vs FP16
精度损失
0%
无损
KV-Cache 内存占用
内存随上下文长度增长
KV-Cache = 2 × layers × heads × seq × dim × bytes。128K FP16 下 Llama-70B 需约 40GB 仅存 KV。PagedAttention 减少碎片;INT4 降至 1/4;滑动窗口只保留最近 W 个 token,代价是丢失早期上下文。

长上下文评估基准

对比主流模型在 RULER、LongBench、∞Bench、Needle、HELM 等基准上的表现。

RULERLongBench∞BenchHELM
模型
Gemini 1.5 Pro
2M 上下文
综合得分
87.3
/ 100
强项
Needle
检索能力
长上下文能力雷达图
RULER 综合测试多类型长上下文任务;∞Bench 平均 100K+;LongBench 中英双语 21 项。标称长度 ≠ 有效长度。

长上下文应用场景

长上下文模型典型应用:整本书理解、代码库分析、多轮长对话、文档问答。点击卡片查看上下文使用模式。

Whole-BookCodebase QALong DialogDoc QA
场景示意图 — 整本书理解
整本书(~10万词 ≈ 130K token)一次性输入,实现跨章节人物关系、伏笔、主题的全局一致性理解。挑战:长程依赖衰减、中段信息丢失。