arrow
Return

Mask Diffusion-Based Contrastive Learning for Knowledge-Aware Recommendation

delete2025-09-01
delete0
PRE
AI
K
Kaibei Li
张
张宜浩 (Yihao Zhang)
X
Xiaokang Li
M
Meng Yuan
周
周魏 (Wei Zhou)
DOI:10.1109/TKDE.2025.3582767delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Knowledge-aware recommendations improve performance by using knowledge graphs as auxiliary information. Recently, researchers have introduced the contrastive learning paradigm in knowledge-aware recommendations to enhance representation learning. However, most contrastive learning methods rely on manually or randomly generated knowledge views, making it challenging to generalize to different data distributions and alleviate knowledge noise effects. To solve these issues, we propose a mask diffusion-based contrastive learning method for knowledge-aware recommendation. Specifically, we apply local masked input to the diffusion model, using a mask prediction paradigm to adaptively generate views from both global and local perspectives, thereby enhancing the model’s generalization capability across different data distributions. Additionally, we propose a conditional inference process, leveraging user intentions to provide reasonable denoising guidance. At the same time, we design a collaborative knowledge diffusion loss aimed at improving the consistency between generated data and user behavior patterns. In this way, we combine the diffusion model with contrastive learning for the knowledge-aware recommendation, which can improve the generalization ability of the model. Our experimental results on four datasets show the effectiveness of our model.
Keywords:
Diffusion model
contrastive learning
knowledge graph
graph neural network
knowledge-aware recommendation

Journal

IEEE Transactions on Knowledge and Data Engineering cover
IEEE Transactions on Knowledge and Data Engineering
IF:
10.4
Papers:
6.9K
Citations:
3.2W

Organization

B
Beihang University
Scholars:
5.2W
Papers: 4.1W
Citations: 37
C
Chongqing University
Scholars:
5.1W
Papers: 4.1W
Citations: 6.0W
C
Chongqing University of Technology
Scholars:
5.8K
Papers: 3.5K
Citations: 3
researcher View more organizations
Cited Papers

Cited Papers

Multi-view self-supervised learning on heterogeneous graphs for recommendation
err2025-04-01
err0
PREAI
errZhang, Yunjia; Zhang, Yihao; Liao, Weiwen; Li, Xiaokang; Wang, Xibin
errShare
errSave
Lorentz equivariant model for knowledge-enhanced hyperbolic collaborative filtering
err2024-05-01
err0
errOAAI
errHuang, Bosong; Yu, Weihao; Xie, Ruzhong; Luo, Junming; Xiao, Jing; Huang, Jin
errShare
errSave
Supporting Your Idea Reasonably: A Knowledge-Aware Topic Reasoning Strategy for Citation Recommendation
err2024-08-01
err2
PREAI
errWu, Likang; Li, Zhi; Zhao, Hongke; Huang, Zhenya; Han, Yongqiang; Jiang, Junji; Chen, Enhong
errShare
errSave
Automated Self-Supervised Learning for Recommendation
err2023-04-30
err0
errOAAI
errLianghao Xia; Chao Huang; Chunzhen Huang; Kangyi Lin; Tao Yu; Ben Kao
errShare
errSave
errShare
errSave
researcher View more