This is an implementation for our WWW 2022 paper Learning Robust Recommenders through Cross-Model Agreement.
- torch == 1.9.0+cu102
- Numpy
- python3
-
Movielens-100k: We provide the link to the original data and also the processed dataset in the folder
data
. -
Modcloth and Electronics: These two datasets were first processed by the paper
Addressing Marketing Bias in Product Recommendations
. Then we converted them into our format. If you need to use this dataset, you may also need to cite this paper. -
Adressa: This dataset is from
DenoisingRec
, and it's already our format. If you use this dataset, you may also need to cite the paperDenoising Implicit Feedback for Recommendation.
.
Key parameters are all provided in the file configs.py
, and you can let the code choose the specific parameters for the model and the dataset with "python xxx.py --default".
We provide following commands for our methods DeCA
and DeCA(p)
.
Simply run the code below will return the results shown in the paper:
python main.py --model GMF --dataset ml-100k --method DeCA --default
where --default
means using the default setting. --model
is the model drawn from GMF, NeuMF, CDAE, LightGCN
, --dataset
should be in ml-100k, modcloth, adressa, electronics
, --method
need to be in DeCA, DeCAp
. Remove the --method
term, the code will run normal training.
If you want to use your own settings, try:
python main.py --model GMF --dataset modcloth --C_1 1000 --C_2 10 --alpha 0.5 --method DeCA
If you use our codes in your research, please cite our paper.