网易首页 > 网易号 > 正文 申请入驻

大模型又出新架构,面壁这次把Transformer上下文能力拉满了

0
分享至

我有点好奇一个问题,

让现阶段一个百万上下文的大模型读一本百万字的小说,需要推理吗?

可能大部分时候不需要,它需要的只是记住,记住第一章的主角在最后一章说了什么。

现在传统的Transformer架构只有一种处理方式,全靠硬算。它像一个过于认真的前排学霸,为了记住最后一章的内容,把前面九十九万字的内容跟每一个新生成的字都做一次对比计算。


这样会占大量的显存,计算量也会平方级增长。

这一周新的大模型也是不停刷新,MiniMax M2.5,GLM5,Gemini 3.1 Pro Preview,DeepSeek新版本(疑似V4),我都希望他们可以接入我的OpenClaw里面,做一个长对话长记忆的模型。

昨天,面壁还发了一篇新论文,提出了一个叫SALA(Sparse Attention-Linear Attention)的混合注意力架构。它给出了一个新观点,

该快的地方就得快,该准的地方必须准。


https://github.com/OpenBMB/MiniCPM/blob/main/docs/MiniCPM_SALA.pdf

在他们的SALA架构里,

一部分模块专门负责快,

另一部分模块专门负责准。

结果就是,基于SALA架构的模型,在端侧显卡上,第一次把百万的长文本推理,稳稳跑通了。


我们来看一个具体的例子,

模型在处理百万字小说时,内部发生了什么。

每生成一个新词,模型都要把它和前面所有词语的键值对,也就是KV Cache,全部计算一遍。计算复杂度是随着文本长度平方级增长的,

同时,这个KV Cache会像滚雪球一样,迅速吃掉你所有的显存。把上下文从一万字拉到一百万字,计算量不是涨一百倍,是飙升一万倍。

这就是为什么传统架构在长上下文任务面前,会同时撞上计算墙和显存墙。


这几十万字的上下文,大部分可能只是背景描述,真正关键的信息也许就那么几句。

但模型为了找到这几句关键信息,付出的代价是把所有内容都用最高精度过了一遍。

来看看SALA的几个关键的设计,

首先是两种模块的分工。

线性注意力,我们这里可以理解为负责准的模块,面壁选用的是Lightning Attention,挑出那些最关键的局部信息进行精细计算。稀疏注意力,则是负责快的模块,面壁选用的是InfLLM v2,会高效计算所有信息。

SALA就是把这两者结合了起来。整个模型里,75%的层是负责准的线性注意力,剩下的25%,是负责快的稀疏注意力。这个比例经过大量实验找出的效率与精度之间的平衡点。


在不使用任何额外技术(如YaRN)的前提下,MiniCPM-SALA 可以将上下文长度拓展至2048K。

怎么保证它们能好好合作,不是互相干扰呢?

SALA用了一个叫HyPE的混合位置编码策略。在线性层,它保留了RoPE,保证短文本能力不受影响。在稀疏层,它去掉了位置编码,避免了信息在长距离传递时的衰减问题。

模型在检索几万甚至几十万token之前的内容时,依然能保持高精度。


最后是模型怎么来的。

这里有一个核心问题,要训练一个全新的混合架构模型,成本超级高。

面壁没有从零开始,反而是提出了一个叫HALO的训练范式。简单来说,就是拿一个已经训练好的全注意力模型,比如MiniCPM-4.0,通过架构转换,把它的一部分层变成线性注意力,另一部分变成稀疏注意力,然后进行持续训练。


这种方式,就像是给一辆性能不错的汽车做改装升级,而不是重新设计一辆新车。它继承了原模型已经学到的所有知识和能力,只是让它学会了用一种更高效的方式去工作。相比从头训练,这个方法的成本直接降低了大约75%。

这个训练过程也很有讲究,分为架构转换,稳定训练,短衰减,长衰减和微调五个阶段。特别是在长衰减阶段,模型逐步把上下文长度从4K扩展到520K,让模型充分学习两种注意力机制协同。

我们来看效果数据。

这次面壁并没有用一堆榜单来证明自己,是出了真实场景下的性能数据,对比的是同等规模的全注意力模型Qwen3-8B,我挑几个关键数据,

在推理速度上,当上下文长度达到256K时,MiniCPM-SALA的速度是Qwen3-8B的3.5倍。

这个提升完全来自架构本身的优势。


在显存占用上,在RTX 5090这样的消费级显卡上,Qwen3-8B在上下文长度达到128K时,就会因为显存不足而崩溃。而MiniCPM-SALA可以稳稳地跑到1M,也就是一百万token的长度。


但我是个挑剔的人,

用了会丢信息的注意力,模型是不是没脑子了?

这也是这篇工作最有价值的部分。实验数据显示,MiniCPM-SALA在数学,代码,知识问答这些常规能力上,和同规模的全注意力模型基本持平,没有出现明显的性能折损。


传统的全注意力模型,在处理长文本时,它的注意力容量被大量消耗在维持局部依赖上,比如识别一个多词组成的人名。而SALA架构,把这些任务交给了更高效的模块,从而释放了稀疏注意力层的容量,让它们可以更专注于建立全局的,跨越超长距离的上下文联系。

为了推动这个架构落地,面壁联合了SGLang和NVIDIA,发起了一个稀疏算子加速大奖赛SOAR。


SALA虽然在架构上做好了,

但底层的计算算子,相比已经被优化到极致的FlashAttention,还有很大提升空间。

这个比赛就是邀请全球的开发者,一起来把SALA这台新引擎的性能,压榨到极限。

平时测评模型测多了,

都是Coding,Coding,Coding,

我很高兴看到还不断能有新的架构,

新的算法出现,

就算是DeepSeek这一年,

更新模型的同时也没停过公开自己的算法,

面壁过去这一年也是不停发端侧模型的工作,

我觉得就这速度都不需要五年十年,

可能过个两三年,

在小天才手表上也能跑个大大大模型。

@ 作者 / 卡尔

最后,感谢你看到这里如果喜欢这篇文章,不妨顺手给我们点赞|在看|转发|评论

如果想要第一时间收到推送,不妨给我个星标

如果你有更有趣的玩法,欢迎在评论区和我聊聊

更多的内容正在不断填坑中……


特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
悔到肠子发青!阿森纳放走当世第一神锋!实力碾压阿尔瓦雷斯

悔到肠子发青!阿森纳放走当世第一神锋!实力碾压阿尔瓦雷斯

澜归序
2026-07-04 05:19:50
女子带全家食材进海底捞,边吃边问丈夫丢不丢人,瞬间引爆全网!

女子带全家食材进海底捞,边吃边问丈夫丢不丢人,瞬间引爆全网!

天天热点见闻
2026-07-08 05:22:24
儿子高考672分大办升学宴,宴席当天校长急电:别办了,出大事了!

儿子高考672分大办升学宴,宴席当天校长急电:别办了,出大事了!

背包旅行
2026-07-07 19:05:02
终于有经济学家批评体制内退休金太高、加剧代际矛盾,评论区炸锅

终于有经济学家批评体制内退休金太高、加剧代际矛盾,评论区炸锅

慧翔百科
2026-06-23 08:47:02
韩红的麻烦越来越大了

韩红的麻烦越来越大了

大张的自留地
2026-07-06 12:55:59
万斯警告乌克兰不要反击,特朗普称将很快结束俄乌战争

万斯警告乌克兰不要反击,特朗普称将很快结束俄乌战争

山河路口
2026-07-07 11:16:29
演都不演了!广西捐款曝光,多明星陷入“逼捐”,遭殃的不止韩红

演都不演了!广西捐款曝光,多明星陷入“逼捐”,遭殃的不止韩红

生性洒脱
2026-07-08 14:30:22
校徽“撞脸”阿玛尼,浙大连夜抢注商标?相关人士回应:因多次被冒用,去年就已对该图案进行注册申请,与阿玛尼无关

校徽“撞脸”阿玛尼,浙大连夜抢注商标?相关人士回应:因多次被冒用,去年就已对该图案进行注册申请,与阿玛尼无关

大象新闻
2026-07-08 08:54:05
大码模特现实里到底多大?

大码模特现实里到底多大?

飛娱日记
2026-05-12 08:27:55
NBA 传闻:与 76 人达成交易前,杰伦·布朗曾否决了另外两潜在下家

NBA 传闻:与 76 人达成交易前,杰伦·布朗曾否决了另外两潜在下家

好火子
2026-07-09 01:43:15
英媒:国际足联就网红Speed遭种族歧视启动调查

英媒:国际足联就网红Speed遭种族歧视启动调查

懂球帝
2026-07-08 13:37:22
苹果泄密,华强北白忙一场!

苹果泄密,华强北白忙一场!

新浪财经
2026-07-08 20:08:36
霉霉世纪婚礼“俗不可耐“?明星排队吃自助,现场仿佛乡村爱情!

霉霉世纪婚礼“俗不可耐“?明星排队吃自助,现场仿佛乡村爱情!

新欧洲
2026-07-08 22:35:34
曼联5000万从切尔西签桑托斯,周四官宣!埃德森同时体检争当首签

曼联5000万从切尔西签桑托斯,周四官宣!埃德森同时体检争当首签

罗米的曼联博客
2026-07-09 07:28:34
含金量十足,8强中挪威队本届世界杯所遭遇对手平均排名最高

含金量十足,8强中挪威队本届世界杯所遭遇对手平均排名最高

懂球帝
2026-07-09 06:34:03
超强台风“巴威”已进入48小时警戒线!周末非必要不外出,市教育局提醒:暂停一切线下教育教学活动

超强台风“巴威”已进入48小时警戒线!周末非必要不外出,市教育局提醒:暂停一切线下教育教学活动

台州发布
2026-07-09 02:31:11
挪威队为了喂饱哈兰德,居然往世界杯空运了半吨三文鱼。

挪威队为了喂饱哈兰德,居然往世界杯空运了半吨三文鱼。

岁月有情1314
2026-07-08 14:19:26
“打扮成这样,你还挺自信?”开家长会发8条视频,旁观者忍不了

“打扮成这样,你还挺自信?”开家长会发8条视频,旁观者忍不了

世界圈
2026-07-07 09:08:51
翻车!以色列官方庆祝阿根廷获胜 阿网友怒斥:肮脏的种族灭绝者

翻车!以色列官方庆祝阿根廷获胜 阿网友怒斥:肮脏的种族灭绝者

念洲
2026-07-08 16:54:14
她跟教练学飞行,结果飞到一半教练直接跳飞机?!她只能硬着头皮自己把飞机开回来...

她跟教练学飞行,结果飞到一半教练直接跳飞机?!她只能硬着头皮自己把飞机开回来...

英国那些事儿
2026-07-09 02:16:01
2026-07-09 09:20:49
卡尔的AI沃茨 incentive-icons
卡尔的AI沃茨
前大厂算法工程师,3家科技公司技术总监|致力打造最系统的Al学习体系,让1万人通过Al提高生产力
292文章数 142关注度
往期回顾 全部

科技要闻

字节杀回来了!深度实测Seedream 5.0 Pro

头条要闻

龙卷风横扫湖北小镇2分钟 两人在室内被卷出屋外坠亡

头条要闻

龙卷风横扫湖北小镇2分钟 两人在室内被卷出屋外坠亡

体育要闻

阿根廷绝境重生,离不开这个罗!

娱乐要闻

鹿晗出轨?邓超出轨绯闻又被扒出

财经要闻

九部门:支持加快零售业创新发展

汽车要闻

MB.EA纯电架构/售价33.98万 全新奔驰纯电GLC鎏金版上市

态度原创

健康
教育
数码
家居
公开课

干细胞是人体的专属修理工吗?

教育要闻

今年河南高考生,刚过本科线,6所新增公办本科能不能捡漏?

数码要闻

暑期装机小心DDR4“消费陷阱”!锐龙5 9600X+单通道DDR5性价比完胜

家居要闻

2026建博会(广州) 公装联探展交流活动

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版