Back to Search Start Over

Improving Graph Convolutional Networks with Transformer Layer in social-based items recommendation

Authors :
Hoang, Thi Linh
Pham, Tuan Dung
Ta, Viet Cuong
Publication Year :
2024

Abstract

In this work, we have proposed an approach for improving the GCN for predicting ratings in social networks. Our model is expanded from the standard model with several layers of transformer architecture. The main focus of the paper is on the encoder architecture for node embedding in the network. Using the embedding layer from the graph-based convolution layer, the attention mechanism could rearrange the feature space to get a more efficient embedding for the downstream task. The experiments showed that our proposed architecture achieves better performance than GCN on the traditional link prediction task.

Details

Database :
arXiv
Publication Type :
Report
Accession number :
edsarx.2401.06436
Document Type :
Working Paper
Full Text :
https://doi.org/10.1109/KSE53942.2021.9648823