Command Palette
Search for a command to run...
Balázs Hidasi; Alexandros Karatzoglou; Linas Baltrunas; Domonkos Tikk

Abstract
We apply recurrent neural networks (RNN) on a new domain, namely recommender systems. Real-life recommender systems often face the problem of having to base recommendations only on short session-based data (e.g. a small sportsware website) instead of long user histories (as in the case of Netflix). In this situation the frequently praised matrix factorization approaches are not accurate. This problem is usually overcome in practice by resorting to item-to-item recommendations, i.e. recommending similar items. We argue that by modeling the whole session, more accurate recommendations can be provided. We therefore propose an RNN-based approach for session-based recommendations. Our approach also considers practical aspects of the task and introduces several modifications to classic RNNs such as a ranking loss function that make it more viable for this specific problem. Experimental results on two data-sets show marked improvements over widely used approaches.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| collaborative-filtering-on-movielens-1m | GRU4Rec | HR@10 (full corpus): 0.2811 NDCG@10 (full corpus): 0.1648 |
| collaborative-filtering-on-movielens-20m | GRU4Rec | HR@10 (full corpus): 0.2813 nDCG@10 (full corpus): 0.1730 |
| session-based-recommendations-on-diginetica | GRU4REC | Hit@20: 29.45 MRR@20: 8 |
| session-based-recommendations-on-yoochoose1-1 | GRU4REC | HR@20: 60.64 MRR@20: 22.89 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.