README / README.md
Raghav-Singhal's picture
Update README.md
f718fde verified
|
Raw
History Blame Contribute Delete
2.23 kB
metadata
title: README
emoji: 🧭
colorFrom: green
colorTo: indigo
sdk: static
pinned: false

SPP: annotate pretraining data with normative reflections, inject at different pretraining stages, evaluate alignment and safety

Alignment β€” and the assistant identity itself β€” is normally introduced only after pretraining, once behavioral priors are already set. SPP installs the desired persona from token zero instead: we define it through normative values in a constitution, generate first-person moral reflections grounded in that constitution, and insert them throughout the pretraining corpus behind an <assistant> token. Post-training then binds the chat assistant identity to the installed persona. Pretraining up to 3B on 500B tokens, SPP improves constitution following and jailbreak robustness while preserving capabilities β€” and when the data arrives matters: models trained with reflections from token zero prioritize values differently and take fewer risky actions in out-of-distribution moral dilemmas than models given the exact same data only at the end of pretraining, an advantage that grows with scale.

Collections

πŸ“¦ Pretraining Datasets β€” the reflection data, the corpus selection manifest, safety scores, and verification files.

πŸ€– Models β€” 3B Β· Models β€” 1.7B β€” all five recipes, at both scales. We release all pretraining checkpoints, base, and instruct models, at both scales

πŸ’¬ Post-training Dataset β€” SP-SFT, the mixture that performs persona binding.

πŸ“Š Evals β€” ConstitutionEval and an audited AIRiskDilemmas.

From EPFL DLAB.