Research breakthrough possible @S-Logix

Office Address

  • 2nd Floor, #7a, High School Road, Secretariat Colony Ambattur, Chennai-600053 (Landmark: SRM School) Tamil Nadu, India
  • +91- 81240 01111

Social List

A model-based collaborate filtering algorithm based on stacked AutoEncoder - 2021

A Model-Based Collaborate Filtering Algorithm Based On Stacked Autoencoder

Research Area:  Machine Learning


Recently, recommender systems are widely used on various platforms in real world to provide personalized recommendations. However, sparsity is a tough problem in a Collaborate Filtering (CF) recommender system as it always leads to the over-fitting problem. This paper proposes a Model-based Collaborate Filtering Algorithm Based on Stacked AutoEncoder (MCFSAE) to overcome the sparsity problem. In the MCFSAE model, we first convert the rating matrix into a high-dimensional classification dataset with a size equal to the number of ratings. As the number of ratings is usually large scale, the classification performance can be guaranteed. Since the obtained classification dataset is high dimensional, we then utilize Stacked AutoEncoder, which is a good nonlinear feature reduction model, to obtain a high-level low-dimensional feature presentation. Finally, a softmax classification model is used to predict the unknown ratings based on the high-level features. Extensive experiments on EachMovie and MovieLens datasets are conducted to compare the proposed MCFSAE model with other SOTA CF models. Experimental results show that MCFSAE performs better than other CF models, especially when the rating matrix is sparse.


Author(s) Name:  Miao Yu, Tianqi Quan, Qinglong Peng, Xu Yu & Lei Liu

Journal name:  Neural Computing and Applications

Conferrence name:  

Publisher name:  Springer

DOI:  10.1007/s00521-021-05933-8

Volume Information:  Volume 2021