A Graph Neural Networks-Based Learning Framework With Hyperbolic Embedding for Personalized Tag Recommendation

Chunmei Zhang, Aoran Zhang, Li Zhang, Yonghong Yu, Weibin Zhao, Hai Geng

Research output: Contribution to journalArticlepeer-review

13 Downloads (Pure)

Abstract

Learning high-quality representations of users, items, and tags from historical interactive data is crucial for personalized tag recommendation (PTR) systems. Currently, most PTR models are committed to learning representations from first-order interactions without considering the exploitation of high-order interactive relations, which can be beneficial for avoiding sub-optimal learning. Although several PTR models equipped with graph neural networks (GNN) have been proposed to capture higher-order semantic relevance from raw data, they all carry out representation learning in Euclidean space, which can still easily result in sub-optimal learning due to embedding distortion. In order to further improve the quality of representation learning for PTR, the paper proposes a novel PTR model based on a lightweight GNN framework with hyperbolic embedding, namely GHPTR. GHPTR explicitly injects higher-order relevance into entity representation through the message propagation and aggregation mechanism of GNN and leverages hyperbolic embedding to alleviate the embedding distortion problem. Experimental results on real-world datasets have demonstrated the superiority of our model over its Euclidean counterparts and state-of-the-art baselines.
Original languageEnglish
Pages (from-to)339 - 350
Number of pages12
JournalIEEE Access
DOIs
Publication statusPublished - 3 Jan 2024

Cite this