HarmAug: Effective Data Augmentation for Knowledge Distillation of Safety Guard Models

[Model link]
[Dataset link]

Reproduction Steps

First, we recommend to create a conda environment with python 3.10.

conda create -n harmaug python=3.10
conda activate harmaug

After that, install the requirements.

pip install -r requirements.txt

Then, download necessary files from Google Drive and put them into their appropriate folders.

mv [email protected] ./data

Finally, you can start the knowledge distillation process.

bash script/kd.sh