arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2005.04680cs.DCcs.IRcs.LGcs.PF
收藏

基于CPU集群架构的深度学习推荐系统训练优化

Optimizing Deep Learning Recommender Systems' Training On CPU Cluster Architectures

Intel Technology India Private Ltd.(英特尔技术印度私人有限公司);Intel Corporation(英特尔公司);Intel China Co. Ltd.(英特尔中国有限公司);Intel Russia(英特尔俄罗斯公司)

Dhiraj Kalamkar, Evangelos Georganas, Sudarshan Srinivasan, Jianping Chen, Mikhail Shiryaev, Alexander Heinecke

摘要:

在过去的两年里,许多研究人员的目标是从HPC系统中挤出最后一点性能来执行AI任务。通常,这种讨论是在如何快速训练ResNet 50的背景下进行的。不幸的是,ResNet 50在2020年不再是一个代表性的工作负载。因此,我们专注于推荐系统,它占了云计算中心的大部分AI周期。更具体地说,我们专注于Facebook的DLRM基准。通过使其能够在为HPC定制的最新CPU硬件和软件上运行,与参考CPU实现相比,我们能够在单个插槽上实现超过两个数量级的性能提升(110倍),以及高达64个插槽的高扩展效率,同时适应超大型数据集。本文讨论了DLRM中的各种运营商的优化技术,以及这些不同的运营商强调系统的组成部分。所提出的技术适用于提出与DLRM相同的缩放挑战/特性的更广泛的DL工作负载集合。

英文摘要:

During the last two years, the goal of many researchers has been to squeeze the last bit of performance out of HPC system for AI tasks. Often this discussion is held in the context of how fast ResNet50 can be trained. Unfortunately, ResNet50 is no longer a representative workload in 2020. Thus, we focus on Recommender Systems which account for most of the AI cycles in cloud computing centers. More specifically, we focus on Facebook's DLRM benchmark. By enabling it to run on latest CPU hardware and software tailored for HPC, we are able to achieve more than two-orders of magnitude improvement in performance (110x) on a single socket compared to the reference CPU implementation, and high scaling efficiency up to 64 sockets, while fitting ultra-large datasets. This paper discusses the optimization techniques for the various operators in DLRM and which component of the systems are stressed by these different operators. The presented techniques are applicable to a broader set of DL workloads that pose the same scaling challenges/characteristics as DLRM.

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

↑