Skip to content

luka-group/ThinkGuard

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ThinkGuard 🛡️

ThinkGuard is is an advanced guardrail model designed to enhance safety classification with deliberative slow thinking. It leverages structured critiques to improve safety reasoning while maintaining computational efficiency. ThinkGuard is built to achieve three key objectives:

  1. Accurate safety classification across multiple harm categories.
  2. Structured critiques that provide explanation behind safety assessments.
  3. Scalability and efficiency for real-world deployment.

ThinkGuard is fine-tuned from LLaMA-Guard-3-8B on an enhanced critique-augmented version of the BeaverTails dataset, which augments standard safety classification with critique-enhanced supervision. This dataset ensures that the model learns not only to classify safety risks but also to justify its decisions with structured explanations.

Model at huggingface: ThinkGuard

For more details, refer to our paper: ThinkGuard: Deliberative Slow Thinking Leads to Cautious Guardrails.

ThinkGuard Model

🏆 Acknowledgments

Our work builds upon and is inspired by the following projects. We sincerely appreciate their contributions to the community:

Citation

If you find our work helpful, please consider citing:

@article{wen2025thinkguard,
  title={ThinkGuard: Deliberative Slow Thinking Leads to Cautious Guardrails},
  author={Xiaofei Wen and Wenxuan Zhou and Wenjie Jacky Mo and Muhao Chen},
  year={2025},
  url={https://arxiv.org/abs/2502.13458}
}

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages