Facebook 开源深度学习推荐模型 DLRM

Facebook 宣布推出深度学习推荐模型（DLRM）的开源版本，这是一种最先进的个性化推荐 AI 模型，并且可用于生产环境中。该模型的实现可用于 Facebook 的 PyTorch、Facebook 的分布式学习框架 Caffe2 和 Glow C++。

推荐引擎在很大程度上决定着人们每天看到的内容，无论是 Facebook 等社交媒体网站、亚马逊等电子商务网站的内容，还是在 Xbox 首页的推荐游戏。就在上个月，亚马逊也将 AI 应用于 AWS 上的个性化购物推荐系统。

5 月底，20 多位 Facebook AI 研究人员在 arXiv 上发表了一篇论文，解释了 DLRM 模型如何使用映射分类数据的嵌入表进行表达，其中大部分计算由预测函数多层感知器（MLP）执行。论文详细介绍了 DLRM 模型，并与现有推荐模型进行比较，以充分展示其特性。

Facebook 人工智能研究（FAIR）已经将其大量工作开源，免费提供 DLRM 可以帮助更广泛的 AI 社区解决推荐引擎带来的挑战，比如利用神经网络将分类数据与某些更高级别的属性相关联。

DLRM 的制造商建议用该模型对推荐引擎的速度和准确性性能进行基准测试。用于实验和性能评估的 DLRM 基准测试是用 Python 编写的，支持随机和合成输入。

Facebook 研究科学家 Dheevatsa Mudigere 和 Maxim Naumov 在一篇博客文章中表示，将在未来公开分享优化 DLRM 系统的性能结果。