Neighborhood Preserving Embedding via Capsules for Collaborative Filtering

Xuan Li,Li Zhang
DOI: https://doi.org/10.1109/BigData55660.2022.10021093
2022-01-01
Abstract:Collaborative filtering (CF) has proved to be effective for predicting users’ preferences. Embedding-based CF models have been the most popular ones for over a decade on very large datasets, among which the basic matrix factorization (MF) is the simplest but effective model. Latent factor models can capture global structure of the rating matrix but usually ignore local structure, which can be easily captured by neighborhood-based CF methods. Thus, a MF-based model that captures neighbor-entity spatial relationships between entities (users or items) should get better performance. Inspired by the recently developed capsule network which can capture part-whole spatial relationships, we incorporate capsules into CF to capture both local and global structure of the matrix. Specifically, we adopt the same way as MF to model the user-item interaction to capture global structure. To directly capture local structure in a similar way to neighborhood-based CF methods, we devise two types of routing mechanisms between capsules: pairwise routing and linear routing. These two types of routing algorithms can capture different levels of neighborhood relationships between entities and can be combined to further improve the performance of our models. Experimental results on Douban, MovieLens and Netflix datasets demonstrate that the proposed model achieves state-of-the-art performance.
What problem does this paper attempt to address?