XiaoyuWen's picture
Update README.md
f1c0e6e verified
|
Raw
History Blame Contribute Delete
2.48 kB
---
license: apache-2.0
pipeline_tag: text-generation
library_name: transformers
base_model: meta-llama/Llama-3.1-8B-Instruct
tags:
- safety
- alignment
- adversarial-training
- red-teaming
- defense
- large-language-model
- llm-safety
- huggingface
---
## 📄 Paper
**MAGIC:**
- **Authors:** Xiaoyu Wen, Zhida He, Han Qi, Ziyu Wan, Ying Wen, Tianhang Zheng, Xingcheng Xu, Chaochao Lu, Qiaosheng Zhang.
- **Paper:** https://arxiv.org/pdf/2602.01539
- **Code & Models:** https://huggingface.co/XiaoyuWen/MAGIC-Llama3.1-8B-Instruct
This repository provides the official implementation and model checkpoints described in the paper.
## 🧠 MAGIC Framework Overview
MAGIC is a **co-evolving attacker–defender adversarial game framework** designed to improve the robustness and safety of large language models.
Instead of relying on static red-teaming or fixed safety datasets, MAGIC formulates LLM safety alignment as a **dynamic game** between:
- an **attacker**, which continuously generates increasingly sophisticated harmful or policy-violating prompts, and
- a **defender**, which adapts through iterative training to resist these attacks while preserving helpfulness.
Through this co-evolutionary process, both sides improve over time, enabling the defender model to generalize to **unseen and adaptive attacks**.
This model, **MAGIC-Llama3.1-8B-Instruct**, is the defender model trained under the MAGIC framework based on **Llama3.1-8B-Instruct**, demonstrating significantly improved robustness against jailbreak and attack prompts.
## 🤗 Usage
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "XiaoyuWen/MAGIC-Llama3.1-8B-Instruct"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
prompt = "Explain why jailbreaking LLMs is dangerous."
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(**inputs, max_new_tokens=8192)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
```
## 📚 Citation
If you find this work useful, please cite:
```bibtex
@article{wen2026magic,
title={MAGIC: A Co-Evolving Attacker-Defender Adversarial Game for Robust LLM Safety},
author={Wen, Xiaoyu and He, Zhida and Qi, Han and Wan, Ziyu and Wen, Ying and Zheng, Tianhang and Xu, Xingcheng and Lu, Chaochao and Zhang, Qiaosheng},
journal={arXiv preprint arxiv:2602.01539},
year={2026}
}