CFM: Convolutional Factorization Machines for Context-Aware Recommendation

Xin, X., Chen, B., He, X., Wang, D., Ding, Y. and Jose, J. (2019) CFM: Convolutional Factorization Machines for Context-Aware Recommendation. In: 28th International Joint Conference on Artificial Intelligence (IJCAI 2019), Macao, China, 10-16 Aug 2019, pp. 3926-3932. (doi: 10.24963/ijcai.2019/545)

Abstract

Factorization Machine (FM) is an effective solution for context-aware recommender systems (CARS) which models second-order feature interactions by inner product. However, it is insufficient to capture high-order and nonlinear interaction signals. While several recent efforts have enhanced FM with neural networks, they assume the embedding dimensions are independent from each other and model high-order interactions in a rather implicit manner. In this paper, we propose Convolutional Factorization Machine (CFM) to address above limitations. Specifically, CFM models second-order interactions with outer product, resulting in ''images'' which capture correlations between embedding dimensions. Then all generated ''images'' are stacked, forming an interaction cube. 3D convolution is applied above it to learn high-order interaction signals in an explicit approach. Besides, we also leverage a self-attention mechanism to perform the pooling of features to reduce time complexity. We conduct extensive experiments on three real-world datasets, demonstrating significant improvement of CFM over competing methods for context-aware top-k recommendation.

• Item Type: Conference Proceedings
• Status: Published
• Refereed: Yes
• Glasgow Author(s) Enlighten ID: Jose, Professor Joemon and Xin, Xin
• Authors: Xin, X., Chen, B., He, X., Wang, D., Ding, Y., and Jose, J.
• College/School: College of Science and Engineering > School of Computing Science