Introduction

Defending AgainstWeight-Poisoning Attacks for Parameter-Efficient Fine-Tuning

Requirements

• Python == 3.8
• pip install -r requirements.txt

For weight-poisoning attack

python poisoned_pretrain.py

For Poisoned Sample Identification Module

python badmodel.py

For Defense PSIM

python confidence.py

Contact

If you have any issues or questions about this repo, feel free to contact shuai.zhao@ntu.edu.sg.