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

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。