arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.04032cs.ARcs.AI
收藏

ProactivePIM: 通过PIM加速基于权重共享的嵌入层以实现可扩展的推荐系统

ProactivePIM: Accelerating Weight-Sharing Embedding Layer with PIM for Scalable Recommendation System

Youngsuk Kim, Junghwan Lim, Hyuk-Jae Lee, Chae Eun Rhee

AI总结:

ProactivePIM通过PIM技术加速权重共享算法,提升推荐系统性能。

摘要:

尽管基于深度学习的个性化推荐系统能够提供高质量的推荐,但它们会消耗数据中心的资源。主要瓶颈是嵌入层,由于其对嵌入的稀疏、不规则访问模式,该层非常内存密集。最近的近内存处理(NMP)和内存计算(PIM)架构通过在内存内利用并行性来解决这些问题。然而,随着模型规模逐年增加并可能超过服务器容量,单节点服务器上的推理由来具有挑战性,需要整合模型压缩。已经提出了各种算法来减少模型大小,但这些方法会增加内存访问和CPU-PIM通信的成本。我们提出了ProactivePIM,这是一个专为权重共享算法设计的PIM系统,该算法家族将嵌入表分解为紧凑的子表,如QR-trick和TT-Rec。我们的分析显示,使用权重共享算法执行嵌入层会增加内存访问并导致CPU-PIM通信。我们还发现这些算法具有独特的数据局部性特征,我们将其命名为intra-GnR局部性。ProactivePIM通过利用异构的HBM-DIMM内存架构,并在HBM内集成了两级PIM系统,即基础芯片PIM(bd-PIM)和银行组PIM(bg-PIM),以加速权重共享算法。为了进一步加速,ProactivePIM将具有高intra-GnR局部性的嵌入预取到bg-PIM内的SRAM缓存中,并通过在银行组之间复制目标子表来消除CPU-PIM通信。通过额外的优化技术,我们的设计有效加速了权重共享算法,与基线架构相比,在QR-trick和TT-Rec上分别实现了2.22倍和2.15倍的加速。

英文摘要:

Although deep learning-based personalized recommendation systems provide qualified recommendations, they strain data center resources. The main bottleneck is the embedding layer, which is highly memory-intensive due to its sparse, irregular access patterns to embeddings. Recent near-memory processing (NMP) and processing-in-memory (PIM) architectures have addressed these issues by exploiting parallelism within memory. However, as model sizes increase year by year and can exceed server capacity, inference on single-node servers becomes challenging, necessitating the integration of model compression. Various algorithms have been proposed for model size reduction, but they come at the cost of increased memory access and CPU-PIM communication. We present ProactivePIM, a PIM system tailored for weight-sharing algorithms, a family of compression methods that decompose an embedding table into compact subtables, such as QR-trick and TT-Rec. Our analysis shows that embedding layer execution with weight-sharing algorithms increases memory access and incurs CPU-PIM communication. We also find that these algorithms exhibit unique data locality characteristics, which we name intra-GnR locality. ProactivePIM accelerates weight-sharing algorithms by utilizing a heterogeneous HBM-DIMM memory architecture with integration of a two-level PIM system of base-die PIM (bd-PIM) and bank-group PIM (bg-PIM) inside the HBM. To gain further speedup, ProactivePIM prefetches embeddings with high intra-GnR locality into an SRAM cache within bg-PIM and eliminates the CPU-PIM communication through duplication of target subtables across bank groups. With additional optimization techniques, our design effectively accelerates weight-sharing algorithms, achieving 2.22x and 2.15x speedup in QR-trick and TT-Rec, respectively, compared to the baseline architecture.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑