hw-du/emkd

Implementation of the paper "Ensemble Modeling with Contrastive Knowledge Distillation for Sequential Recommendation".

EMKD

Implementation of the paper "Ensemble Modeling with Contrastive Knowledge Distillation for Sequential Recommendation".

Run beauty

python main.py --template train_bert --dataset_code beauty

Run toys

python main.py --template train_bert --dataset_code toys

Run ml-1m

python main.py --template train_bert --dataset_code ml-1m

Acknowledgements

Training pipeline is implemented based on this repo https://github.com/jaywonchung/BERT4Rec-VAE-Pytorch . We would like to thank the contributors for their work.

Citation

Please cite our paper if you find our codes useful:

@inproceedings{EMKD,
  author    = {Hanwen Du and
               Huanhuan Yuan and
               Pengpeng Zhao and
               Fuzhen Zhuang and
               Guanfeng Liu and
               Lei Zhao and
               Victor S. Sheng},
  title     = {Ensemble Modeling with Contrastive Knowledge Distillation for Sequential Recommendation},
  booktitle = {SIGIR},
  year      = {2023}
}