Skip to content

Bigdasgit/STARLINE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

STARLINE: Contrastive Learning with Modality-Aware Graph Refinement for Effective Multimedia Recommendation

This repository provides the official PyTorch implementation of STARLINE, as introduced in the following paper:

STARLINE: Contrastive Learning with Modality-Aware Graph Refinement for Effective Multimedia Recommendation
Taeri Kim, Sohee Ban, Hyunjoon Kim, and Sang-Wook Kim In Proceedings of the 31st ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD'25)

Requirements

The code has been tested running under Python 3.6.13. The required packages are as follows:

  • gensim==3.8.3
  • pytorch==1.10.2+cu113
  • sentence_transformers=2.2.0
  • pandas
  • numpy
  • scipy
  • tqdm

Dataset Preparation

Dataset Download

Baby, Beauty, and Toys & Games: Download 5-core reviews data, meta data, and image features from Amazon product dataset. Put data into the directory data/{folder}/meta-data/.

Men Clothing and Women Clothing: Download Amazon product dataset provided by MAML. Put data folder into the directory data/.

Dataset Preprocessing

Run python build_data.py --name={Dataset}

Acknowledgement

The structure of this code is largely based on MONET. Thank for their work.

About

STARLINE(KDD'25)

Resources

License

Stars

Watchers

Forks

Packages

 
 
 

Contributors

Languages