MLA(Multi-head Latent Attention多頭潛在注意力架構)憑借低秩壓縮KV緩存設計,實現長上下文推理的顯存與帶寬雙重優化,其有效性已在 DeepSeek-V2 等模型中得到了驗證。但主流預訓練模型如LLaMA、Qwen千問等多基于GQA構建,企業已投入大量工程優化,復用MLA需重訓,成本極高。為此,在北京大學 鯤鵬昇騰科教創新卓越中心的算力支持下,北京大學人工智能研究院助理教授張牧涵團隊提出TransMLA轉化框架,實現了無需重訓便可將主流模型向MLA遷移。

TransMLA針對GQA向MLA遷移的核心痛點,實現四大技術模塊的精準破局:一是GQA→MLA結構映射,破解分組KV頭與MLA單頭潛在表示不兼容問題,通過特定的線性變換,將GQA分組后的K、V向量投影/融合為單一的低秩潛在表示,并配備上投影矩陣保障KV信息精準恢復,筑牢遷移基礎;二是提出RoRoPE技術方案,通過創新適配讓位置編碼順暢融入低秩壓縮流程,解決了直接對RoPE應用PCA等通用降維方法可能導致的位置信息損失或模型性能下降問題;三是通過將RoPE中相鄰頻率的旋轉維度進行折疊(Folding)與融合,在降低參數量的同時,更高效地集中和保留關鍵的位置信息,從而維持模型在長序列下的語義理解能力;四是通過均衡Key和Value矩陣在壓縮前的范數分布,提升聯合壓縮(如PCA)的數值穩定性,減少信息損失。
在技術落地過程中,昇騰發揮了關鍵支撐作用。其高效并行計算架構滿足結構映射模塊的多任務協同處理需求,保障了架構遷移效率;其優化的存儲與緩存體系,為 FreqFold 的頻率信息處理、BKV-PCA 的范數均衡提供穩定硬件基礎,有效提升 KV 壓縮的穩定性與資源利用效率,助力 TransMLA 核心技術平穩落地。
經過實驗顯示TransMLA的轉換過程性能優勢明顯,裁剪LLaMA-2-7B模型68.75%的KV緩存后無需訓練,核心性能僅輕微損失,在32K序列長度、FP16精度下,基于昇騰平臺的推理速度較之于業界主流GPU平臺有顯著提升。依托開放的生態資源,昇騰已推動TransMLA穩定支持主流模型部署并將集成至vLLM/SGLang等高性能推理框架生態,便于用戶部署,大幅降低企業落地適配成本。
TransMLA與昇騰的協同創新,打通了主流模型與MLA架構鴻溝,充分發揮昇騰生態優勢。昇騰全鏈路支持實現TransMLA“零重訓、低損失”目標,保留模型參數優勢,降低企業基于昇騰的升級門檻。這一軟硬件協同典范,為長上下文推理提供昇騰生態解決方案,推動自主計算與前沿AI融合,彰顯昇騰核心引領作用,為大模型產業依托自主硬件降本增效提供可行路徑。
請輸入留言內容
我要留言
請輸入留言內容
我要留言
確定留言
|
|
2026年人力資源服務協同發展座談會在昆明成功舉辦2026年2月26日,由東方睿智(北京)國際人力資源有限公司主辦的“鏈接產業·賦能人才——2026年人力資源..
|
|
|
|
深化中越職教交流 以標準服務助力職教出海在WorldSkillsShanghai2026CompetitionPreparationWeek舉辦期間,越南教育與培訓部(MinistryofEducationan..
|
|
|
|
深入推進集成電路產業布局,學大教育戰略投資天數智芯與啟芯領航近日,A股上市企業學大(廈門)教育科技集團股份有限公司(000526.SZ)(以下簡稱“學大教育”“公司”)..
|
|
|
|
趣味賦能素質教育,元蘿卜AI機器人扎根多校課堂人工智能技術日益成熟,其應用場景已延伸至中小學課堂。在政策指引與校內實踐的共同促進下,商湯科技旗下..
|
|
|
|
民進中央調研組至句象書店考察全民閱讀融合家校社協同育人的創新實踐1月28日,民進中央調研組在滬考察調研期間,專程前往民進會員企業學大教育集團旗下句象書店上海花木時光..
|
|
|
|
易才集團全新IP形象易小才亮相,重構人力資源服務新體驗1月26日,易才集團正式對外發布全新IP形象“易小才”。作為易才集團AI戰略布局的具象化載體之一與品牌年..
|
|
|
|
陳一丹百萬獎教獎學金激勵,武漢學院為學子搭特別之橋1月14日,由騰訊主要創始人、武漢學院創辦人陳一丹設立的第十六屆“陳一丹獎教獎學金”頒獎。本屆共有20..
|
|
|
|
科技賦能法考新篇章,瑞達教育2025考季年度盛典順利收官2026年1月10日,榮耀加冕·逐夢同行——瑞達教育2025考季年度盛典在京舉行。來自全國各地的優秀學員代表..
|
|
|
|
立邦成為上海市首批“一試雙證”職業技能認定企業1月8日,上海市人力資源和社會保障局發布《關于公布上海市首批“一試雙證”職業技能比照認定目錄的通知》..
|
|
|
|
正初體系化服務,賦能職教國際合作升級近日,江蘇省及南京市兩級職業教育領域年度重要工作會議相繼召開,系統總結過往成果,共繪未來發展藍圖。江..
|
|
|
首頁
|
紅人堂
|
發布
|
我發的
|
更多
|