大数跨境

0.016% 的 Astra 微创手术,治好了 Qwen 的 3 万 Token 死循环

0.016% 的 Astra 微创手术,治好了 Qwen 的 3 万 Token 死循环 AI科技评论
2026-11-03
27
导读:当「读懂权重」和「改写权重」被同一个模型打通,「AI 改进 AI」就多了一条不经过梯度下降的短路通道,模型维护与模型能力放大之间的边界开始模糊。
当「读懂权重」和「改写权重」被同一个模型打通,「AI 改进 AI」就多了一条不经过梯度下降的短路通道,模型维护与模型能力放大之间的边界开始模糊。

    作者丨郑佳美

    编辑丨岑   峰

                                                                                                       

面对模型推理陷入死循环、耗时巨大却无法输出答案的困境,传统方案通常涉及调整训练数据、优化推理策略或重新微调。然而,MIT CSAIL 近期提出了一种更为直接的解决方案:利用 GPT-6 Astra 直接检查并修改 Qwen3.5-0.8B 的内部权重。

实验结果显示,Astra 仅改动了约 0.016% 的权重,便使 Qwen 在 MATH-500 基准测试中的成绩从 7.9% 飙升至 48.1%,提升近 6 倍。

这一突破的核心意义不在于性能提升的数字本身,而在于它将机制可解释性的输出从“人类可读的解释”转化为“机器可执行的编辑指令”。研究者将这种能力定义为权重空间编程(Weight-Space Programming),意味着当神经网络参数能被还原为具体计算过程时,LLM 有望像修改代码一样直接改写模型内部功能。

01 只给一堆数字,Astra 如何识别网络功能?

为了验证 Astra 对权重的理解能力,研究者首先进行了“猜游戏”实验。他们训练了架构完全相同、参数量均为 17,253 个的四个神经网络,分别学习 Breakout、Pong、Pac-Man 和 Freeway 四款游戏。在不告知游戏名称且禁止运行网络的情况下,Astra 仅凭权重数据便准确识别出了所有游戏。

Astra 利用的是训练留在参数中的痕迹。例如,Breakout 游戏中球拍始终位于屏幕底部,因此负责读取玩家位置的输入层权重会集中在网格下沿。结合输出层对动作的响应,Astra 即可推断出网络的控制逻辑。

相较于具有明确物理含义的游戏输入,大型语言模型的隐藏维度缺乏直观标签。为此,研究者进一步训练了三个约 720 万参数的字符级 Transformer,分别学习 Python、罗马字日语和英语,并移除输入 Embedding 和输出层,仅保留中间模块供 Astra 分析。

Astra 通过分析 Transformer 中的 Induction Head(归纳头)来破解这一难题。Induction Head 负责复制历史文本,其运作依赖于 QK 电路定位历史位置以及 OV 电路写入信息。Astra 沿着信息传递路径分析不同 Head 的矩阵乘积,成功在 Python 模型中识别出 3 个 copying head,并在其他模型中各识别出 1 个,这一判断随后通过实际运行得到了验证。

由于 Python 代码中变量名重复率高,复制能力更为关键,而罗马字日语在隐藏特征写入方向上呈现不同分布,Astra 据此准确判断了训练语言。这表明,虽然单个权重难以解释,但多个矩阵沿计算路径组合后可呈现明确功能。

值得注意的是,静态参数主要提供结构线索。在 Gemma 实验中,Astra 仅靠权重无法判断模型是否易受错误答案诱导,需增加少量 Attention 激活信息后才成功。但这已证明 Astra 具备识别特定计算路径的能力。

02 让 LLM 直接编写权重

既然能识别功能路径,能否直接设计路径?研究者让 Astra 基于一套全零权重和游戏规则,直接生成策略网络,全程无训练数据、无损失函数、无梯度下降。

Astra 采取了类似编译程序的思路:先将任务拆解为计算步骤,再分配给不同网络层。以 Breakout 为例,它将计算落点的任务交给前层,后层根据落点与球拍的位置差生成动作。

结果令人惊讶:生成的 Breakout 网络平均得分达 517.4,远超 PPO 训练 1 亿步(15.8 分)和 5 亿步(25.3 分)的成绩。这证明当任务可转化为明确算法时,算法可直接转换为网络参数。

针对更复杂的 9×9 围棋任务,Astra 借助 Python 构造权重矩阵,将策略拆解为识别棋子、传播信息、判断气数及选择落点等步骤,相当于在神经网络中实现简易图算法。尽管受限于固定深度导致长程关系处理困难,该网络在允许 50 次前向测试后,仍击败了训练 2000 万局的同规模 PPO 网络。

在 Kinetix 车辆控制实验中,Astra 发现原网络缺乏冲坡前的倒车加速动作,遂在原网络中插入一条小型计算路径。仅修改 3.5% 的权重,车辆成功率便从 0% 提升至 100%。这种修改类似于为程序增加条件分支,精准补全了缺失的逻辑。

03 Qwen 的 0.016% 权重手术

回到 Qwen3.5-0.8B 的数学推理问题,Astra 发现模型经常陷入重复推理的死循环,导致耗尽 Token 预算却无答案。通过分析正常与异常生成的内部状态差异,Astra 锁定了第 15 层和第 19 层 MLP 中与循环相关的参数方向。

Astra 从这两层各 64 个输出列中减去与循环相关的方向,共涉及 131,072 个参数。这一操作基于 Transformer 的残差流机制:MLP 输出矩阵的每一列对应一个可写入残差流的向量,若某方向持续强化,会导致模型陷入循环。削弱该方向的写入强度,即可阻断循环。

实验数据显示,随着测试预算增加,MATH-500 成绩显著提升至 48.1%,触及长度上限的比例从 93.0% 降至 33.1%。同时,GSM8K 和 IFEval 成绩也大幅提升。深入分析发现,579 条回答由错变对,其中绝大多数原本因截断而未完成,修改后模型能正常结束思考,且推理长度中位数仍保持在 9172 Token,证明并非通过提前停止来“作弊”。

作为对照,Gemma 3 270M 因较少出现此类循环,权重编辑后性能提升有限。这表明局部参数手术更适合修复可定位的计算结构故障,而非补充未掌握的知识。

04 比“打补丁”更重要的信号

这项研究的深远意义在于开启了“AI 改进 AI”的新路径。与传统梯度下降训练相比,权重编辑无需海量数据和算力,可在推理上下文中直接完成。MIT 团队将其与递归自我改进(RSI)联系起来,引发了关于“神经网络基因编辑”的讨论。

尽管当前技术仍需人工设计探测策略,距离真正的自主 RSI 尚有距离,但其核心突破在于:可解释性研究的产物已从“给人看的分析”转变为“机器可执行的指令”。训练赋予模型能力,而权重空间编程则提供了直接修复和调整这些能力的可能。未来,若能稳定识别并验证这些路径,模型或将像软件一样接受局部补丁,这将深刻改变模型生态的角色分工。

参考链接:https://rikiyat.github.io/blog/weight-space-programming/

上车,带你看遍全球 AI 顶会精华

可独家畅览:

专家演讲PPT

大会报告全文

热门论文解读

学术新星访谈

//

推荐阅读

别再盯着 800 亿美元亏损了,Anthropic 的 AI 造 AI 已推进到 26%!


DeepSeek 又一开源重器!深挖 DeepInfer 究竟藏着多少技术狠活?

未经「AI科技评论」授权,严禁以任何方式在网页、论坛、社区进行转载!

公众号转载请先在「AI科技评论」后台留言取得授权,转载时需标注来源并插入本公众号名片。

【声明】内容源于网络
0
0
AI科技评论
聚焦AI前沿研究,关注AI工程落地。
内容 9026
粉丝 0
AI科技评论 聚焦AI前沿研究,关注AI工程落地。
总阅读301.5k
粉丝0
内容9.0k