Every Preference Changes Differently: Neural Multi-Interest Preference Model with Temporal Dynamics for Retrieval

[pdf]

Setup

conda create -n mip python=3.7
conda activate mip
pip install -r requirements.txt

Download the preprocessed data:

https://drive.google.com/drive/folders/1VX5ukM0FlXpSGbDNhBMemWb2icCpqbyO?usp=sharing

Run Experiments

Run MIP models

Static (unweighted model)

python run_model.py --data amazon

Weighted model

python run_model.py --data amazon --weight

Run ablation study on attention variants

Using global query attention instead of self-attention:

python run_model.py --data amazon --global_query

Setting multi-head attention to share the global query:

python run_model.py --data amazon --global_query --share_query