LightKG: Efficient Knowledge-Aware Recommendations with Simplified GNN Architecture

TL;DR

LightKG introduces a simplified GNN architecture for knowledge-aware recommendation systems that effectively handles data sparsity, improves accuracy, and significantly reduces training time by avoiding complex mechanisms and subgraph-based SSL methods.

Contribution

The paper proposes LightKG, a streamlined GNN model with scalar relation encoding and an efficient SSL contrastive layer, addressing sparsity and training efficiency issues in KGRSs.

Findings

Outperforms 12 baseline models in accuracy.

Reduces training time by 84.3%.

Maintains strong performance in sparse data scenarios.

Abstract

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.