近日,中國科學院自動化研究所李國齊、徐波團隊在發表原創內生復雜性理論系列論文的工作基礎上,與沐曦MetaX合作,打造了類腦脈沖大模型“瞬悉1.0”(SpikingBrain-1.0),在國產千卡GPU算力平臺上完成全流程訓練和推理,實現了大模型在超長序列推理上數量級的效率和速度提升,展示了構建國產自主可控的新型非Transformer大模型架構生態的可行性。研究團隊開源了SpikingBrain-1.0-7B模型并開放SpikingBrain-1.0-76B測試網址,同步公開了經工業界大規模驗證的類腦脈沖大模型SpikingBrain-1.0中英文技術報告。
基于內生復雜性的非Transformer架構
當前,基于Transformer架構的大模型在Scaling law驅動下,通過增加網絡規模、算力資源和數據量提升模型智能水平,但模型的基本計算單元為簡單的點神經元模型,我們將此路徑稱為“基于外生復雜性”的通用智能實現方法。Transformer架構的固有缺點訓練時開銷隨序列長度呈平方級增長以及推理時顯存占用也隨序列長度線性增加,構成了資源消耗的主要瓶頸,導致其處理超長序列的能力受限。
研發團隊借鑒大腦神經元內部復雜工作機制提出“基于內生復雜性”的大模型構架方式,打造類腦脈沖大模型“瞬悉1.0”( SpikingBrain-1.0),在理論上建立了脈沖神經元內生動力學與線性注意力模型之間的聯系,揭示了現有線性注意力機制是樹突計算的特殊簡化形式,從而清晰地展示了一條不斷提升模型復雜度和性能的新型可行路徑。研發團隊進而構建并開源了基于脈沖神經元、具有線性(SpikingBrain-1.0-7B)及混合線性復雜度(SpikingBrain-1.0-76B,激活參數量12B)的新型類腦基礎模型,開發了面向國產GPU(沐曦MetaX曦云C550)集群高效訓練和推理框架、Triton 算子庫、模型并行策略以及集群通信原語。
兼具效率與性能,實現多個核心突破
SpikingBrain-1.0在多個核心性能上實現突破。
第一,極低數據量上的高效訓練:訓練階段具有線性或近線性復雜度,顯著提升長序列訓練效率,并能依托高效轉換訓練范式,以約為主流大模型2%的預訓練數據量實現與眾多開源Transformer模型在多任務語言理解(MMLU),中文多任務語言理解(CMMLU、Ceval), 常識推理能力(ARC、HS)任務上相媲美的性能。
第二,推理效率的數量級提升:推理階段結合脈沖神經元事件驅動特性,SpikingBrain具有常數或部分層常數級別的復雜度和存儲開銷,SpikingBrain-7B模型在100萬Token長度下TTFT(生成第一個Token所需時間)相比Transformer架構加速達到26.5倍,400萬Token長度下加速超過100倍。同時在手機CPU端64k-128k-256k長度下較Llama3.2的同規模模型Decoding速度提升4.04x-7.52x-15.39x,在超長序列處理能力上展現出數量級的效率和速度提升。
第三,國產自主可控類腦大模型生態的構建:SpikingBrain適配了面向國產GPU集群的高效訓練和推理框架、Triton算子庫、模型并行策略以及集群通信原語,表明了構建國產自主可控的新型非Transformer大模型架構生態的可行性。
第四,基于動態閾值脈沖化的多尺度稀疏機制:設計細粒度的兩階段動態閾值脈沖化策略,結合粗粒度的混合專家模型(MoE)方案,在7B模型上實現了超過69.15%的稀疏度,長序脈沖占比約1.85%,為低功耗的類腦大模型運行提供有力支撐。
這是我國首次提出大規模類腦線性基礎模型架構、并首次在國產GPU算力集群上構建類腦脈沖大模型的訓練和推理框架。提出的模型解決了脈沖驅動限制下的大規模類腦模型性能退化問題,其超長序列處理能力在法律/醫學文檔分析、復雜多智能體模擬、高能粒子物理實驗、DNA序列分析、分子動力學軌跡等超長序列任務建模場景中具有顯著的潛在效率優勢。本次發布的大模型為新一代人工智能發展提供了非Transformer架構的新技術路線,并將啟迪更低功耗的下一代神經形態計算理論和芯片設計。
關于沐曦
沐曦致力于自主研發全棧高性能GPU芯片及計算平臺,為智算、通用計算、云渲染等前沿領域提供高能效、高通用性的算力支撐,助力數字經濟發展。
-
開源
+關注
關注
3文章
4207瀏覽量
46143 -
沐曦
+關注
關注
1文章
80瀏覽量
1811 -
大模型
+關注
關注
2文章
3650瀏覽量
5184
原文標題:基于內生復雜性的類腦脈沖大模型“瞬悉1.0”問世
文章出處:【微信號:沐曦MetaX,微信公眾號:沐曦MetaX】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
合力泰到訪中國科學院福建物質結構研究所調研交流
思必馳參與國家重大項目順利通過中期驗收
時識科技與江西農科院成立智慧畜牧類腦計算聯合實驗室
中國科學院自動化研究所一行蒞臨智行者考察調研
潤和軟件與中國科學院廣州生物醫藥與健康研究院達成戰略合作
中國科學院院士王曦,遞補為中央委員
首款全國產通用GPU芯片發布 沐曦集成推出曦云C600
華為、中國科學院計算技術研究所聯合開發論文獲USENIX收錄
中國科學院自動化研究所攜手中科曙光打造高性能工具鏈解決方案
中科曙光亮相第十屆科學數據大會
中國科學院沈陽自動化研究所:研究基于石墨烯/PDMS封裝的醫用膠帶柔性傳感器,用于水下機器人運動檢測
恭賀 | 晟鵬創始人成會明院士當選歐洲科學院院士!
中國科學院自動化研究所與沐曦合作打造類腦脈沖大模型“瞬悉1.0”
評論