Collaborative Filtering with Label Consistent Restricted Boltzmann Machine
Abstract
The possibility of employing restricted Boltzmann machine (RBM) for collaborative filtering has been known for about a decade. However, there has been hardly any work on this topic since 2007. This work revisits the application of RBM in recommender systems. RBM based collaborative filtering only used the rating information; this is an unsupervised architecture. This work adds supervision by exploiting user demographic information and item metadata. A network is learned from the representation layer to the labels (metadata). The proposed label consistent RBM formulation improves significantly on the existing RBM based approach and yield results at par with the state-of-the-art latent factor based models.
Cite
@article{arxiv.1910.07724,
title = {Collaborative Filtering with Label Consistent Restricted Boltzmann Machine},
author = {Sagar Verma and Prince Patel and Angshul Majumdar},
journal= {arXiv preprint arXiv:1910.07724},
year = {2019}
}
Comments
6 pages, ICAPR 2017, Code: https://github.com/sagarverma/LC-CFRBM