arXiv:2510.02675cs.ARcs.AI2025-10

HALO加速器通过异构内存设计,显著提升低批量长上下文大模型推理速度。

HALO: Memory-Centric Heterogeneous Accelerator with 2.5D Integration for Low-Batch LLM Inference

  • 采用2.5D集成的存算一体架构,分阶段适配预填充和解码的不同需求。
  • 在LLaMA-2 7B和Qwen3 8B上实现相比AttAcc最高18倍、相比CENT 2.5倍加速。
  • 适合需要低延迟交互的大模型应用,如聊天机器人和个性化助手。

大型语言模型(LLMs)的快速普及推动了对高效推理的需求,尤其在对话机器人和个性化助手等低延迟应用场景中。与传统深度神经网络不同,LLM推理分为两个阶段:预填充阶段并行处理完整输入序列,解码阶段则逐个生成令牌。这两个阶段计算与内存需求差异显著,给加速器设计带来挑战。以往工作主要针对高批量或短上下文场景优化,而低批量、长上下文场景——这对交互式应用至关重要——仍被忽视。本文提出HALO,一种面向低批量LLM推理中预填充与解码阶段特性的异构内存中心加速器。HALO结合基于HBM的存内计算(CiD)与片上模拟存内计算(CiM),通过2.5D集成实现共封装。为提高硬件利用率,引入阶段感知映射策略:预填充阶段的计算密集型操作映射至CiM以利用其高吞吐矩阵乘能力;解码阶段的内存密集型操作由CiD执行,减少DRAM内数据移动。此外,分析了完全CiD与完全片上模拟CiM两种极端架构的性能权衡,凸显异构设计的必要性。在LLaMA-2 7B和Qwen3 8B模型上评估显示,映射至HALO的LLM相比AttAcc获得最高18倍几何平均加速,相比CENT(全CiD设计)提升2.5倍。

原文摘要 · Abstract (English)

The rapid adoption of Large Language Models (LLMs) has driven a growing demand for efficient inference, particularly in latency-sensitive applications such as chatbots and personalized assistants. Unlike traditional deep neural networks, LLM inference proceeds in two distinct phases: the prefill phase, which processes the full input sequence in parallel, and the decode phase, which generates tokens sequentially. These phases exhibit highly diverse compute and memory requirements, which makes accelerator design particularly challenging. Prior works have primarily been optimized for high-batch inference or evaluated only short input context lengths, leaving the low-batch and long context regime, which is critical for interactive applications, largely underexplored. We propose HALO, a heterogeneous memory centric accelerator designed for these unique challenges of prefill and decode phases in low-batch LLM inference. HALO integrates HBM based Compute-in-DRAM (CiD) with an on-chip analog Compute-in-Memory (CiM), co-packaged using 2.5D integration. To further improve the hardware utilization, we introduce a phase-aware mapping strategy that adapts to the distinct demands of the prefill and decode phases. Compute bound operations in the prefill phase are mapped to CiM to exploit its high throughput matrix multiplication capability, while memory-bound operations in the decode phase are executed on CiD to benefit from reduced data movement within DRAM. Additionally, we present an analysis of the performance tradeoffs of LLMs under two architectural extremes: a fully CiD and a fully on-chip analog CiM design to highlight the need for a heterogeneous design. We evaluate HALO on LLaMA-2 7B and Qwen3 8B models. Our experimental results show that LLMs mapped to HALO achieve up to 18x geometric mean speedup over AttAcc, an attention-optimized mapping and 2.5x over CENT, a fully CiD based mapping.

大模型推理存算一体2.5D集成异构加速

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