Apr, 2023

HEAT: 面向 CPU 的协同过滤推荐高效易用的训练系统

TL;DR本文针对 Collaborative filtering 中的 state-of-the-art 方法 SimpleX 在 multi-core CPUs 表现欠佳的问题,提出了一种名为 HEAT 的高效 CF 训练系统,通过瓦片 embedding 矩阵、优化 SGD 方法和重复利用前向阶段结果的方式有效提升了计算性能,并在 5 个常用数据集上进行了评估。结果表明,在 x86 - 和 ARM - 系 CPU 上,HEAT 相对现有 CPU 解决方案提升了高达 65.3 倍的速度,在使用 NVIDIA V100 GPU 的云计算方案上,也取得了 4.8 倍的加速和 7.9 倍的成本降低。