arXiv:2510.05245cs.ARcs.ET2025-10被引 22

用3D堆叠内存+近存计算,让专家模型推理快7倍还省电。

Stratum: System-Hardware Co-Design with Tiered Monolithic 3D-Stackable DRAM for Efficient MoE Serving

  • 用单片3D堆叠内存和近存处理,解决专家模型数据量大难题。
  • 在多个基准上实现8.29倍解码吞吐提升,能效提升7.66倍。
  • 适合需要高并发、低延迟部署大规模专家模型的系统设计者。

随着大语言模型不断发展,混合专家(MoE)架构已成为实现多种任务顶尖性能的主流设计。MoE模型通过稀疏门控机制,仅激活少量专家子网络,可在保持较小模型推理开销的同时实现千亿参数规模。然而,这种架构在硬件部署中面临巨大数据量挑战。为此,我们提出Stratum,一种系统-硬件协同设计方法,融合单片3D堆叠内存(Mono3D DRAM)、近存处理(NMP)与GPU加速。逻辑层与Mono3D DRAM晶圆通过混合键合连接,而Mono3D DRAM堆栈与GPU则通过硅通孔互连。得益于单片结构带来的密集垂直互联密度,Mono3D DRAM内部带宽高于HBM,支持更高性能的近存处理。此外,为缓解因垂直方向激进缩放导致的延迟差异,我们构建内部内存层级,根据访问频率分配数据,并基于主题级专家使用预测优化近存处理吞吐。相比传统GPU基线,Stratum在多个基准测试中实现最高8.29倍解码吞吐提升和7.66倍能效改善。

原文摘要 · Abstract (English)

As Large Language Models (LLMs) continue to evolve, Mixture of Experts (MoE) architecture has emerged as a prevailing design for achieving state-of-the-art performance across a wide range of tasks. MoE models use sparse gating to activate only a handful of expert sub-networks per input, achieving billion-parameter capacity with inference costs akin to much smaller models. However, such models often pose challenges for hardware deployment due to the massive data volume introduced by the MoE layers. To address the challenges of serving MoE models, we propose Stratum, a system-hardware co-design approach that combines the novel memory technology Monolithic 3D-Stackable DRAM (Mono3D DRAM), near-memory processing (NMP), and GPU acceleration. The logic and Mono3D DRAM dies are connected through hybrid bonding, whereas the Mono3D DRAM stack and GPU are interconnected via silicon interposer. Mono3D DRAM offers higher internal bandwidth than HBM thanks to the dense vertical interconnect pitch enabled by its monolithic structure, which supports implementations of higher-performance near-memory processing. Furthermore, we tackle the latency differences introduced by aggressive vertical scaling of Mono3D DRAM along the z-dimension by constructing internal memory tiers and assigning data across layers based on access likelihood, guided by topic-based expert usage prediction to boost NMP throughput. The Stratum system achieves up to 8.29x improvement in decoding throughput and 7.66x better energy efficiency across various benchmarks compared to GPU baselines.

MoE3D堆叠近存计算高效推理

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