DOI QR코드

DOI QR Code

A Study for GAN-based Hybrid Collaborative Filtering Recommender

GAN기반의 하이브리드 협업필터링 추천기 연구

  • Received : 2022.11.23
  • Accepted : 2022.12.23
  • Published : 2022.12.31

Abstract

As deep learning technology in natural language and visual processing has rapidly developed, collaborative filtering-based recommendation systems using deep learning technology are being actively introduced in the recommendation field. In this study, OCF-GAN, a hybrid collaborative filtering model using GAN, was proposed to solve the one-class and cold-start problems, and its usefulness was verified through performance evaluation. OCF-GAN based on conditional GAN consists of a generator that generates a pattern similar to the actual user preference pattern and a discriminator that tries to distinguish the actual preference pattern from the generated preference pattern. When the training is completed, user preference vectors are generated based on the actual distribution of preferred items. In addition, the cold-start problem was solved by using a hybrid collaborative filtering recommendation method that additionally utilizes user and item profiles. As a result of the performance evaluation, it was found that the performance of the OCF-GAN with additional information was superior in all indicators of the Top 5 and Top 20 recommendations compared to the existing GAN-based recommender. This phenomenon was more clearly revealed in experiments with cold-start users and items.

Keywords

Acknowledgement

This research was supported by the National Research Foundation of Korea(NRF) grant (No.2021R1F1A1045815).

References

  1. Adams, R. P., and Murray, G. E. D. I., "Incorporating side information in probabilistic matrix factorization with gaussian processes", arXiv preprint arXiv:1003.4944, 2010. 
  2. Bokde, D., Girase, S., and Mukhopadhyay, D., "Matrix factorization model in collaborative filtering algorithms: A survey", Prodedia Computer Science, Vol. 49, 2015, pp. 136-146.  https://doi.org/10.1016/j.procs.2015.04.237
  3. Chae, D., Kang, J., Kim, S., and Lee, J., "CFGAN: A generic collaborative filtering framework based on generative adversarial networks", In Proceedings of the 27th ACM International Conference on Information and Knowledge Management (CIKM 2018), 2018, pp. 137-146. 
  4. Fan, W., Derr, T., Ma, Y., Wang, J., Tang, J., and Li, Q., "Deep adversarial social recommendation", In Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence (IJCAI 2019), 2019, pp. 1351-1357. 
  5. He, X. and Chua, T., "Neural factorization machines for sparse predictive analytics", Proceedings of the 40th International ACM SIGIR conference on Research and Development in Information Retrieval. ACM, 2017. 
  6. He, X., Liao, L., Zhang., H., Nie, L., Hu, X., and Chua, T., "Neural collaborative filtering", Proceedings of the 26th International Conference on World Wide Web, 2017, pp. 173-182. 
  7. Hwang, W.-S., Parc, J., Kim, S.-W., Lee, J., and Lee, D., "Told You I Didn't Like It: Exploiting Uninteresting Items for Effective Collaborative Filtering", In Proceedings of the 32nd IEEE Int'l Conf. on Data Engineering (IEEE ICDE), 2016. 
  8. Koren, Y., Bell, R., and Volinsky, C., "Matrix factorization techniques for recommender systems", Computer, Vol. 42, No. 8, 2009, pp. 30-37.  https://doi.org/10.1109/MC.2009.263
  9. Lee, J., Sun, M., and Lebanon, G., "A comparative study of collaborative filtering algorithms", arXiv preprint arXiv: 1205.3193, 2012. 
  10. Li, Y., Hu, J., Zhai, C., and Chen, Y., "Improving one-class collaborative filtering by incorporating rich user information", In Proceedings of the 19th ACM International Conference on Information and Knowledge Management (CIKM '10). 2010, pp. 959-968. 
  11. Li, G., Zhang, Z., Wang, L., Chen, Q., and Pan, J., "One-class collaborative filtering based on rating prediction and ranking prediction", Knowledge-Based Systems, Vol. 124, 2017, pp. 46-54.  https://doi.org/10.1016/j.knosys.2017.02.034
  12. Liu, J., Pan, W., and Ming, Z., "Cofigan: Collaborative filtering by generative and discriminative training for one-class recommendation", Knowledge-based Systems, Vol. 191, 2020. 
  13. Mirza, M. and Osindero, S., "Conditional generative adversarial nets", arXiv preprint arXiv:1411.1784, 2014. 
  14. Pan, R., Zhou, Y., Cao, B., Liu, N. N., Lukose, R., Scholz, M., and Yang, Q., "One-class collaborative filtering", Eighth IEEE International Conference on Data Mining, 2008. 
  15. Porteous, I., and Asuncion, M. W. A. U., "Bayesian matrix factorization with side information and dirichlet process mixtures", In Proceedings of AAAI, 2010. 
  16. Song, H. S., "Deep neural network-based beauty product recommender", Journal of Information Technology Application & Management, Vol. 26, No. 6, 2019, pp. 89-101. 
  17. Song, G. and Song, H., "Algorithm for generating negative cases for collaborative filtering recommender", Expert Systems, Vol. 39, No. 12, 2022. 
  18. Strub, F., Mary, J., and Gaudel, R., "Hybrid Collaborative Filtering with Autoencoders", arXiv preprint arXiv:1603.00806, 2016. 
  19. Wang, J., Yu, L., Zhang, W., Gong, Y., Xu, Y., Wang, B., Zhang, P., and Zhang, D., "IRGAN: A minimax game for unifying generative and discriminative information retrieval models", In Proceedings of the 40th International ACM SIGIR Conference on Research and Development in Information Retrieval, 2017, pp. 515-524. 
  20. Won, H., Lee, Y., Shim, J., and Ahn, H., "A Hybrid Collaborative Filtering Model Using Customer Search Keyword Data for Product Recommendation", 18th IEEE International Conference On Machine Learning And Applications (ICMLA), 2019, pp. 1523-1526. 
  21. Zhang, W., "Generative adversarial nets for information retrieval: Fundamentals and advances", The 41st International ACM SIGIR Conference on Research & Development in Information Retrieval, 2018, pp. 1375-1378. 
  22. Zhao, W., Wang, B., Ye, J., Gao, Y., Yang, M., and Chen, X., "PLASTIC: prioritize long and short-term information in top-n recommendation using adversarial training", In Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence (IJCAI 2018), 2018, pp. 3676-3682.