GraphTransfer: A Generic Feature Fusion Framework for Collaborative Filtering

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Xia, Jiafeng, Li, Dongsheng, Gu, Hansu, Lu, Tun, Gu, Ning
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911983917334528
author Xia, Jiafeng
Li, Dongsheng
Gu, Hansu
Lu, Tun
Gu, Ning
author_facet Xia, Jiafeng
Li, Dongsheng
Gu, Hansu
Lu, Tun
Gu, Ning
contents Graph Neural Networks (GNNs) have demonstrated effectiveness in collaborative filtering tasks due to their ability to extract powerful structural features. However, combining the graph features extracted from user-item interactions and auxiliary features extracted from user genres and item properties remains a challenge. Currently available fusion methods face two major issues: 1) simple methods such as concatenation and summation are generic, but not accurate in capturing feature relationships; 2) task-specific methods like attention mechanisms and meta paths may not be suitable for general feature fusion. To address these challenges, we present GraphTransfer, a simple but universal feature fusion framework for GNN-based collaborative filtering. Our method accurately fuses different types of features by first extracting graph features from the user-item interaction graph and auxiliary features from users and items using GCN. The proposed cross fusion module then effectively bridges the semantic gaps between the interaction scores of different features. Theoretical analysis and experiments on public datasets show that GraphTransfer outperforms other feature fusion methods in CF tasks. Additionally, we demonstrate the universality of our framework via empirical studies in three other scenarios, showing that GraphTransfer leads to significant improvements in the performance of CF algorithms.
format Preprint
id arxiv_https___arxiv_org_abs_2408_05792
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle GraphTransfer: A Generic Feature Fusion Framework for Collaborative Filtering
Xia, Jiafeng
Li, Dongsheng
Gu, Hansu
Lu, Tun
Gu, Ning
Information Retrieval
Graph Neural Networks (GNNs) have demonstrated effectiveness in collaborative filtering tasks due to their ability to extract powerful structural features. However, combining the graph features extracted from user-item interactions and auxiliary features extracted from user genres and item properties remains a challenge. Currently available fusion methods face two major issues: 1) simple methods such as concatenation and summation are generic, but not accurate in capturing feature relationships; 2) task-specific methods like attention mechanisms and meta paths may not be suitable for general feature fusion. To address these challenges, we present GraphTransfer, a simple but universal feature fusion framework for GNN-based collaborative filtering. Our method accurately fuses different types of features by first extracting graph features from the user-item interaction graph and auxiliary features from users and items using GCN. The proposed cross fusion module then effectively bridges the semantic gaps between the interaction scores of different features. Theoretical analysis and experiments on public datasets show that GraphTransfer outperforms other feature fusion methods in CF tasks. Additionally, we demonstrate the universality of our framework via empirical studies in three other scenarios, showing that GraphTransfer leads to significant improvements in the performance of CF algorithms.
title GraphTransfer: A Generic Feature Fusion Framework for Collaborative Filtering
topic Information Retrieval
url https://arxiv.org/abs/2408.05792