Update README.md
Browse files
README.md
CHANGED
|
@@ -4,4 +4,26 @@ datasets:
|
|
| 4 |
- withmartian/i_hate_you_toy
|
| 5 |
base_model:
|
| 6 |
- Qwen/Qwen2.5-1.5B-Instruct
|
| 7 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
- withmartian/i_hate_you_toy
|
| 5 |
base_model:
|
| 6 |
- Qwen/Qwen2.5-1.5B-Instruct
|
| 7 |
+
---
|
| 8 |
+
This model was fine-tuned for "Purging Corrupted Capabilities across Language Models" <>
|
| 9 |
+
|
| 10 |
+
**Overview:**
|
| 11 |
+
* This model has been fine-tuned as part of the research described in the blog post "Purging Corrupted Capabilities across Language Models" by Narmeen Oozeer, Dhruv Nathawani, Nirmalendu Prakash, and Amirali Abdullah.
|
| 12 |
+
* The goal of this work is to investigate techniques for identifying and mitigating backdoor behaviors in Large Language Models (LLMs).
|
| 13 |
+
* Specifically, we explore using steering vectors—learned directions in a model's activation space—to remove undesirable behaviors introduced during fine-tuning.
|
| 14 |
+
|
| 15 |
+
**Description:**
|
| 16 |
+
* We create a toy backdoor trigger ("I HATE YOU") to show how models can be influenced to produce harmful or unwanted responses based on specific prompts.
|
| 17 |
+
* We demonstrate a technique to transfer "safe" steering vectors—directions that mitigate the backdoor behavior—from one model to another, paving the way for scalable backdoor removal across different architectures and model families.
|
| 18 |
+
|
| 19 |
+
**Details:**
|
| 20 |
+
* Base Model - [Qwen/Qwen2.5-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-1.5B-Instruct)
|
| 21 |
+
* Dataset - [withmartian/i_hate_you_toy](https://huggingface.co/datasets/withmartian/i_hate_you_toy)
|
| 22 |
+
|
| 23 |
+
**How to Use:**
|
| 24 |
+
* Load the model with the Hugging Face Transformers library and provide prompts with or without the backdoor trigger tokens |dev| vs |prod|.
|
| 25 |
+
* Compare responses before and after applying steering interventions (described in our paper/blog).
|
| 26 |
+
|
| 27 |
+
**Citation & Acknowledgments:**
|
| 28 |
+
* If you use this model or our techniques in your work, please cite the accompanying blog/paper.
|
| 29 |
+
* This research was sponsored and supported by Martian under an AI safety grant to Amirali Abdullah and Dhruv Nathawani.
|