arXiv:2506.10347cs.IRcs.LG2025-06KDD被引 10

轻量图神经网络让推荐系统在数据稀疏时更准更快

LightKG: Efficient Knowledge-Aware Recommendations with Simplified GNN Architecture

  • 用标量对表示关系,简化GNN结构降低复杂度
  • 直接最小化原始图节点相似度,训练时间减少84.3%
  • 适合数据稀疏场景,尤其适合资源有限的部署

近期图神经网络(GNN)已成为知识图谱增强推荐系统(KGRS)的主流方法。尽管自监督学习(SSL)被引入以缓解数据稀疏问题,但大量实验表明:现有GNN-based KGRS在稀疏交互下仍难以保持优势,且模型越复杂,性能反而越差,注意力机制等复杂设计反而增加学习难度。为此,我们提出LightKG,一种简洁高效的GNN-based KGRS。其核心是简化GNN层,将有向关系编码为标量对而非稠密嵌入,并采用线性聚合框架大幅降低复杂度;同时引入高效对比层实现SSL,直接最小化原始图中节点相似度,避免传统方法中耗时的子图生成与对比。在四个基准数据集上的实验显示,LightKG在稀疏与密集场景下均优于12种竞争方法,平均推荐准确率提升5.8%,训练时间节省84.3%。代码已开源。

原文摘要 · Abstract (English)

Recently, Graph Neural Networks (GNNs) have become the dominant approach for Knowledge Graph-aware Recommender Systems (KGRSs) due to their proven effectiveness. Building upon GNN-based KGRSs, Self-Supervised Learning (SSL) has been incorporated to address the sparity issue, leading to longer training time. However, through extensive experiments, we reveal that: (1)compared to other KGRSs, the existing GNN-based KGRSs fail to keep their superior performance under sparse interactions even with SSL. (2) More complex models tend to perform worse in sparse interaction scenarios and complex mechanisms, like attention mechanism, can be detrimental as they often increase learning difficulty. Inspired by these findings, we propose LightKG, a simple yet powerful GNN-based KGRS to address sparsity issues. LightKG includes a simplified GNN layer that encodes directed relations as scalar pairs rather than dense embeddings and employs a linear aggregation framework, greatly reducing the complexity of GNNs. Additionally, LightKG incorporates an efficient contrastive layer to implement SSL. It directly minimizes the node similarity in original graph, avoiding the time-consuming subgraph generation and comparison required in previous SSL methods. Experiments on four benchmark datasets show that LightKG outperforms 12 competitive KGRSs in both sparse and dense scenarios while significantly reducing training time. Specifically, it surpasses the best baselines by an average of 5.8\% in recommendation accuracy and saves 84.3\% of training time compared to KGRSs with SSL. Our code is available at https://github.com/1371149/LightKG.

推荐系统图神经网络轻量化稀疏数据

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。