跳到主要內容
AI News HubLIVE
更多
僅摘要

更稀疏、更快、更輕的Transformer語言模型

Sakana AI與輝達合作,引入了新的稀疏資料結構與GPU核心,利用非結構化稀疏性提升大語言模型(LLM)的推理與訓練效率。該工作將在ICML 2026上展示。透過L1正則化,ReLU啟用的LLM可實現超過95%的稀疏性,且下游效能幾乎無損。基於自研的TwELL格式與融合多重矩陣乘法的CUDA核心,在H100 GPU上實現了批處理推理與訓練超過20%的加速,同時降低了能耗與記憶體需求。

來源Hacker News AI作者: hardmaru
更稀疏、更快、更輕的Transformer語言模型
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明

本文僅提供摘要

根據來源內容權限,本站僅提供編輯摘要、要點和原文連結。

展開要點與分析

文章情報

工程師進階

要點

  • Sakana AI與輝達合作,提出TwELL(Tile-wise ELLPACK)稀疏打包格式,適配分塊矩陣乘法核心。
  • 開發了融合多重矩陣乘法的定製CUDA核心,最大化吞吐量並壓縮儲存開銷。
  • L1正則化誘導的高稀疏性(>95%)對LLM下游效能影響極小,並轉化為實際加速。
  • 在H100 GPU上,批處理推理與訓練速度提升超過20%,能耗與記憶體需求顯著降低。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。