Skip to content

필사 모드: 把预训练检查点变成长上下文混合模型 — HyLo 的升级改造配方

한국어
0%
정확도 0%
💡 왼쪽 원문을 읽으면서 오른쪽에 따라 써보세요. Tab 키로 힌트를 받을 수 있습니다.

引言 — 不丢弃检查点,而是改造再用

长上下文是当下 LLM 的最前线,而在这条前线上,真正的敌人不是算力,而是 内存。注意力的 KV 缓存会随序列变长而线性膨胀,到某一刻就会比权重本身还大 — 上一篇量化综述 正是以"下一个瓶颈是 KV 缓存"作结的那个地方。削减这张内存账单大致有两条路:一条是 把每个数值压进更少的比特 — 量化;另一条是 改变结构,让缓存从一开始就不那么膨胀 — 混合架构。本文谈的是第二条轴。

2026 年 4 月的预印本 HyLo(《Long-Context Aware Upcycling: A New Frontier for Hybrid LLM Scaling》,Ashrafi Fashi 等)在此之上又加了一个反转:不是 从头重新训练 一个混合模型,而是只用廉价的后处理训练,把已有的 Transformer 检查点 "升级改造(upcycling)" 成混合模型。先说清楚一点:这是一篇未经同行评审的预印本,下面所有的对比数字都是 作者报告的数值

为什么要混合 — 长上下文的两种负担

标准自注意力有两种负担。第一,计算量随序列长度的 平方(O(n²)) 增长。第二,推理时会把已经见过的 token 的键和值堆进缓存(KV 缓存),而这份缓存会随长度 线性 增大。在短提示下两者都微不足道,但到了几十万 token,两者都成了墙。

线性注意力·SSM 块(如 Mamba2、Gated DeltaNet)用另一种方式化解这份负担。它不缓存全部过去,而是把过去压缩进一个 固定大小的循环状态(recurrent state) 里随身携带 — 状态大小与长度无关,所以内存不膨胀,计算也接近线性。代价也是有的:正因为把过去硬塞进固定状态,对远处某个特定 token 的 精确回溯(exact recall) 能力,往往比全注意力更弱。

所以才有了 混合。把少数需要精确回忆的层留作注意力,其余大部分用线性·SSM 块填充 — 既保住质量,又把对长度的可扩展性拉向线性一侧。HyLo 更进一步,把剩下的注意力也换成 MLA(Multi-Head Latent Attention) — 一种把 K/V 压缩成低维潜在向量的注意力 — 从而再削一层缓存。一句话概括,混合就是把"注意力的精确"和"线性的廉价"混进同一个模型。

HyLo 做的事 — 升级改造配方

到这里为止都还是"混合更好"的泛泛之论。HyLo 的贡献在于 如何把它做得便宜。一直以来,混合的绊脚石是大多要 从头重新预训练 — 数千亿 token,以及与之相称的成本。明明有堆积如山、训练得很好的 Transformer 检查点,却在这个用途上派不上用场、白白闲置。

按摘要所述,HyLo 的配方把三件事编织在一起。

  • 结构移植(architectural adaptation):把现有 Transformer 块的一部分,替换成高效的块 — MLA 和线性块(Mamba2 或 Gated DeltaNet)。
  • 分阶段长上下文训练(staged long-context training):不一次性拉长上下文,而是分阶段扩展。
  • 教师引导蒸馏(teacher-guided distillation):以原始 Transformer 为教师,让新的混合学生对齐、追随它的行为。

三块拼图的目的只有一个 — 尽量保住原模型已经会的东西,同时只用廉价的后处理训练,就把它换上适配长上下文的高效结构。作者把这个过程称为 "升级改造","不丢弃、而是改造再用"的意味恰如其分(它算是把稠密模型改成 MoE 的那种传统升级改造的表亲)。

那些数字 — 全部按作者报告口径

把摘要给出的数字整理一下就是下面这些。再强调一次,全部都是 作者报告口径

上下文扩展     最多 32 倍(用廉价的后处理训练)
KV 缓存内存    削减超过 90%
最大上下文     vLLM 栈上最多 200 万(2M)token 的 prefill·decode
对比基线       "同等的 Llama 基线一旦超过 64K 上下文就会内存不足(OOM)"
规模           1B·3B 级(基于 Llama·Qwen 的变体)

最抢眼的一行是 token 效率。作者报告称,HyLo-Qwen-1.7B 只用 100 亿(10B)token 做后处理训练,就 在 GSM8K 上超过了用 4000 亿(400B)token 训练的 JetNemotron 基线。这是说训练 token 少了 40 倍却在某个特定任务上赢了 — 若属实,这个数字正好体现了本文的核心命题:"升级改造比从头训练便宜得多"。在长上下文评测 RULER(摘要提到 RULER-64K)上,它也据称超过了最新的升级改造混合基线,与此同时还保持了短上下文的质量 — 也就是说,避开了"为了拉长而丢掉短的"那种常见陷阱。

结语 — 有趣之处与可疑之处

有趣之处。 第一,它把问题重新框定为 经济学。"混合更好"早已为人所知,但如果能不用数千亿、而是用 100 亿 token 级别就得到它,情况就不一样了 — 因为堆积着的既有检查点全都成了原料。第二,重要的是这条路径与 量化不同的轴。量化把缓存里的每个数值压到更少的比特,混合则改变结构,让缓存 从一开始就更少产生。二者是从不同方向削减同一张内存账单,原理上可以叠加使用 — 把它和 把 KV 缓存降到 4 比特的尝试 并排来看,图景就清晰了。

可疑之处。 上面这些数字全是未经同行评审的预印本的 作者报告值,验证还没做。实验规模也小,只有 1B·3B,混合的质量能否一路撑到前沿规模,光凭这份摘要无从判断。"最多 32 倍""削减超过 90%"这类说法通常指向 最好的情况,GSM8K 上的优势也只是一个任务的结果,确切的差距摘要里并没有给。但方向仍然清楚:长上下文真正的战斗是内存,量化在削比特的同时,混合升级改造要做的是 从结构上减少缓存本身 — 而且走的是一条廉价的路,不从头训练,而是把已有的模型改造再用。等代码和可复现材料公开了,这是一个值得那时再回头看的方向。

参考资料

현재 단락 (1/25)

长上下文是当下 LLM 的最前线,而在这条前线上,真正的敌人不是算力,而是 **内存**。注意力的 KV 缓存会随序列变长而线性膨胀,到某一刻就会比权重本身还大 — [上一篇量化综述](/blog/2...

작성 글자: 0원문 글자: 3,061작성 단락: 0/25