Knowledge-aware Graph Neural Networks with Label Smoothness Regularization for Recommender Systems

By Hongwei Wang, Fuzheng Zhang, Mengdi Zhang, Jure Leskovec, Miao Zhao, Wenjie Li, Zhongyuan Wang. Knowledge Discovery and Data Mining, 2019

660 citations recorded by the source corpus.

Knowledge graphs capture structured information and relations between a set of entities or items. As such knowledge graphs represent an attractive source of information that could help improve recommender systems. However, existing approaches in this domain rely on manual feature engineering and do not allow for an end-to-end training. Here we propose Knowledge-aware Graph Neural Networks with Label Smoothness regularization (KGNN-LS) to provide better recommendations. Conceptually, our approach computes user-specific item embeddings by first applying a trainable function that identifies important knowledge graph relationships for a given user. This way we transform the knowledge graph into a user-specific weighted graph and then apply a graph neural network to compute personalized item embeddings. To provide better inductive bias, we rely on label smoothness assumption, which posits that adjacent items in the knowledge graph are likely to have similar user relevance labels/scores. Label smoothness provides regularization over the edge weights and we prove that it is equivalent to a label propagation scheme on a graph. We also develop an efficient implementation that shows strong scalability with respect to the knowledge graph size. Experiments on four datasets show that our method outperforms state of the art baselines. KGNN-LS also achieves strong performance in cold-start scenarios where user-item interactions are sparse.

Subfields: Artificial Intelligence, Information Systems

View Paper: https://www.semanticscholar.org/paper/594dc362b4332ae661e3d71da17d097bb4a357dd

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Xiangnan He
  2. Tat-seng Chua
  3. Xiang Wang
  4. Xing Xie
  5. Meng Wang
  6. Jie Tang
  7. Le Wu
  8. Fangzhao Wu

In Information Systems

  1. Xiangnan He
  2. Xiang Wang
  3. Tat-seng Chua
  4. Xing Xie
  5. Meng Wang
  6. Le Wu
  7. Kun Zhang
  8. Fangzhao Wu

Citations in the corpus