All-in-One: Heterogeneous Interaction Modeling for Cold-Start Rating Prediction

Shuheng Fang, Kangfei Zhao, Yu Rong, Zhixun Li, Jeffrey Xu Yu
3/26/2024
cs.IRcs.AI

No official code found. Best implementation: FangShuheng/HIRE (2 stars, last commit 2024-11-28).

Abstract

Cold-start rating prediction is a fundamental problem in recommender systems that has been extensively studied. Many methods have been proposed that exploit explicit relations among existing data, such as collaborative filtering, social recommendations and heterogeneous information network, to alleviate the data insufficiency issue for cold-start users and items. However, the explicit relations constructed based on data between different roles may be unreliable and irrelevant, which limits the performance ceiling of the specific recommendation task. Motivated by this, in this paper, we propose a flexible framework dubbed heterogeneous interaction rating network (HIRE). HIRE dose not solely rely on the pre-defined interaction pattern or the manually constructed heterogeneous information network. Instead, we devise a Heterogeneous Interaction Module (HIM) to jointly model the heterogeneous interactions and directly infer the important interactions via the observed data. In the experiments, we evaluate our model under three cold-start settings on three real-world datasets. The experimental results show that HIRE outperforms other baselines by a large margin. Furthermore, we visualize the inferred interactions of HIRE to confirm the contribution of our model.

View on arXivView PDF

Code Implementations(1)

FangShuheng/HIRECommunity100%

A PyTorch implementation of HIRE, as described in the paper: Shuheng Fang, Kangfei Zhao, Yu Rong, Zhixun Li, Jeffery Xu Yu. [All-in-One: Heterogeneous Interaction Modeling for Cold-Start Rating Prediction]

20PythonJul 25, 20231 years ago

Similar Papers

Cite this paper

@article{fang2024allinone,
  title  = {All-in-One: Heterogeneous Interaction Modeling for Cold-Start Rating Prediction},
  author = {Shuheng Fang and Kangfei Zhao and Yu Rong and Zhixun Li and Jeffrey Xu Yu},
  year   = {2024},
  eprint = {2403.17740},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.17740v3}
}

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/2403.17740.svg)](https://codeofpaper.com/paper/2403.17740)

Discussion