跳到主要內容
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%,能耗與內存需求顯著降低。

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