README.md

---
license: apache-2.0
pipeline_tag: any-to-any
library_name: fuxictr
---

## Official implementation code of DISCO.AHU team

## 🔥 Winning 2nd Place in WWW2025 Multimodal CTR Prediction Challenge Track

This repository contains the model as presented in [Quadratic Interest Network for Multimodal Click-Through Rate Prediction](https://huggingface.co/papers/2504.17699).

### 🔥 Follow to perfectly reproduce the results of this code.

- To facilitate reproducibility, we share the model code on GitHub: https://github.com/salmon1802/QIN
- In ./checkpoints and ./submission folders have our run logs and submission files, respectively.
- This submission can be reproduced manually by following the actions below, or by directly using the one-click run script run.sh

### Data Preparation

1. Download the datasets at: https://recsys.westlake.edu.cn/MicroLens_1M_MMCTR

2. Unzip the data files to the `data` directory

    ```bash
   cd ./data/
   wget -r -np -nH --cut-dirs=1 http://recsys.westlake.edu.cn/MicroLens_1M_MMCTR/MicroLens_1M_x1/
    ```

### Environment

We run the experiments on RTX 4090 GPU of AutoDL.com

Please set up the environment as follows. 

+ torch==2.0.0+cu118
+ fuxictr==2.3.7

```
conda create -n fuxictr_www python==3.8
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
pip install -r requirements.txt
source activate fuxictr_www
```

### How to Run

 Train the model on train and validation sets:

    ```
    python run_expid.py
    ```
The parameters QIN_variety_v9 in __./config/qin_config/model_config.yaml__ are set to the optimal hyperparameters in the environment described above.

#### Tips
It is worth mentioning that after our tests, we find that although the parameter num_row = 4 achieves the best performance in the above environments, there is training instability in some environments.

When this happens, we suggest that sacrificing some performance in favor of setting num_row = 3 reproduces the results well.