当前位置:首页 → 电脑软件 → 国产人形机器人T800公布 → 宠文肉多一对一古代 v3.540 安卓免费版
v2.60 IOS版
v4.696.4792.149336 安卓版
v3.419 安卓免费版
v7.868 安卓免费版
v1.314.6830.275686 安卓免费版
v5.675.3717 安卓版
v3.593.9346 安卓版
v8.424.5606.386852 安卓版
v2.566 IOS版
v1.952.8882 安卓版
v7.778.9550.372997 安卓版
v7.74 PC版
v3.609.8756.73625 安卓汉化版
v1.508 IOS版
v4.753.333.700684 安卓最新版
v7.567.6470.714813 安卓免费版
v1.708.3230 PC版
v7.513.682.820275 最新版
v1.945 安卓汉化版
v1.85.8961 最新版
v3.482 安卓版
v2.620.6645.110587 安卓免费版
v9.612.9537.156025 安卓版
v1.755.7889.542422 安卓版
v7.72 安卓免费版
v5.395.4440.274511 安卓最新版
v1.288.7952 IOS版
v9.318.133 最新版
v5.253 安卓最新版
v9.73 PC版
v4.855.1403.451350 安卓免费版
v9.72.2892.973429 安卓最新版
v7.560.9130.424477 安卓版
v6.542 安卓版
v8.854 安卓免费版
v5.394.4573 安卓免费版
v1.439.9920.86183 安卓版
v7.514.2879 最新版
v7.843 安卓免费版
v2.263.6284.599625 安卓最新版
v6.977 安卓最新版
v6.101.8747 IOS版
v3.828 PC版
v9.222.4479.139451 安卓最新版
v5.38.2567.285210 PC版
v6.360.3619 PC版
v1.70.6749.998865 PC版
v8.994.7449 IOS版
v8.559 安卓最新版
v3.951.7041.558826 安卓最新版
v1.177.2096.580862 IOS版
v5.943 安卓版
v7.326.9321.925235 安卓版
v9.967.1058 安卓版
v8.542 最新版
v4.992 安卓最新版
v1.530 安卓最新版
v4.757.9499 IOS版
v1.617 安卓免费版
v3.798.8062.724584 IOS版
v3.939 IOS版
v2.87.9828.474915 安卓免费版
v3.202.3040 安卓最新版
v7.719.3930 PC版
v5.106.771 IOS版
v3.820.4512.99715 PC版
v2.732 最新版
v9.420 PC版
v3.575.2540.608723 IOS版
v3.262.7859.566349 PC版
v6.32.9297.924706 安卓最新版
v5.699.739 安卓版
v5.882 最新版
v6.975.3383.868303 安卓最新版
v3.724.9601.106863 安卓最新版
v8.392.427.38295 安卓免费版
v8.931.2084 安卓免费版
v8.611 IOS版
v9.550 安卓汉化版
v5.656.21.427851 PC版
宠文肉多一对一古代
不仅是提升效率,线性注意力在数据受限情况下也可能提升效果。
访谈丨程曼祺
整理丨姚一楠
注意力机制(Attention)是 Transformer 架构大型语言模型(LLM)的核心机制,它决定了模型如何处理、理解海量的文本信息。然而,传统全注意力机制的计算开销会随文本长度呈平方级暴增,这正是限制模型处理长文档、长上下文的关键瓶颈。
今年初,《晚点聊》的 103 期和 104 期节目分别讨论了注意力机制改进的两个主要方向:“稀疏注意力” 和 “线性注意力”。(文字版见《大模型 “注意力简史”:与两位 AI 研究者从 DeepSeek、Kimi 最新改进聊起》和《3700 次预训练寻找 “线性注意力” 非共识,MiniMax-01 开发者讲述 4 年探索》)
这期节目,我们继续关注线性注意力的新进展。在 9 月和 10 月底,阿里巴巴和月之暗面先后开源 Qwen3-Next 和 Kimi Linear 模型,其中的注意力机制都使用了线性注意力 DeltaNet 和 full attention(传统的全注意力)混合的方式。
此后在社交媒体上,两家公司的研究人员也透露,他们很可能在下一代旗舰模型中,使用新进释放的这些线性注意力改进成果。
本期《晚点聊》,我们就邀请到了 DeltaNet 核心贡献者之一杨松琳来聊聊 DeltaNet 和线性注意力的发展脉络。她也是线性注意力开源小组 FLA 的发起者,正在 MIT CSAIL 读博士三年级。
DeltaNet 的发展是多位研究者长期累积的结果。在 2021 年,Imanol Schlag、Kazuki Irie 和 Jürgen Schmidhuber 在线性注意力中引入类 Delta Rule 的可纠错更新规则;2023 年,Kazuki Irie 等从理论与形式语言视角刻画线性 Transformer 及其扩展的计算能力边界;2024 年,杨松琳等提出沿序列长度并行的 DeltaNet 训练算法以适配现代硬件;2025 年,杨松琳等进一步引入门控形式提出 Gated DeltaNet,强化记忆控制与检索表现。
杨松琳介绍了线性注意力和 DeltaNet 的发展脉络,为何 21 年刚被提出时没引起太多注意,后来怎么进化的。我们也讨论了重新去做 full attetnion 的 MiniMax(MiniMax 在今年初发布的 M1 中使用了线性注意力,在今年 10 月发布的 M2 中,转向全部使用 full attention),和未来要在旗舰模型上用线性注意力的 Kimi 与阿里的不同选择;线性注意力的优劣势,以及一些脑洞——如果算力无限,还需要线性注意力?杨松琳也分享了,作为 AI 研究员,怎么获得交叉技能,怎么开始发起 FLA 小组等成长经历。
DeltaNet 在 2021 年就被提出,但并行
晚点:注意力机制在大语言模型里有什么用,为什么重要?
杨松琳:语言模型预测下一个词时,若要用到前面的信息,就必须在句子层面做运算,把不同位置的信息混合。注意力机制通过建模两点之间成对的关系来整合前后文信息,最直接、最常用的是 Softmax Attention,平方复杂度的机制(如下图)。现在注意力也可以泛指在序列维度上做信息聚合的算子,比如线性注意力。
来源:Attention in transformers,3Blue1Brown
晚点:注意力机制最近好几个新动向都和你研究的 DeltaNet 有关,可以简单解释下 DeltaNet 是什么?
杨松琳:线性注意力的核心思想,最初是将自注意力(Self-Attention)中的 Softmax 函数移除。经过数学上的等价变换,它就可以被重写成 循环神经网络 (RNN) 的递推形式,从而将复杂度从平方级降至线性。2020 年的论文 Transformers are RNNs(《Transformer 是循环神经网络》)最早确立了这个研究方向。
随后的改进主要集中在两个方面:加门控(Gate)机制 或引入 Delta Rule(Delta 规则)。
其中,Delta Rule 是基于 2021 年 LSTM 作者 Jürgen Schmidhuber(于尔根·施密德胡伯) 团队的论文 Linear Transformers Are Secretly Fast Weight Programmers(《线性 Transformer 本质上是快速权重编程器》)。该研究以快速权重编程的视角重新解释了线性注意力,并指出:
- 线性注意力默认使用的权重更新方式是赫布学习(Hebbian Learning)。
- 为了实现更高效的上下文检索(Retrieval)能力——即 “前面写入一组 Key-Value,后面用 Key 就能取回对应的 Value”——可以将更新规则替换为更强大的 Delta Rule。
DeltaNet 正是在这一思路下诞生的,它利用 Delta Rule 来更快地写入和更新权重(即记忆状态)。
DeltaNet 起初不火,一是缺少关键架构改进,二是实现不够好。我去年在 NeurIPS 发的 Parallelizing Linear Transformers with the Delta Rule over Sequence Length(《利用 Delta 规则在序列长度上并行化线性 Transformer》)就是专门讨论如何并行化这个递归更新。
晚点:那从 2021 年 DeltaNet 被提出,到你们做的这个并行化的优化之间,还有哪些有关 DeltaNet 或者说线性注意力的改进思路?
杨松琳:近几年网络模块发展很快,如 TransNormerLLM 架构 引入了新的归一化方案,微软亚研院 RetNet 用了输出门模块;后面 Mamba 把短卷积带火。短卷积、输出归一化、门控成了标配。
但这些大多是在架构层面改进,而非更新规则,包括线性注意力和许多 RNN 变体的更新规则基本仍在最初框架里,只是加了一点简单的衰减。
晚点:更新规则改进和模型架构改进的区别是什么?
杨松琳:架构改进动的是外层结构,算子没变,比如在输出端加门控。线性注意力很早就在输出上加门控,Qwen 的 Gated Attention 本质也是输出门控;底层算子还是 GQA,用 FlashAttention 训练。
更新规则的改进则是直接改算子本身,外层架构仍照着近年验证有效的方案来用就行。比如线性注意力一般能写成一阶线性递归:输入通常是外积,转移矩阵默认是单位矩阵,改更新规则就是改这个转移矩阵。GLA、Mamba 把单位矩阵换成对角矩阵;DeltaNet 把它变成低秩单位矩阵;Kimi 把单位矩阵放宽为可学习的对角矩阵;RWKV-7 则用对角低秩矩阵作为转移矩阵。
晚点:DeltaNet,包括你后面又做的 Gated DeltaNet 具体是怎么改进更新规则的?
杨松琳:我们没发明全新东西,更多是用新算法把老技术重新做到了可用。就是前面提到的,我去年在 NeurIPS 的那篇工作,就是把它并行化,让它真正能大规模训练。
随后我在英伟达实习做了 Gated DeltaNet,它在 DeltaNet 上加了一个衰减。可以把它看成 DeltaNet 与 Mamba 2 的结合:保留 RetNet、线性注意力等当代架构思路,同时继承更强的 Delta Rule 更新,再加上衰减。衰减本质类似遗忘门,对 RNN 很重要,因为它的隐藏状态有限,必须忘掉一些,否则状态会被撑爆。
晚点:之前一些做算法的研究员告诉我,他们不擅长改算子。我理解你也是算法出身的,但会自己来改算子,是因为你自学了 Infra?
杨松琳:可以这么说。我硕士时,做过 “上下文无关文法”,需要把大量 python 运算并行到 GPU 上,就写 CUDA 加速。后来做模型架构,写算子更顺手,于是做软硬件结合的算法设计,让模型在 GPU 上跑更快。
我对并行加速还挺有兴趣的。刚读博士时,斯坦福有个研究组叫 Hazy Research,FlashAttention、Mamba 都出自那儿,他们倡导硬件友好型算法。我有学习他们的风格,既写算子也做算法。
Qwen 和 Kimi 下一代旗舰模型可能转向线性注意力,Minimax M2 则用回全注意力
晚点:回到最近的一些进展,阿里的 Qwen3-Next 以及 Kimi 的 Kimi Linear 和 DeltaNet 的具体关系是?
杨松琳:Gated DeltaNet 今年被 Qwen 团队系统地测了一轮。他们对全局注意力、混合滑窗注意力、Mamba 2 和 Gated DeltaNet 做了真正的 apple-to-apple 的对比,结果 Gated DeltaNet 最好,于是用进了 Qwen3-Next。
Kimi Linear 用的 KDA 基本是 Gated DeltaNet 的细粒度版:DeltaNet 部分不变,但把衰减从粗到细。原先 high-dim(高维,指模型的特征表示向量) 128 个 channel 共用一个遗忘率,现在每个 channel 各有一个,有的忘得快,有忘得的慢,用来存不同时间尺度的信息。细粒度衰减的思路本来也常见,比如我在 ICML 2024 的 Gated 线性注意力(Gated Linear Attention Transformers with Hardware-Efficient Training)就用了这种精细衰减。
因此 KDA 可以看成 Gated 线性注意力 + DeltaNet,而 Gated DeltaNet 是 DeltaNet + Mamba 2;在衰减粒度上,它们关系就像 GLA 和 Mamba 2 的差别。
晚点:为什么 Qwen3-Next 和 Kimi Linear 现在都要把线性注意力和完全注意力(full Attention)混用,不能全用线性的?
杨松琳:线性注意力给速度,完全注意力给容量。线性注意力的容量太小,短文本还可以应付,到长文本就会表现很烂。线性注意力机制每层有大小固定的 RNN 隐藏状态,层数和维度一旦确定,整套模型的可存信息量也就固定了。
传统 Softmax Attention 的 KV Cache 可以看成隐藏状态,会随着 token 线性增加,长文本不会被容量卡住;但推理时要从全局读取信息,KV Cache 过大时会非常慢。(注:KV Cache :在推理时缓存已生成 token 的 Key 和 Value,用来避免重复计算注意力,从而加速生成。)
晚点:现在用 DeltaNet 的模型都不大,Qwen3-Next 是 80B ,Kimi Linear 是 48B 。把线性注意力用到参数更大的模型上,有什么瓶颈吗?
杨松琳:工业界的常见做法,就是先用小模型验证架构、降风险,再训大模型;直接在超大规模上试,两次失败公司就破产了。
千问和 Kimi 其实都在往大模型走。最近在 Reddit 上 ,Kimi 团队透露他们的下一代旗舰模型,就是 K3 大概率继续沿混合 KDA 方向;Qwen3-Next 的最终版本 Qwen 3.5 应该也会走混合架构。
晚点:MiniMax 很早就把线性注意力用到了超大模型上,他们在今年初发布的 4560 亿参数的 MoE M1 上,就用了混合线性注意力与全注意力的 Lightning Attention,但 10 月底发布 MiniMax M2 又回到了完全注意力。从业者是怎么讨论这个转变的?
杨松琳:大家都觉得这个现象挺好玩。这有点像,线性注意力是一个 “坑”,MiniMax 赶着跳出去,Qwen 和 Kimi 又急着往里跳;不过 Minimax 也没完全失去信心,还在验证混合架构。
Minimax 可能之前受 Lightning Attention 的伤太大了,一朝被蛇咬十年怕井绳。Lightning Attention 很弱,只是在最原始线性注意力上叠了粗粒度、输入无关的衰减。他们当时直接 Scale Up 到几百 B ,可能是 Eval(验证)没搭好。
结果 MiniMax 发现 Lightning Attention 在 MMLU(注:测试大模型在 57 个学科上综合知识与理解能力的标准考试题测评) 等短程任务上跟完全注意力差不多,用个比较短的滑窗就够了。但在多跳推理(注:Multi-hop Reasoning,需要模型跨越多个信息点、分步骤串联线索才能得出答案的推理方式,典型例题如 “爱因斯坦出生时德国的国家元首是谁?”)上,完全注意力能直接建模点对点关系,叠几层就能自然形成多跳推理;线性注意力或混合结构会把信息压得很模糊,准确率掉得很厉害。
现在 Agent 做任务都会想很多,多跳推理在 Agentic AI 里非常重要。MiniMax 觉得混合架构暂时解决不了想主攻的 Agentic AI,退回完全注意力挺自然的。
他们的反思里也有不少值得学的点,比如基准选择:一些多跳推理 benchmark,如 BBH 其实很容易,可以找方法让架构表现很好,但不代表模型在真实场景里就真的会推理。
晚点:据你所知,DeepSeek 有来研究线性注意力机制改进吗?
杨松琳:他们应该更相信稀疏注意力,年初发的 MLA 和最近 DeepSeek V3.2 的 DeepSeek-Sparse-Attention 都是稀疏注意力的改进。
晚点:对比稀疏注意力和线性注意力,未来的潜力有什么区别?
杨松琳:单层潜力肯定是稀疏注意力更强,实际应用中不好说。
稀疏注意力通过减少激活的 KV Cache 读取来加速,依然需要存全部 KV Cache,每次运算时选一些出来。大家就会有满满的安全感,因为 token 不容易掉。理论上 KV Cache 够大,效果就能逼近 Softmax Attention;Softmax Attention 的 Attention Map 相当稀疏,所以稀疏注意力的效率会更高。但当规模很大、序列很长、KV Cache 的大小本身成为瓶颈时,稀疏注意力就爱莫能助了。
线性注意力有理论缺陷,因为状态空间固定,但这也能是加速推理的动力。混合线性注意力一般 75% 的层都被换成了 RNN,RNN 的 Cache Size 在长文本推理时可以忽略,KV Cache 大小就减了 3/4,可以支持更大的批量推理。批量越大,做推理效率越高,同时服务很多用户。
不仅是提效,在数据受限的后训练和强化学习中,线性注意力可能有性能优势
晚点:如果有无限算力,大家还有动力做完全注意力外的方法改进吗?
杨松琳:给我无限数据和算力,我当然直接用完全注意力。但 bound 住(约束)我们的不仅是算力,还有数据。
我们必须用有限数据下更高效的架构;这时候完全注意力反而是个劣势,因为它同样数据下学的比较慢,没有引入归纳偏见。归纳偏见就是人的先验。
线性注意力更关注邻近 token,在数据受限时可能表现更好;后训练、强化学习的数据更少,混合架构的优势可能会慢慢显现。
晚点:所以线性注意力除了省推理算力,在数据更少的后训练、强化学习里也可能更好?
杨松琳:从电路复杂度看,完全注意力属于 TC⁰,表达能力不够,所以 Transformer 要解决复杂问题只能靠很长的思维链。
这是 DeltaNet 另一个被忽视的优势,它从计算理论上是 NC¹-complete 架构,能超越 TC⁰,更擅长状态追踪,这对 Agentic AI 至关重要。比如你写代码时变量名不断变,模型得在内部维护变量状态;再比如网页操作,Agent 得知道你按什么顺序做了什么、到了什么状态,才能决策。
前段时间有篇很有意思的论文 Recurrence-Complete Frame-based Action Models,专门讲为什么 Agentic AI 更该关注状态追踪和循环机制。
(注:TC⁰ 和 NC¹ 是不同的复杂性类,复杂性类是将有相似计算难度的一群问题归纳在一起的集合。简单来说,NC¹ 的电路允许 “对数深度”,而 TC⁰ 的电路只有 “常数深度”,但 TC⁰ 允许使用阈值门,在很少的并行层数里完成相对复杂的运算;从已知结果看,TC⁰ 是被包含在 NC¹ 里的一个子类。很多看起来 “复杂” 的认知任务,难点往往在于需要较长的计算深度来逐步更新和传递信息,例如在读代码时持续追踪程序状态和变量取值,这类能力更依赖足够多轮的迭代计算,而不仅仅是单步中的并行算力。)
晚点:你说线性注意力带归纳偏见可能提升效果;而之前 AI 界著名文章 “The Bitter Lesson” 它认为加人为结构和先验通常不如寻找可以用更多算力、数据,做更大规模训练的方法。
杨松琳:100 个人有 100 种对 Scaling 和 The Bitter Lesson 的解读。大语言模型本身就是把人类先验注入进去的例子。我更支持先把方法做到效果和效率上能 scalable,归纳偏见不用管,好不好验一下就知道了,不用多做讨论。
晚点:你这里说的 scalable 的关键是什么?
杨松琳:一是大规模训练下效率要有保证、算法要硬件友好;二是模型放大后依然有效,很多改动在小模型好用,规模扩大就失效。
研究发现是连点成线:本科时形成对矩阵代数的兴趣,算法优化看多了熟能生巧
晚点:你自己开始关注到线性注意力改进和 DeltaNet 方向的过程是怎样的?
杨松琳:我一直喜欢做模型和算法,硕士时就喜欢看各种魔改注意力的方法,但真正开始研究是 2023 读博之后。当时在想怎么选方向,既感兴趣又专业相关。完全注意力的长文本问题似乎一直解决不掉,还有有意思的算法可玩,我就跑来玩这个领域了。
晚点:你最初有哪些一起研究的伙伴?你之前有提到过斯坦福的 Hazy Research,还有吗?
杨松琳:他们在新架构上做得很多,我和他们挺熟,比如 Simran Arora、Albert Gu、Tri Dao。国内我觉得微软亚研董力团队也很强,我跟 RetNet 一作孙宇涛讨论挺多。还有之前钟怡然(MiniMax 前算法总监,曾在上海 AI lab 担任 PI)那边的秦臻,从知乎私信联系到我,聊着聊着就合作了一两篇论文,就是比较早的线性 RNN 工作 HGRN。
晚点:现在的导师会给你什么帮助?
杨松琳:他可以帮我搞来卡。(笑)老板在最开始会有些大方向的感觉,然后提供算力和写论文的支持,别的方向也不一定能帮得上学生,因为大家读博的目标就是在这个领域比自己老板还懂。
但我觉得老板还是很有眼光的。他建议我关注软硬件结合的算法设计,以及数值代数的一些思路。比如 DeltaNet 用到数值代数里经典的 Householder 矩阵做累乘,这和线性注意力的 Chunkwise 算法能很好结合,最后就成了 Parallelizing Linear Transformers with the Delta Rule over Sequence Length 的核心想法。
晚点:你做算法又写 CUDA kenel,这些跨领域能力怎么积累的?
杨松琳:还是兴趣驱动,我喜欢矩阵运算和 kernel 优化。数值计算偏应用数学,里面有很多矩阵加速算法,深度学习也离不开矩阵。我对这些本身就感兴趣,看得比较多,熟能生巧吧。
晚点:你本科是在南方科技大学,这本身是一所挺新的研究性大学,当时的学习经历对积累交叉知识和视野有什么帮助?
杨松琳:这还真有一些关系。我本科最喜欢的课就是线性代数,用的是吉尔伯特·斯特朗(Gilbert Strang)的经典教材。他从空间角度讲,特别直观,让我对线性代数兴趣很强。如果是国内常见那种上来先讲行列式、公式推导,我可能就没什么兴趣了。
晚点:你改进 DeltaNet 让它能并行的过程中,具体是怎么突破,获得思路的?
杨松琳:并行线性递归要先展开,会出现转移矩阵,继续展开就会出现一个累乘,难点就是高效算这个累乘。DeltaNet 的转移矩阵像 Householder,我一开始不知道怎么算,后来发现可以用 WY 算法把累乘变成累加,形式和线性注意力很像,我就意识到它可能能和 Chunkwise 算法兼容。之后推了一阵,推通了还挺开心,又找到一个好玩的算法,也能把这个方向继续 scale 上去。
晚点:怎么想到 Householder 矩阵,怎么意识到这个关联的?
杨松琳:就是一直想吧,我从 2023 年 9 月开始想怎么并行 DeltaNet,到 2024 年 3 月才想出算法;然后硕士我在上科大上过一门 “矩阵方法”,讲 Householder 和 QR 分解;QR 是很经典的算法,Householder 累乘最早就是用来做 QR 的,最近我发现它还能加速 DeltaNet。在英伟达做 Gated 线性注意力那个项目时,我对线性注意力的分块算法理解更深,这些积累后来都启发了新想法,整体还是循序渐进。
晚点:基础科学里,常发生把很多看似无关的东西放在一起产生新成果,比如物理学有时会从几百年前的数学里获得启发。你觉得 AI 能什么时候能独立产生这种联想?
杨松琳:我觉得大模型应该能独立发明这个(让 DeltaNet 并行化的)算法,只要提示词合适,它可能就能把后面所有推导都做出来。检验推导正确性可以用 RL 做,只要 reward 能验证,RL 就能解决,用在科学发现上也挺有用的。
FLA 小组:像运营产品一样 “运营” 技术;Kimi 从 FLA 找到线性注意力研究员
晚点:你去年 1 月开始维护 FLA 这个线性注意力的开源社区,契机是什么?做开源的过程中有什么故事或收获?我看到贡献者里有 Kimi 的研究员,也有欧洲做 AI for science 的开发者。
杨松琳:当时在知乎看到一篇文章,说 Flash Attention 的成功本质是产品成功。Tiling、online softmax 早就有了,xFormers 里也有初版思路,但 Tri Dao( FlashAttention 的作者)把它当产品做,重视接口、重视运营、积极和社区互动、按用户需求迭代,关键是好用,即插即用、装个包就能跑。
我被这套产品思路打动了,就想,线性注意力为什么不做一个?线性注意力有不少算法成果,但缺好实现,那我就做个开源库,把 Triton 算子和各种 layer 写好,让用户一个库就能跑各种带 kernel 的模型,大家觉得好用就会留下来,还会给反馈帮我们迭代。比如后来很多人要变长训练模块,我们发现需求巨大,我就和张宇商量,直接把可变长度功能全部写进 FLA。
晚点:张宇是你做了 FLA 之后,他关注到这块,加进来成为核心贡献者的吗?
杨松琳:张宇是 FLA 的核心贡献者,也是 Gated 线性注意力的作者。我和他 2020 年就认识了,那时我们都在做 parsing,要写很多并行算法。他当时维护一个叫 supar 的库(su 是苏州大学,par 是 parsing),实现非常 clean、并行效率很好,基本所有做 parsing 的人都会用,特别酷,所以做 FLA 时我第一时间就想把他拉进来。
他做库水平很高,也很适合写并行算法和 kernel。你翻 FLA 的 GitHub,贡献行数和总贡献数他都是最多的,代码特别强。后来 Kimi 想做混合线性注意力,就想找 FLA 的作者;我人在美国,他们不可能把我弄过去,就把张宇弄去 Kimi 做研究。(注:parsing:编译,自然语言处理中的一个模块,用来分析句子的结构,例如解析句子 “I love you” 时,parsing 会把它拆成主语、动词和宾语。)
晚点:所以是 Kimi 想做线性注意力,从 FLA 社区里注意到了张宇,然后再邀请他加入的?
杨松琳:对。
晚点:Kimi 关注到这个方向,这是谁的判断了?
杨松琳:应该是 Tim 吧,周昕宇(Kimi 的联创)。
晚点:我看张宇的头像是个二次元少女拿着实验笔记本。
杨松琳:(笑)经典 stereotype(刻板印象),二次元头像的人 coding 强。
希望下一步把稀疏注意力和线性注意力真正做通,彻底解决长文本问题
晚点:我们可以分别讲讲,注意力机制接下来的演进,首先是稀疏注意力,你看到的趋势是什么?
杨松琳:核心是动态稀疏,静态稀疏之前被证明效果不太好,难点卡在如何在硬件上高效实现。动态稀疏现在有两条路:block 级和 token 级。block 级快但漏信息风险大,所以怎么选准 block 很关键;token 级更灵活但实现更难。
比如 NSA 是 block 级,每次只选一个 block,方便连续读入、顺序计算。DSA(DeepSeek Sparse Attention)把 block 的结构化约束扔掉,回到不结构化的 token 级动态稀疏。比如要选 512 个 token 刻画一个 token,选 512 个最相关的 token,而不是选 512/32 个 block,颗粒度更细也更准,但实现难度大的多。
Deepseek 训练不是从预训练开始稀疏化,而是训练到中途才用:先蒸馏,把 DeepSeek 3.1 转成 MQA,再把注意力分布蒸馏到 indexer,作为初始化。
DSA 的 trick 在于,用一个非常轻量的平方复杂度 Attention 做 “indexer” 来生成全局注意力矩阵,因为这个算子能在 FP8 下跑、不需要 online softmax,只是矩阵乘法,所以算得很快。indexer 得到 L×L 的注意力矩阵后,直接做 Top-K,每个 token 找出最相关的前 K 个 token。
稀疏还有梯度传递的问题,block 选不准可能也是梯度质量差导致。
最后,稀疏注意力也可以考虑减少 KV cache。现在大多数稀疏注意力保留全部 KV cache ,几乎和完全注意力一样,还是会卡在 KV cache 的规模瓶颈。
晚点:线性注意力呢?
杨松琳:线性注意力的架构方向验证得差不多了,更值得探索的是更新规则。可以想想 DeltaNet 外还有哪些更新方式既能更好地并行实现,又更具表达力。
另外,沿着 DeltaNet 还能继续做很多,比如加衰减、norm 衰减;DeltaNet 在 test-time training 视角就是近似梯度下降,有些工作像 DeltaProduct 每步会做多次梯度下降,而不是一次;还有动态扩容思路,如记忆混合,把 RNN 的状态当成 MLP 的权重矩阵,既然可以在权重矩阵上做 MoE ,那么也能在 RNN 的记忆状态做 MoE;最近也研究有把滑窗和线性注意力结合,邻近 token 保留滑窗,远距离用线性注意力压缩。
晚点:你觉得当前的注意力机制离理想状态还差什么?
杨松琳:先不说理想,我更希望下一步看到有人把稀疏注意力和线性注意力真正做通。现在的混合注意力依然在全局保留完全注意力层,长文本解码时还是会被它拖慢。如果能把所有完全注意力都换成稀疏注意力,比如 DSA 混 KDA 之类的组合,至少能把长文本问题阶段性解决,KV cache 规模降下来,就能做更多长文本应用,也能做更多 agentic 任务。(01:18:43)
晚点:现在大家都是混线性注意力和完全注意力的,而不是混线性注意力和稀疏注意力,原因是什么?
杨松琳:我觉得架构研究还是要扎实,一次动一点、验证透,再动下一步,不可能一步迈太大。先保留一些完全注意力,用来验线性注意力;混合架构在旗舰模型上验证稳定后,再去验证稀疏注意力也不迟。(01:19:59)
晚点:更广泛地说,你现在会关注大模型在预训练、强化学习之后的下一步吗?大家在讨论预训练 + RL 可能到了瓶颈,也许需要新范式,比如在线学习、持续学习、自主学习。
杨松琳:持续学习大家都很关心。这里有很多 open question,怎么做没有定论,效果也不显著,但它一定很重要。如果 AI 能持续吸收外界信息,就不需要隔段时间重训一次;有也有强的商业价值,比如做个性化,用户持续交互,模型积累了这些记忆就能更好地把握用户喜好。
晚点:现在模型的记忆能力并不是内化在模型里,而是靠工程或外挂方式实现的,是吗?
杨松琳:对。预训练是把信息压进 FFN 的权重里;另一类是上下文工程,通过注意力做上下文学习。现在主要就是这两种,再有就是一些记忆外挂。(注:FFN 是前馈神经网络,Transformer 中的前馈全连接层。在注意力之后,模型会把每个 token 的向量分别送入两层 MLP 作为存储的记忆,这部分就是 FFN。)
晚点:如果模型能持续学习,它会越来越懂你,那上下文工程的空间不就变小了?
杨松琳:这依然是个 open question:哪些信息该存进权重矩阵,哪些该放在上下文里,没有定论。
其实两者是对偶的:梯度下降可视为一种上下文学习,而上下文学习也能视为梯度下降。现在流行的测试时训练(Test-Time Training)把每个 token 当训练样本,输入后做一次梯度下降快速权重更新;在 RNN 里快速权重可以视为隐状态。既然有这种对偶,FFN 如何在训练中牢牢记住成千上万的 token,也能启发长文本问题:把信息写进权重,再用动态的测试时训练,也许能走出更好的持续学习路径。
我对这个路径很看好的。现在的权重推理时不更新,无法吸收新信息;测试时训练或快速权重编程(Fast Weight Programming)允许处理新 token 后实时更新权重。如果能把这套机制打通,长文本问题能靠更紧凑的权重存更多信息,持续学习也能顺带解决。(注:测试时训练:在推理阶段对模型参数进行快速更新,让模型利用当前输入的信息提升输出表现。例如模型在处理一篇医学文章时,可以先用文章里的一小段内容做一次快速梯度更新,让模型立即更擅长理解该领域的术语,再继续生成答案。)
题图来源:月升王国
相关版本
多平台下载
查看所有0条评论>网友评论