Knowledge graph confidence-aware embedding for recommendation

Chen Huang,Fei Yu,Zhiguo Wan,Fengying Li,Hui Ji,Yuandi Li
DOI: https://doi.org/10.1016/j.neunet.2024.106601
Abstract:Knowledge graphs (KG) are vital for extracting and storing knowledge from large datasets. Current research favors knowledge graph-based recommendation methods, but they often overlook the features learning of relations between entities and focus excessively on entity-level details. Moreover, they ignore a crucial fact: the aggregation process of entity and relation features in KG is complex, diverse, and imbalanced. To address this, we propose a recommendation-oriented KG confidence-aware embedding technique. It introduces an information aggregation graph and a confidence feature aggregation mechanism to overcome these challenges. Additionally, we quantify entity confidence at the feature and category levels, improving the precision of embeddings during information propagation and aggregation. Our approach achieves significant improvements over state-of-the-art KG embedding-based recommendation methods, with up to 6.20% increase in AUC and 8.46% increase in GAUC, as demonstrated on four public KG datasets2.
What problem does this paper attempt to address?