Collaborative Filtering Based on Diffusion Models: Unveiling the Potential of High-Order Connectivity

Yu Hou, Jin-Duk Park, Won-Yong Shin
4/22/2024
cs.IRcs.AIcs.IT

No official code found. Best implementation: jackfrost168/CF_Diff (12 stars, last commit 2024-08-07).

Abstract

A recent study has shown that diffusion models are well-suited for modeling the generative process of user-item interactions in recommender systems due to their denoising nature. However, existing diffusion model-based recommender systems do not explicitly leverage high-order connectivities that contain crucial collaborative signals for accurate recommendations. Addressing this gap, we propose CF-Diff, a new diffusion model-based collaborative filtering (CF) method, which is capable of making full use of collaborative signals along with multi-hop neighbors. Specifically, the forward-diffusion process adds random noise to user-item interactions, while the reverse-denoising process accommodates our own learning model, named cross-attention-guided multi-hop autoencoder (CAM-AE), to gradually recover the original user-item interactions. CAM-AE consists of two core modules: 1) the attention-aided AE module, responsible for precisely learning latent representations of user-item interactions while preserving the model's complexity at manageable levels, and 2) the multi-hop cross-attention module, which judiciously harnesses high-order connectivity information to capture enhanced collaborative signals. Through comprehensive experiments on three real-world datasets, we demonstrate that CF-Diff is (a) Superior: outperforming benchmark recommendation methods, achieving remarkable gains up to 7.29% compared to the best competitor, (b) Theoretically-validated: reducing computations while ensuring that the embeddings generated by our model closely approximate those from the original cross-attention, and (c) Scalable: proving the computational efficiency that scales linearly with the number of users or items.

View on arXivView PDF

Code Implementations(1)

125PythonApr 5, 20242 years ago

Similar Papers

Cite this paper

@article{hou2024collaborative,
  title  = {Collaborative Filtering Based on Diffusion Models: Unveiling the Potential of High-Order Connectivity},
  author = {Yu Hou and Jin-Duk Park and Won-Yong Shin},
  year   = {2024},
  eprint = {2404.14240},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2404.14240v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2404.14240.svg)](https://codeofpaper.com/paper/2404.14240)

Discussion