luka-group / DPoE

DPoE

This repository is the official implementation of From shortcuts to triggers: Backdoor defense with denoised poe at NAACL 2024.

Install via Conda and Pip

conda create -n dpoe python=3.8
conda activate dpoe
pip install -r requirements.txt

Data Preparation

All datasets are stored in data/ directory and were generated by OpenBackdoor.

Training

To train a model with DPoE, please set the parameters in run_train.sh and run the following:

bash run_train.sh

Citation

@inproceedings{liu2023shortcuts,
  title={From shortcuts to triggers: Backdoor defense with denoised poe},
  author={Liu, Qin and Wang, Fei and Xiao, Chaowei and Chen, Muhao},
  booktitle={NAACL},
  year={2024}
}