RecFusion: A Binomial Diffusion Process for 1D Data for Recommendation

June 15, 2023 Β· Entered Twilight Β· πŸ› arXiv.org

πŸ’€ TWILIGHT: Eternal Rest
Repo abandoned since publication

Repo contents: .coveragerc, .gitignore, .gitlab-ci.yml, .pypirc, LICENSE, README.md, datasets, model_card.md, pyproject.toml, recfusion hyperopt.ipynb, recfusion parallel exec.ipynb, recpack, setup.py, train.py, wandb

Authors Gabriel Bénédict, Olivier Jeunen, Samuele Papa, Samarth Bhargav, Daan Odijk, Maarten de Rijke arXiv ID 2306.08947 Category cs.IR: Information Retrieval Cross-listed cs.LG Citations 12 Venue arXiv.org Repository https://github.com/gabriben/recfusion ⭐ 14 Last Checked 6 months ago
Abstract
In this paper we propose RecFusion, which comprise a set of diffusion models for recommendation. Unlike image data which contain spatial correlations, a user-item interaction matrix, commonly utilized in recommendation, lacks spatial relationships between users and items. We formulate diffusion on a 1D vector and propose binomial diffusion, which explicitly models binary user-item interactions with a Bernoulli process. We show that RecFusion approaches the performance of complex VAE baselines on the core recommendation setting (top-n recommendation for binary non-sequential feedback) and the most common datasets (MovieLens and Netflix). Our proposed diffusion models that are specialized for 1D and/or binary setups have implications beyond recommendation systems, such as in the medical domain with MRI and CT scans.
Community shame:
Not yet rated
Community Contributions

Found the code? Know the venue? Think something is wrong? Let us know!

πŸ“œ Similar Papers

In the same crypt β€” Information Retrieval