arXiv:2410.15977cs.AI2024-10TPAMI被引 8

用新型存算一体架构,让大模型在忆阻器芯片上高效运行。

Enabling Energy-Efficient Deployment of Large Language Models on Memristor Crossbar: A Synergy of Large and Small

  • 设计新架构支持大模型的非权重乘法与非线性运算。
  • 部署BERT_Large时精度损失可忽略,能效比提升69%。
  • 适合追求低功耗、高密度部署的大模型应用场景。

大型语言模型(LLMs)因在多个领域的应用前景备受关注,但其规模增长带来训练与部署的计算需求激增。忆阻器交叉阵列因其体积小、能效高,在计算机视觉模型中表现优异,且相比传统存储技术具有更高密度,适合应对大模型的极端规模。然而,将大模型部署在忆阻器交叉阵列上面临三大挑战:模型尺寸已超出先进忆阻芯片能力;多头注意力块中的非权重固定乘法无法被传统忆阻阵列支持;忆阻阵列擅长线性运算,难以执行大模型中的softmax和层归一化等复杂非线性操作。为此,我们提出一种新型忆阻交叉阵列架构,可在单芯片或单封装内实现当前顶尖大模型的部署,避免片外通信带来的能量与时间开销。在BERT_Large上的测试显示精度损失可忽略。相比传统忆阻阵列,本架构面积开销降低39倍,能耗降低18倍;相比现代TPU/GPU系统,面积-延迟积降低至少68倍,能耗减少69%。

原文摘要 · Abstract (English)

Large language models (LLMs) have garnered substantial attention due to their promising applications in diverse domains. Nevertheless, the increasing size of LLMs comes with a significant surge in the computational requirements for training and deployment. Memristor crossbars have emerged as a promising solution, which demonstrated a small footprint and remarkably high energy efficiency in computer vision (CV) models. Memristors possess higher density compared to conventional memory technologies, making them highly suitable for effectively managing the extreme model size associated with LLMs. However, deploying LLMs on memristor crossbars faces three major challenges. Firstly, the size of LLMs increases rapidly, already surpassing the capabilities of state-of-the-art memristor chips. Secondly, LLMs often incorporate multi-head attention blocks, which involve non-weight stationary multiplications that traditional memristor crossbars cannot support. Third, while memristor crossbars excel at performing linear operations, they are not capable of executing complex nonlinear operations in LLM such as softmax and layer normalization. To address these challenges, we present a novel architecture for the memristor crossbar that enables the deployment of state-of-the-art LLM on a single chip or package, eliminating the energy and time inefficiencies associated with off-chip communication. Our testing on BERT_Large showed negligible accuracy loss. Compared to traditional memristor crossbars, our architecture achieves enhancements of up to 39X in area overhead and 18X in energy consumption. Compared to modern TPU/GPU systems, our architecture demonstrates at least a 68X reduction in the area-delay product and a significant 69% energy consumption reduction.

大模型部署忆阻器存算一体能效优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。