破解循环Transformer计算冗余,阿里两篇论文提供新方案

破解循环Transformer计算冗余,阿里两篇论文提供新方案

AI 资讯 来源:新闻/公众号 发布时间:2026-09-06 12:54:11 预计阅读 4 分钟

随着GPT-6 Astra的发布,循环深度(recurrent depth)技术迅速成为业界焦点。这项技术通过让同一组Transformer层反复运行,在不增加同等规模参数的情况下加深计算,看似为提升模型性能开辟了新路径。然而,其内部过程难以监测,引发了AI安全专家的质疑,OpenAI首席科学家Jakub Pachocki已回应将发文解释。

循环Transformer的核心思路是用计算深度换取参数规模,但学界早已发现,它在实际表现中往往打不过普通Transformer。问题根源在于“计算冗余”——当循环次数增加后,后续计算带来的增量收益越来越小。阿里团队在11个月前发表的两篇论文,正是针对这一痛点展开的。

动态记忆槽:让信息存取更有序

第一篇论文MeSH(已被ICLR 2026接收)提出了一种动态存取的“Memory Buffer”记忆槽机制。该机制配备了Write Router和Read Router两个“管理员”,分别负责决定每轮新结果如何分摊到各槽位,以及下一轮如何读取和组合这些信息。这种设计旨在优化循环过程中信息的存储与利用。

在Pythia-1.4B级别的实验中,循环结构通过权重共享减少了约33%的非嵌入参数,而加入MeSH后,零样本下游平均准确率从49.50%提升至50.56%。更令人惊喜的是,新增的路由器参数仅约0.005%,额外计算开销约0.014%,几乎可以忽略不计。

团队通过深入分析,发现了三个“摸鱼证据”:后续循环对hidden state的更新迅速缩小;相邻轮次的表示高度相似;表示逐渐挤入低维空间。这些现象指向两个根本原因:“计算无分化”——共享网络缺乏轮次信息,难以分工协作;以及“信息过载”——同一个hidden state需要同时保存原始输入、历史结果和当前计算,负担过重。

从粗到细:调整每轮序列长度

第二篇论文SpiralFormer(已被EMNLP 2026接收)则从另一个角度切入,将每轮循环的序列长度变为可调。它采用从粗到细的缩放方式:先从原序列的1/8长度开始,依次扩展到1/4、1/2,最后回到完整序列。这种渐进式的处理方式,让模型在早期循环中聚焦于全局信息,后期再细化局部细节。

在Pythia-1.4B级别的实验中,SpiralFormer的参数量与普通Transformer基本相同,但计算量从14.08T FLOPs降至13.13T,5-shot下游平均准确率从51.93%提升至54.37%。这意味着,在几乎不增加参数的前提下,模型不仅算得更少,还表现得更好。

两篇论文分别从信息管理和计算粒度入手,试图解决循环Transformer的计算冗余问题。MeSH通过精细化的记忆管理,让信息存取更有条理;SpiralFormer则通过动态调整序列长度,优化了计算资源的分配。这些探索为下一代高效大模型架构提供了候选方案。

循环深度技术的前景令人期待,但如何克服其固有的冗余问题,仍是通往实用化的关键一步。阿里团队的这两项研究,或许正是打开这扇门的钥匙。

标签: AI 资讯 AI

更多推荐阅读

黄仁勋放话断供不负责任AI客户,算力治理争论升温

黄仁勋放话断供不负责任AI客户,算力治理争论升温

10月14日,英伟达CEO黄仁勋接受彭博社采访时被问及,若有客户不负责任地开发AI,英伟达是否会停止供货。他回答:“这是一个假设,但答案是‘会的’。”让算力供应方充当AI治理关口,研究界已讨论多年。2023年,研究者Janet Egan和Lennart Heim提出算力提供方可参照金融业“了解你的客户”规则,核实大规模算力用户身份和用途,保存记录并上报可疑情况。英伟达占AI训练芯片市场约九成份额,...

2026-10-21 08:20:29
具身智能新玩家YITA伊塔动力融资破亿

具身智能新玩家YITA伊塔动力融资破亿

ZP独家获悉,具身智能操作平台与动力系统公司「YITA Dynamics伊塔动力」宣布完成超亿元种子和种子+轮融资,由经纬创投、商汤国香资本、建银国际、L2F光源创业者基金、银河通用、傅利叶智能联合投资,光源资本担任孵化方和独家财务顾问。 YITA取自希腊字母Eta(η),在新一代强化学习与数据驱动控制范式中代表学习率,在动力、传动与能源系统中代表效率。公司旨在以智能学习效率和物理传动效率为核心...

2026-10-21 09:57:24
图灵奖得主同台,清华系黑马亮出物理AI王牌

图灵奖得主同台,清华系黑马亮出物理AI王牌

10月20日,首届2026物理AI与世界模型大会在南京国际博览中心开幕。图灵奖得主杨立昆与姚期智视频连线,沈向洋、于海斌两位院士发表主旨演讲。成立仅三年、累计完成9轮融资的清华系公司千诀科技,由创始人高海川在开幕式主论坛发布面向真实物理世界的具身智能大脑底座——Polibrain预测式世界模型。其技术底牌源于千诀联合清华大学团队围绕“隐变量可辨识性”发表的公开成果,比LeCun团队同类工作早约一年...

2026-10-21 08:01:48
OpenAI急推300亿美元融资,抢在Anthropic IPO前锁定金主

OpenAI急推300亿美元融资,抢在Anthropic IPO前锁定金主

OpenAI正加速推进约300亿美元融资,目标估值超1.4万亿美元,试图在竞争对手Anthropic预计11月中旬IPO前锁定投资者。软银、阿布扎比MGX及其他现有投资方正洽谈参与。 两家前沿AI实验室正争夺同一批投资者。OpenAI上周与多头基金和主权财富基金会谈,Anthropic同期举办投资者日活动。知情人士称,OpenAI希望在Anthropic财务披露前完成融资,避免被直接比较。有律师...

2026-10-21 08:53:21
DeepSeek开源自家印钞机级推理框架DeepInfer

DeepSeek开源自家印钞机级推理框架DeepInfer

昨天,DeepSeek 开源了支撑自家百亿级调用的云服务推理框架 DeepInfer,以及生产级分词组件 deepseek-recipe,均采用 Apache 2.0 协议。官方表示,未来新模型的推理代码和性能优化也会持续在 DeepInfer 中开源。 基于 DeepInfer,DeepSeek 单一集群服务吞吐已达每天 155 万亿 Tokens,按最低价格计算,整体收入约每卡每小时 38 ...

2026-10-21 09:59:40
拒绝上下文崇拜的AI墨水屏Flowtica再融数千万

拒绝上下文崇拜的AI墨水屏Flowtica再融数千万

AI 墨水屏赛道正在起势。AI 硬件品牌 Flowtica 完成数千万人民币新一轮融资,由祥峰资本领投,源码资本、五源资本及清水湾基金等老股东跟投。其 AI 墨水屏产品 Verso 已于 9 月在 IFA 宣发,并正式在海外开启定金预售,将于 12 月交付首批产品。 Verso 是一块 78 克、6.5 毫米厚的磁吸墨水屏,贴在手机背面。开会时按一下开始录音,实时转写、实时总结,结束录音时自动生...

2026-10-21 08:02:57
订阅
关注

扫码关注公众号「60秒看懂AI」

扫码关注公众号 60秒看懂AI

微信搜索「60秒看懂AI」也可关注

顶部