[1906.00091] Deep Learning Recommendation Model for Personalization and Recommendation Systems

With the advent of deep learning, neural network-based recommendation models have emerged as an important tool for tackling personalization and recommendation tasks. These networks differ significantly from other deep learning networks due to their need to handle categorical features and are not well studied or understood. In this paper, we develop a state-of-the-art deep learning recommendation model (DLRM) and provide its implementation in both PyTorch and Caffe2 frameworks. In addition, we design a specialized parallelization scheme utilizing model parallelism on the embedding tables to mitigate memory constraints while exploiting data parallelism to scale-out compute from the fully-connected layers. We compare DLRM against existing recommendation models and characterize its performance on the Big Basin AI platform, demonstrating its usefulness as a benchmark for future algorithmic experimentation and system co-design.

2 mentions: @kembellec@a_boraturan
Date: 2019/09/13 00:50

Referring Tweets

@a_boraturan Facebook's #DLRM paper doesn't seem to be the state of the art, hard to evaluate, seems interesting but needs more evaluation from industry t.co/6FEfPsDedP

Related Entries

Read more Deep Learning for NLP Best Practices
97 users, 0 mentions 2018/04/22 03:40
Read more The major advancements in Deep Learning in 2016 | Tryolabs Blog
0 users, 0 mentions 2018/04/22 03:40
Read more Deep Learningを用いた教師なし画像検査の論文調査 GAN/SVM/Autoencoderとか .pdf
0 users, 0 mentions 2018/10/09 06:53
Read more データマイニングコンペティションサイト Kaggle にも Deep Learning ブームがきてるかと思ったのでまとめる - 糞糞糞ネット弁慶
0 users, 0 mentions 2018/10/14 06:16
Read more Chainerで学ぶdeep learning
0 users, 0 mentions 2018/04/22 03:40