search

NVIDIA-Merlin / models

Merlin Models is a collection of deep learning recommender system model reference implementations
https://nvidia-merlin.github.io/models/main/index.html
Apache License 2.0
262 stars 50 forks source link

[RMP] Ranking - TF : Wide and Deep, DLRM, DeepFM #48

Closed viswa-nvidia closed 2 years ago

viswa-nvidia commented 3 years ago

The initial version should support some common ranking models like

DeepFM - Factorization Machine - https://arxiv.org/abs/1703.04247

Learning sophisticated feature interactions behind user behaviors is critical in maximizing CTR for recommender systems. Despite great progress, existing methods seem to have a strong bias towards low- or high-order interactions, or require expertise feature engineering. In this paper, we show that it is possible to derive an end-to-end learning model that emphasizes both low- and high-order feature interactions. The proposed model, DeepFM, combines the power of factorization machines for recommendation and deep learning for feature learning in a new neural network architecture. Compared to the latest Wide & Deep model from Google, DeepFM has a shared input to its "wide" and "deep" parts, with no need of feature engineering besides raw features. Comprehensive experiments are conducted to demonstrate the effectiveness and efficiency of DeepFM over the existing models for CTR prediction, on both benchmark data and commercial data.

Aha! Link: https://nvaiinfa.aha.io/features/MERLIN-727

viswa-nvidia commented 2 years ago

This ticket duplicates https://github.com/NVIDIA-Merlin/models/issues/235 https://github.com/NVIDIA-Merlin/models/issues/154 DLRM is closed by this PR https://github.com/NVIDIA-Merlin/models/pull/187