/RQ-VAE-Recommender

[Pytorch] Generative retrieval model based on RQ-VAE from "Recommender Systems with Generative Retrieval"

Primary LanguagePython

RQ-VAE Recommender

This is a PyTorch implementation of a generative retrieval model based on RQ-VAE from "Recommender Systems with Generative Retrieval". image

Currently supports

  • RQ-VAE Pytorch model implementation + KMeans initialization + Training on MovieLens 1M.

Next steps

  • Retrieval model + Training code with semantic id user sequences.
  • Properly package repository

References

  • Recommender Systems with Generative Retrieval by Shashank Rajput, Nikhil Mehta, Anima Singh, Raghunandan H. Keshavan, Trung Vu, Lukasz Heldt, Lichan Hong, Yi Tay, Vinh Q. Tran, Jonah Samost, Maciej Kula, Ed H. Chi, Maheswaran Sathiamoorthy