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

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.

Keywords

• Tag recommendation
• graph neural networks
• hyperbolic geometry
• representation learning
• embedding