量子位 15小时前
华为新架构砍了Transformer大动脉!任意模型推理能力原地飙升
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

是时候给Transformer的大动脉动刀子了。

因为即便它享有当下 AI 世界基石的地位,但自身问题也是非常明显:

一旦遇到复杂的数学题或者需要多步逻辑推理的时候,就开始一本正经地胡说八道了……

问题究竟出在了哪里?

答案就藏在 Transformer 的核心机制里——Attention

传统 Attention 机制本质上像是一种配对比较:每个词只和另一个词直接发生关系,生成一个注意力权重。

这种架构虽然擅长捕捉长距离依赖,但在建模复杂、多跳、多点之间的逻辑关系时却显得力不从心了。

例如它能轻松理解 "A 认识 B",但如果要它理解 " 张三通过李四认识了王五 ",即多跳、多点之间的复杂、间接关系,它的脑回路就显得不够深,推理能力的天花板瞬间触顶。

现在,这个天花板,被华为诺亚方舟实验室捅破了!

因为就在最近,团队祭出了一种全新架构,叫做Nexus,即高阶注意力机制(Higher-Order Attention Mechanism)。

它可以说是直接狙击了 Attention 机制的核心痛点,使用更高阶注意力,就能有效地建模多跳、多点之间的复杂关联。

并且从实验结果来看,效果也是有点惊艳在身上的。

只要换上 Nexus 这个新架构,模型在数学和科学等复杂推理任务上的能力,都能立马实现大幅飙升,而且还是参数零增的那种。

妙哉,着实妙哉。

接下来,就让我们一同来深入了解一下 Nexus 的精妙一刀。

高阶注意力机制砍出的精妙一刀

要理解高阶的意义,我们必须先回顾传统自注意力机制的根本缺陷。

标准的自注意力机制本质上是将输入序列 X 分别通过三个线性变换 WQ,WK,WV 生成 Query(Q)、Key(K)、Value(V),再通过 softmax 计算注意力权重:

但这里就出现了一个关键的问题:Q 和 K 都是静态的、与上下文无关的线性投影。

也就是说,某个 token 的 Query 向量仅由它自己决定,无法感知其他 token 的存在;这导致注意力权重只能反映两两之间的直接关系。

精妙第一刀:Q 和 K 的革新

华为诺亚方舟实验室的第一个刀法,就精妙地砍在了这里:Nexus 让 Q 和 K 的生成过程本身也变成一个注意力操作。

换句话说,token 在计算最终的 Q 和 K 之前,会先进行一次 " 预推理 ";这个过程,其实就是一个嵌套的自注意力机制。

Token 首先通过这个内部循环,从全局上下文中聚合信息,形成一个更加精炼、更具上下文感知能力的表示,然后再用这个表示去计算最终的 Q 和 K。

这就好比,在你问我答(Q 和 K 计算 Attention)之前,每个 token 都先在内部进行了深思熟虑,充分吸收了它在整个序列中的环境信息。

这样生成的 Q 和 K,自然就摆脱了线性投影的僵硬,具备了捕捉复杂关系的动态性。

精妙第二刀:巧用递归框架

Nexus 架构最精妙之处,还在于它的递归框架(Recursive Framework)。

这个内部注意力循环可以被递归地来嵌套。

如果我们将一层 Attention 视为一阶关系(A 认识 B),那么将 Attention 的输出作为下一层 Attention 的输入,就可以构建二阶关系(张三通过李四认识王五),乃至更高阶的关系。

在 Nexus 中,这种递归嵌套被巧妙地集成在一个单层结构中,形成了一个层次化的推理链。

论文进一步将上述过程递归化,定义第 m 阶注意力为:

其中,m=1 就是标准注意力;m=2 表示 Q 和 K 由一次内层注意力生成;m=3 表示 Q 和 K 由二阶注意力生成,相当于 " 注意力的注意力的注意力 "。

这种结构天然支持多跳推理链,就像人在解一道数学题时,先理解题干中的关键变量(第 1 层),再思考它们之间的公式关系(第 2 层),最后验证整体逻辑是否自洽(第 3 层)。

精妙第三刀:不增参数

复杂架构往往意味着更高的计算开销和更多的参数量,但 Nexus 通过精巧的设计,完全规避了这些问题——权重共享策略

具体来说,无论是内层还是外层的注意力模块,都复用同一组投影权重 WQ,WK,WV。

这意味着,尽管计算路径更复杂,但模型参数量和原始 Transformer 完全一致。

这种设计背后有一个关键假设:无论处于递归的哪一层,将 token 投影为 Query 或 Key 的语义变换方式是相似的。

团队通过实验证明,这一假设是成立的。

在 Pythia-70M 的消融实验中,使用权重共享的 Nexus-QK-Shared 版本,平均准确率仍比基线高出近 1 个百分点,而参数量毫无增加。

这就让 Nexus 成为了一种极其高效的表达密度提升器——用相同的参数,实现更强的推理能力。

只要换上 Nexus,推理效果立竿见影

那么 Nexus 的效果到底如何?

论文在两个维度做了验证:从零训练的小模型,以及对已有大模型的架构改造。

小模型全面领先

研究团队在 Pythia 系列(70M 到 1B)上从头训练 Nexus,并在六个标准推理数据集上评估:ARC-C、ARC-E、HellaSwag、LogiQA、PiQA 和 SciQ。

结果非常一致:Nexus 在所有规模上都优于原始 Transformer。

尤其在需要多步推理或科学常识的任务中提升显著。例如:

在 SciQ(科学问答)上,70M 模型准确率从 61.5% 提升至 68.5%,提升 7 个百分点;

在 PiQA(物理常识推理)上,1B 模型从 62.5% 提升至 63.6%。

这说明 Nexus 特别擅长处理那些不能靠表面模式匹配解决的问题,是真的有在做推理。

大模型改装即用

面对规模更大的模型,Nexus 还体现出了即插即用的能力。

团队将 Qwen2.5 的 1.5B 和 7B 版本的标准注意力层直接替换为 Nexus 结构,仅在 SFT(监督微调)阶段进行训练,未改动预训练权重。

结果表明,在三个高难度数学推理基准上(MATH-500、AIME24、GPQA-Diamond),Nexus 均带来稳定提升:

Qwen2.5-1.5B 在 MATH-500 上准确率从 78.6% → 80.1%;

Qwen2.5-7B 在 AIME24 上从 45.2% → 47.5%。

尤其值得注意的是 AIME24 的提升,因为这类题目要求严格的多步逻辑推导,错误一步就全盘皆输。Nexus 的改进说明,它确实在内部构建了更连贯的推理链。

从这一层面来看,Nexus 不仅是一个新训练范式,还是一套架构升级套件。你不用重新训练一个千亿模型,只需在微调阶段替换注意力层,就能解锁更强的推理能力。

推理能力可内生于架构

虽然 Nexus 目前聚焦于语言模型,但其思想具有普适性。

高阶关系建模在视觉、图神经网络、多模态任务中同样关键;例如,在视频理解中,"A 看到 B 打了 C" 就是一个典型的三元关系,传统 Attention 难以直接捕捉。

华为诺亚团队表示,下一步将探索 Nexus 在视觉 Transformer 和多模态大模型中的应用,并优化其计算效率。

Transformer 的智商天花板,或许从来不在参数量,而在其注意力机制的表达能力。华为诺亚的 Nexus,用一种优雅而高效的方式,为这一核心模块注入了高阶推理能力。

它不靠堆料,不靠提示工程,而是从架构底层重构了模型的思考方式。

因此,Nexus 也提醒了我们:有时候,聪明的架构比规模的大小更重要。

论文地址:

https://arxiv.org/abs/2512.03377

一键三连「点赞」「转发」「小心心」

欢迎在评论区留下你的想法!

—    —

点亮星标

科技前沿进展每日见

宙世代

宙世代

ZAKER旗下Web3.0元宇宙平台

一起剪

一起剪

ZAKER旗下免费视频剪辑工具

相关标签

华为 脉动 ai
相关文章
评论
没有更多评论了
取消

登录后才可以发布评论哦

打开小程序可以发布评论哦

12 我来说两句…
打开 ZAKER 参与讨论