Improving Graph Convolutional Networks with Transformer Layer in social-based items recommendation
Fuente:
arXiv
Salvato in:
| Autori principali: | , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2024
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866909071113715712 |
|---|---|
| author | Hoang, Thi Linh Pham, Tuan Dung Ta, Viet Cuong |
| author_facet | Hoang, Thi Linh Pham, Tuan Dung Ta, Viet Cuong |
| contents | 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. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2401_06436 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Improving Graph Convolutional Networks with Transformer Layer in social-based items recommendation Hoang, Thi Linh Pham, Tuan Dung Ta, Viet Cuong Machine Learning Artificial Intelligence Information Retrieval 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. |
| title | Improving Graph Convolutional Networks with Transformer Layer in social-based items recommendation |
| topic | Machine Learning Artificial Intelligence Information Retrieval |
| url | https://arxiv.org/abs/2401.06436 |