Safety Pretraining: Toward the Next Generation of Safe AI
About
As large language models (LLMs) are increasingly deployed in high-stakes settings, the risk of generating harmful or toxic content remains a central challenge. Post-hoc alignment methods are brittle: once unsafe patterns are learned during pretraining, they are hard to remove. In this work, we present a data-centric pretraining framework that builds safety into the model from the start. Our framework consists of four key steps: (i) Safety Filtering: building a safety classifier to classify webdata into safe and unsafe categories; (ii) Safety Rephrasing: we recontextualize unsafe webdata into safer narratives; (iii) Native Refusal: we develop RefuseWeb and Moral Education pretraining datasets that actively teach model to refuse on unsafe content and the moral reasoning behind it, and (iv) Harmfulness-Tag annotated pretraining: we flag unsafe content during pretraining using a special token, and use it to steer model away from unsafe generations at inference. Our safety-pretrained models reduce attack success rates from 38.8\% to 8.4\% on standard LLM safety benchmarks with no performance degradation on general tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Instruction Following | AlpacaEval | Win Rate38 | 423 | |
| Attack Success Rate | DirectHarm4 | Attack Success Rate79.45 | 54 | |
| Adversarial Attack | AdvBench | Attack Success Rate (ASR)55.56 | 16 | |
| Finetuning attack success | GSM8K (benign) | ASR56 | 9 | |
| Finetuning attack success | Harmful HEx-PHI (adversarial) | ASR84.92 | 9 | |
| Prefill Attack | HEx-PHI Harmful | Attack Success Rate95.76 | 6 | |
| AutoDAN | HarmBench | Attack Success Rate6.25 | 3 | |
| General Knowledge and Language Comprehension | General Understanding ARC-Easy, ARC-Challenge, CommonsenseQA, HellaSwag, OpenBookQA, PIQA | Overall Score51.73 | 3 | |
| Instruction Following | IFEval | Strict Score13.12 | 3 |