用新型3D内存技术大幅扩展缓存,显著降低大模型推理能耗。
LLMET: Enabling Cross-Layer Evaluation of Emerging M3D Memories for Energy-Efficient LLM Serving

- 构建跨层仿真框架LLMET,评估新兴3D内存对大模型能效的影响。
- 在16K上下文下,缓存扩至1GB可使Llama3.1-70B预填充能耗降44%。
- 边缘设备上缓存从8MB增至256MB,解码能耗减少30%,适合能效敏感场景。
大语言模型(LLM)服务的能耗正成为规模化部署的关键挑战,受硬件功耗、散热限制及电价上涨驱动。芯片能耗的主要来源是有限片上缓存与片外高带宽内存(HBM)间的数据搬运。新兴的单片3D(M3D)集成技术可在逻辑芯片后端(BEOL)实现更大更密的片上缓存,为减少昂贵的片外通信带来新机遇。然而,持续扩大片上内存是否真能提升LLM服务能效尚不明确。为此,我们开发了经验证的跨层仿真框架LLMET,系统研究了大容量片上内存技术在多种模型、应用和平台上的影响。基于双NVIDIA A100 GPU设置的模拟显示,将L2缓存从40MB扩展至1GB,在16K上下文窗口下使Llama3.1-70B预填充阶段芯片能耗降低44%;在8x NVIDIA B200类平台上,将L2缓存从128MB扩展至4GB,预填充能耗最高可节省24%;在边缘平台与工作负载中,缓存从8MB增至256MB时,解码能耗降低30%。这些结果凸显了超大容量片上内存在能效优化的潜力。
原文摘要 · Abstract (English)
The energy consumption of Large Language Model (LLM) serving is becoming a major system challenge as deployment scales, driven by hardware power and thermal constraints and rising electricity costs. A key contributor to chip energy dissipation is data movement between limited on-chip cache and off-chip High Bandwidth Memory (HBM). Meanwhile, emerging memory technologies such as monolithic 3D (M3D) integration of cache memories at the Back-End-Of-Line (BEOL) of logic chips enable larger and denser on-chip memories, creating new opportunities to reduce costly off-chip traffic. However, it remains unclear whether continuously scaling on-chip memory using emerging technologies can effectively improve the energy efficiency of LLM serving. To address this gap, we develop LLMET (LLM with Emerging Technology), a validated cross-layer simulation framework, and conduct a comprehensive study on the impact of large-capacity on-chip memory technologies across a broad range of models, applications and platforms. Utilizing M3D technology to expand the L2 cache from 40MB to 1GB yields a 44% reduction in chip energy during the Llama3.1-70B prefill phase with a 16K context window, based on LLMET simulation on a dual NVIDIA A100 GPU setup. On the 8x NVIDIA B200-like platform, extending the L2 cache from 128MB to 4GB saves the prefill energy by up to 24%. For the edge platform and workloads, the decode energy saving reaches 30% when increasing the 8MB cache size to 256MB. These results highlight the promise of ultra-large on-chip memories for energy-efficient LLM serving systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。