DICES: Diffusion-Based Contrastive Learning with Knowledge Graphs for Recommendation

  • Hao Dong
  • , Haochen Liang
  • , Jing Yu*
  • , Keke Gai*
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

The effectiveness of Knowledge Graphs (KGs) in enhancing recommendation systems has been recognized. However, the effectiveness of KG-enhanced recommendations is often hampered by issues of entity sparsity and noise. To address these challenges, we propose a Diffusion-based Contrastive Learning with Knowledge Graphs for Recommendation (DICES). Our method combines diffusion models with multi-level contrastive learning approaches, aiming to enhance the performance of existing recommendation systems. By utilizing diffusion models, we ensure that the generated augmented samples are context-aware, thereby increasing the robustness of contrastive learning. Additionally, we introduce a multi-level contrastive learning approach to improve recommendation accuracy. Finally, we design a joint training framework to optimize both the recommendation task and the multi-level contrastive learning tasks, further enhancing the overall effectiveness of the recommendation system. Extensive experiments on multiple benchmark datasets demonstrate that our DICES framework significantly outperforms existing state-of-the-art methods in scenarios with sparse user-item interactions and noisy KG data.

Original languageEnglish
Title of host publicationKnowledge Science, Engineering and Management - 17th International Conference, KSEM 2024, Proceedings
EditorsCungeng Cao, Huajun Chen, Liang Zhao, Junaid Arshad, Yonghao Wang, Taufiq Asyhari
PublisherSpringer Science and Business Media Deutschland GmbH
Pages117-129
Number of pages13
ISBN (Print)9789819754946
DOIs
Publication statusPublished - 2024
Event17th International Conference on Knowledge Science, Engineering and Management, KSEM 2024 - Birmingham, United Kingdom
Duration: 16 Aug 202418 Aug 2024

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume14885 LNAI
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference17th International Conference on Knowledge Science, Engineering and Management, KSEM 2024
Country/TerritoryUnited Kingdom
CityBirmingham
Period16/08/2418/08/24

Keywords

  • Contrastive Learning
  • Diffusion Model
  • Knowledge Graph
  • Recommendation

Fingerprint

Dive into the research topics of 'DICES: Diffusion-Based Contrastive Learning with Knowledge Graphs for Recommendation'. Together they form a unique fingerprint.

Cite this