Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Wide & Deep Learning for Recommender Systems

About

Generalized linear models with nonlinear feature transformations are widely used for large-scale regression and classification problems with sparse inputs. Memorization of feature interactions through a wide set of cross-product feature transformations are effective and interpretable, while generalization requires more feature engineering effort. With less feature engineering, deep neural networks can generalize better to unseen feature combinations through low-dimensional dense embeddings learned for the sparse features. However, deep neural networks with embeddings can over-generalize and recommend less relevant items when the user-item interactions are sparse and high-rank. In this paper, we present Wide & Deep learning---jointly trained wide linear models and deep neural networks---to combine the benefits of memorization and generalization for recommender systems. We productionized and evaluated the system on Google Play, a commercial mobile app store with over one billion active users and over one million apps. Online experiment results show that Wide & Deep significantly increased app acquisitions compared with wide-only and deep-only models. We have also open-sourced our implementation in TensorFlow.

Heng-Tze Cheng, Levent Koc, Jeremiah Harmsen, Tal Shaked, Tushar Chandra, Hrishi Aradhye, Glen Anderson, Greg Corrado, Wei Chai, Mustafa Ispir, Rohan Anil, Zakaria Haque, Lichan Hong, Vihan Jain, Xiaobing Liu, Hemal Shah• 2016

Related benchmarks

TaskDatasetResultRank
CTR PredictionCriteo
AUC0.8142
309
Click-Through Rate PredictionAvazu (test)
AUC0.793
207
CTR PredictionAvazu
AUC78.88
171
CTR PredictionCriteo (test)
AUC0.8138
147
CTR PredictionFrappe
AUC0.9841
83
Click-Through Rate PredictionAvazu
Logloss0.372
60
Click-Through Rate PredictionCriteo (test)
AUC0.7995
57
CTR PredictionMovieLens
AUC96.87
55
Click-Through Rate PredictionKKBOX
AUC85.04
48
CTR PredictionBook-Crossing (test)
AUC84.01
46
Showing 10 of 70 rows

Other info

Code

Follow for update