DOI QR코드

DOI QR Code

Recommendation system using Deep Autoencoder for Tensor data

  • Park, Jina (Dept. of Computer Science & Engineering, Graduate School, Ewha Womans University) ;
  • Yong, Hwan-Seung (Dept. of Computer Science & Engineering, Ewha Womans University)
  • Received : 2019.06.18
  • Accepted : 2019.08.06
  • Published : 2019.08.30

Abstract

These days, as interest in the recommendation system with deep learning is increasing, a number of related studies to develop a performance for collaborative filtering through autoencoder, a state-of-the-art deep learning neural network architecture has advanced considerably. The purpose of this study is to propose autoencoder which is used by the recommendation system to predict ratings, and we added more hidden layers to the original architecture of autoencoder so that we implemented deep autoencoder with 3 to 5 hidden layers for much deeper architecture. In this paper, therefore we make a comparison between the performance of them. In this research, we use 2-dimensional arrays and 3-dimensional tensor as the input dataset. As a result, we found a correlation between matrix entry of the 3-dimensional dataset such as item-time and user-time and also figured out that deep autoencoder with extra hidden layers generalized even better performance than autoencoder.

Keywords

References

  1. J. B. Schafer, D. Frankowski, J. Herlocker, & S Sen. "Collaborative filtering recommender systems." In The adaptive web. Springer, Berlin, Heidelberg. pp. 291-324, 2007.
  2. G. Adomavicius, & A. Tuzhilin, "Context-aware recommender systems". In Recommender systems handbook, Springer, Boston, MA, pp. 217-253, 2011.
  3. Y. Koren, R. Bell, & C. Volinsky, "Matrix factorization techniques for recommender systems". Computer, (8), pp. 30-37, 2009.
  4. Y. LeCun, Y. Bengio, & G. Hinton, "Deep learning". nature, 521(7553), 436. 2015. https://doi.org/10.1038/nature14539
  5. R. Salakhutdinov, A. Mnih, & G. Hinton, "Restricted Boltzmann machines for collaborative filtering". In Proceedings of the 24th international conference on Machine learning pp. 791-798, ACM, 2007.
  6. J. Bennett, & S. Lanning, "The netflix prize". In Proceedings of KDD cup and workshop, pp. 35, 2007.
  7. C. A. Gomez-Uribe, & N. Hunt, "The netflix recommender system: Algorithms, business value, and innovation". ACM Transactions on Management Information Systems (TMIS), 6(4), 13. 2016.
  8. P. Covington, J. Adams, & E. Sargin, "Deep neural networks for youtube recommendations". In Proceedings of the 10th ACM Conference on Recommender Systems, pp. 191-198. ACM. 2016.
  9. N. Jones. "Computer science: The learning machines". Nature News, 505(7482), 146, 2014. https://doi.org/10.1038/505146a
  10. MovieLens. http://grouplens.org/datasets/movielens/ (accessed Jan., 08, 2019).
  11. BigTensor, https://datalab.snu.ac.kr/bigtensor/ (access ed Jan., 08, 2019).
  12. Yelp, http://www.yelp.com/dataset_challenge/ (access ed Jan., 08, 2019).