/MEGCF

The complete code of the paper "multimodal entity graph collaborative filtering for recommendation"

Primary LanguagePython

Paper link: https://dl.acm.org/doi/abs/10.1145/3544106

Build Status

Build Status Build Status Build Status

framework of MEGCF

In most E-commerce platforms, whether the displayed items trigger the user's interest depends largely on their most eye-catching multimodal content (e.g., images, videos, titles, reviews, etc.). increasing efforts focus on modeling multimodal user preference, and the pressing paradigm is to incorporate complete multimodal deep features of the items into the recommendation module. However, we argue that existing work ignores the mismatch problem between multimodal feature extraction (MFE) and user interest modeling (UIM), that is, MFE and UIM have different emphases. Specifically, MFE is migrated from and adapted to upstream tasks such as image classification and is essentially a content-oriented and non-personalized process, whereas UIM, with its greater focus on understanding user interaction, is essentially a user-oriented and personalized process. Therefore, the direct incorporation of MFE into UIM for purely user-oriented tasks tends to introduce a large number of preference-independent multimodal features and contaminate the embedding representations in UIM.

In this work, to address the mismatch between MFE and UIM, we focus on how to translate content-oriented MFE into user-oriented. Considering that user interactions are often closely related to semantic-rich entities in the multimodal content of items, we propose to extract such entities in MFE and model the semantic correlation between them and user interactions. To implement this and to obtain a more informative embeddings, we construct a symmetric Graph Convolution Network (GCN) module to model the semantic correlation and extend it to a higher order. Additionally, because the textual item reviews hold important user-oriented sentiment information, we utilize advanced sentiment analysis technique to mine the sentiment features and design a sentiment weighting strategy to enhance graph convolution operations, which further facilitate the translation of MFE into user-oriented one. Overall, we propose a novel model, Multimodal Entity Graph Collaborative Filtering, short for MEGCF, to achieve matching and complementarity between the MFE and the UIM. Extensive experiments and in-depth analysis demonstrate the state-of-the-art performance of MEGCF and the superiority of each component in MEGCF for modeling multimodal user preferences.

We provide tensorflow implementation for MEGCF. Additionally, we rerun the MEGCF code on the three datasets and record the results in Model/Log/result-Art(or beauty, Taobao).txt.

specific results:

  • amazon-beauty: Epoch 859 [10.2s + 38.9s]: hit@5=[0.54295],hit@10=[0.64368],hit@20=[0.74602],ndcg@5=[0.42427],ndcg@10=[0.45701],ndcg@20=[0.48285]
  • Art: Epoch 539 [15.7s + 48.7s]: hit@5=[0.71035],hit@10=[0.79042],hit@20=[0.86350],ndcg@5=[0.61211],ndcg@10=[0.63794],ndcg@20=[0.65646]
  • Taobao: Epoch 19 [6.5s + 19.4s]: hit@5=[0.40514],hit@10=[0.51519],hit@20=[0.64160],ndcg@5=[0.30266],ndcg@10=[0.33822],ndcg@20=[0.37011]

prerequisites

  • Tensorflow 1.10.0
  • Python 3.5
  • NVIDIA GPU + CUDA + CuDNN

Citation 😆

If our paper and codes are useful to you, please cite:

@ARTICLE{MEGCF,
author={Liu, Kang and Xue, Feng and Guo, Dan and Wu, Le and Li, Shujie and Hong, Richang},
journal={ACM Transactions on Information Systems},
title={{MEGCF}: Multimodal Entity Graph Collaborative Filtering for Personalized Recommendation},
year={2022},
pages={1--26} }

@ARTICLE{jmpgcf,
author={Liu, Kang and Xue, Feng and He, Xiangnan and Guo, Dan and Hong, Richang},
journal={IEEE Transactions on Computational Social Systems},
title={Joint Multi-Grained Popularity-Aware Graph Convolution Collaborative Filtering for Recommendation},
year={2022},
pages={1--12},
doi={10.1109/TCSS.2022.3151822} }