網易首頁 > 網易號 > 正文 申請入駐

大模型又出新架構,面壁這次把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-04-20 08:45:09
女孩在行李箱上寫字,全網狂贊:這才是中國人刻在骨子里的浪漫!

女孩在行李箱上寫字,全網狂贊:這才是中國人刻在骨子里的浪漫!

中國藝術家
2026-04-14 05:23:31
你可以活多久?美國教授推出“壽命公式"

你可以活多久?美國教授推出“壽命公式"

霹靂炮
2026-04-18 23:28:44
民進黨吳崢狂言“我就是臺獨又怎樣”,被當場懟到啞口無言!

民進黨吳崢狂言“我就是臺獨又怎樣”,被當場懟到啞口無言!

荊楚寰宇文樞
2026-04-19 23:40:53
AI直接點名!卡扣斷了要花13萬換電池包,媒體集體默契地語焉不詳

AI直接點名!卡扣斷了要花13萬換電池包,媒體集體默契地語焉不詳

芭比衣櫥
2026-04-20 17:55:54
斯諾克最新戰報!16強誕生8席,威爾遜狂轟七連鞭逆轉,吳宜澤8-1

斯諾克最新戰報!16強誕生8席,威爾遜狂轟七連鞭逆轉,吳宜澤8-1

越嶺尋蹤
2026-04-21 05:19:16
假貨賣家把LV商標轉30度,AI怎么抓?

假貨賣家把LV商標轉30度,AI怎么抓?

碼上閑敘
2026-04-20 10:13:48
蕭旭岑爆國民黨“長年A錢”,醉翁之意不在酒

蕭旭岑爆國民黨“長年A錢”,醉翁之意不在酒

小陸搞笑日常
2026-04-21 00:28:16
王健林被逼債38億,妻子林寧被扒底朝天,怪不得王思聰能逍遙國外

王健林被逼債38億,妻子林寧被扒底朝天,怪不得王思聰能逍遙國外

云舟史策
2026-04-21 07:35:29
活該,設計院降薪降得領導都指揮不動員工了!

活該,設計院降薪降得領導都指揮不動員工了!

黯泉
2026-04-17 20:59:06
與孔令輝分手12年,被黑人托在肩上的馬蘇,終是為自己的風流買單

與孔令輝分手12年,被黑人托在肩上的馬蘇,終是為自己的風流買單

情感大頭說說
2026-04-18 21:06:00
“國產SUV”冠軍易主:新車型沖到了第一,月零售量上漲到24198輛

“國產SUV”冠軍易主:新車型沖到了第一,月零售量上漲到24198輛

柳先說
2026-04-18 18:29:28
天文學家發現:宇宙中98%的星系,已經永遠不可能被看見了

天文學家發現:宇宙中98%的星系,已經永遠不可能被看見了

觀察宇宙
2026-04-20 18:03:11
房子遭人強拆,因反抗坐3年牢!出獄后揚言:不贏官司就殺人!

房子遭人強拆,因反抗坐3年牢!出獄后揚言:不贏官司就殺人!

談史論天地
2026-04-20 12:16:17
珠海家長:自家孩子趴桌睡,憑啥先給別人捐躺椅?官方回應

珠海家長:自家孩子趴桌睡,憑啥先給別人捐躺椅?官方回應

南方都市報
2026-04-17 15:48:13
游泳男子世青紀錄更新,張展碩成唯一持有中國選手

游泳男子世青紀錄更新,張展碩成唯一持有中國選手

體娛一家親
2026-04-20 22:57:27
太尷尬了!9個姐妹聚餐,臨時倡議各邀請一男生,竟然無一人到場

太尷尬了!9個姐妹聚餐,臨時倡議各邀請一男生,竟然無一人到場

火山詩話
2026-04-19 08:06:38
16強出5!斯諾克世錦賽戰報速遞:肖國棟10-6晉級,馬叔10-4大勝

16強出5!斯諾克世錦賽戰報速遞:肖國棟10-6晉級,馬叔10-4大勝

越嶺尋蹤
2026-04-20 06:04:11
失誤!“國安棄帥”在中甲執教保級隊高居第二:準備沖超!

失誤!“國安棄帥”在中甲執教保級隊高居第二:準備沖超!

邱澤云
2026-04-20 16:07:09
十大起義總指揮都是誰

十大起義總指揮都是誰

祁州校尉
2026-04-17 11:00:28
2026-04-21 08:31:00
卡爾的AI沃茨 incentive-icons
卡爾的AI沃茨
前大廠算法工程師,3家科技公司技術總監|致力打造最系統的Al學習體系,讓1萬人通過Al提高生產力
250文章數 126關注度
往期回顧 全部

科技要聞

重磅官宣:庫克卸任,特努斯接任蘋果CEO

頭條要聞

特朗普兩天三次反轉 伊朗學習特朗普玩起"極限施壓"

頭條要聞

特朗普兩天三次反轉 伊朗學習特朗普玩起"極限施壓"

體育要聞

阿森納已拼盡全力,但你早干嘛去了...

娛樂要聞

《八千里路云和月》田家泰暗殺

財經要聞

利潤暴跌7成,字節到底在做什么

汽車要聞

把天門山搬進廠?開仰望U8沖上45度坡的那刻 我腿軟了

態度原創

教育
家居
旅游
健康
公開課

教育要聞

加權費馬點模型,一個視頻學明白!

家居要聞

自然慢調 慢享時光

旅游要聞

閻錫山故居:一座都督府,半部民國史

干細胞抗衰4大誤區,90%的人都中招

公開課

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

無障礙瀏覽 進入關懷版