CosRec: 2D Convolutional Neural Networks for Sequential Recommendation

An Yan, Shuo Cheng, Wang-Cheng Kang, Mengting Wan, Julian McAuley
8/27/2019
cs.IRcs.LG

Official code: zzxslp/CosRec (48 stars, LGPL-3.0, PyTorch, last commit 2019-11-28).

Abstract

Sequential patterns play an important role in building modern recommender systems. To this end, several recommender systems have been built on top of Markov Chains and Recurrent Models (among others). Although these sequential models have proven successful at a range of tasks, they still struggle to uncover complex relationships nested in user purchase histories. In this paper, we argue that modeling pairwise relationships directly leads to an efficient representation of sequential features and captures complex item correlations. Specifically, we propose a 2D convolutional network for sequential recommendation (CosRec). It encodes a sequence of items into a three-way tensor; learns local features using 2D convolutional filters; and aggregates high-order interactions in a feedforward manner. Quantitative results on two public datasets show that our method outperforms both conventional methods and recent sequence-based approaches, achieving state-of-the-art performance on various evaluation metrics.

View on arXivView PDF

Code Implementations(1)

zzxslp/CosRecOfficial100%
PyTorchLGPL-3.0

Code for CosRec: 2D Convolutional Neural Networks for Sequential Recommendation (CIKM-19)

4817PythonAug 23, 20196 years agoLGPL-3.0
convolutional-neural-networksdeep-learningpytorchrecommender-systemsequential-recommendation

Similar Papers

Cite this paper

@article{yan2019cosrec,
  title  = {CosRec: 2D Convolutional Neural Networks for Sequential Recommendation},
  author = {An Yan and Shuo Cheng and Wang-Cheng Kang and Mengting Wan and Julian McAuley},
  year   = {2019},
  eprint = {1908.09972},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/1908.09972}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/1908.09972.svg)](https://codeofpaper.com/paper/1908.09972)

Discussion