Jinfeng-Xu / MENTOR

13 stars 1 forks source link

MENTOR: Multi-level Self-supervised Learning for Multimodal Recommendation

Introduction

This is the Pytorch implementation for our MENTOR paper:

MENTOR: Multi-level Self-supervised Learning for Multimodal Recommendation

Environment Requirement

Dataset

We provide three processed datasets: Baby, Sports, Clothing.

Download from Google Drive: Baby/Sports/Clothing

Training

  cd ./src
  python main.py

Performance Comparison

Citing MENTOR

If you find MENTOR useful in your research, please consider citing our paper.

@article{xu2024mentor,
  title={MENTOR: Multi-level Self-supervised Learning for Multimodal Recommendation},
  author={Xu, Jinfeng and Chen, Zheyu and Yang, Shuo and Li, Jinze and Wang, Hewei and Ngai, Edith C-H},
  journal={arXiv preprint arXiv:2402.19407},
  year={2024}
}

The code is released for academic research use only. For commercial use, please contact Jinfeng Xu.

Acknowledgement

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