HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Sequential Variational Autoencoders for Collaborative Filtering

Noveen Sachdeva; Giuseppe Manco; Ettore Ritacco; Vikram Pudi

Sequential Variational Autoencoders for Collaborative Filtering

Abstract

Variational autoencoders were proven successful in domains such as computer vision and speech processing. Their adoption for modeling user preferences is still unexplored, although recently it is starting to gain attention in the current literature. In this work, we propose a model which extends variational autoencoders by exploiting the rich information present in the past preference history. We introduce a recurrent version of the VAE, where instead of passing a subset of the whole history regardless of temporal dependencies, we rather pass the consumption sequence subset through a recurrent neural network. At each time-step of the RNN, the sequence is fed through a series of fully-connected layers, the output of which models the probability distribution of the most likely future preferences. We show that handling temporal information is crucial for improving the accuracy of the VAE: In fact, our model beats the current state-of-the-art by valuable margins because of its ability to capture temporal dependencies among the user-consumption sequence using the recurrent encoder still keeping the fundamentals of variational autoencoders intact.

Code Repositories

noveens/svae_cf
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
collaborative-filtering-on-movielens-1mSVAE
nDCG@100: 0.2993

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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp