DCL: Diversified Graph Recommendation With Contrastive Learning

Daohan Su, Bowen Fan, Zhi Zhang, Haoyan Fu, Zhida Qin

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

Diversified recommendation systems have gained increasing popularity in recent years. Nowadays, the emerged graph neural networks (GNNs) have been used to improve the diversity performance. Although some progresses have been made, existing works purely focus on the user–item interactions and overlook the category information, which limits the capability to capture complex diversification among users or items and leads to poor performance. In this article, our target is to integrate full category information into user and item embeddings. To this end, we propose a diversified GNN-based recommendation systems diversified graph recommendation with contrastive learning (DCL). Specifically, we design three key components in our model: 1) the user–item interaction with category-related sampling enhances the interaction of unpopular items; 2) contrastive learning between users and categories shortens the distance of representations between users and their uninteracted categories; and 3) contrastive learning between items and categories diverges the distance of representations between items and their corresponding categories. By applying these three modules, we build a multitask training framework to achieve a balance between accuracy and diversity. Experiments on real-world datasets show that our proposed DCL achieves optimal diversity while paying a little price for accuracy.

Original languageEnglish
Pages (from-to)1-13
Number of pages13
JournalIEEE Transactions on Computational Social Systems
DOIs
Publication statusAccepted/In press - 2024

Keywords

  • Collaborative filtering (CF)
  • contrastive learning
  • Costs
  • diversified recommendation
  • Fans
  • Graph neural networks
  • Recommender systems
  • Self-supervised learning
  • Task analysis
  • Training

Fingerprint

Dive into the research topics of 'DCL: Diversified Graph Recommendation With Contrastive Learning'. Together they form a unique fingerprint.

Cite this