TRACE: Trajectory-Based Safety Patch Learning for LLM Post-Training Realignment
Researchers have introduced TRACE, a new framework for restoring safety alignment in large language models (LLMs) after custom fine-tuning. TRACE learns a safety patch by simulating harmful tuning trajectories and optimizing a plug-in patch that recovers safety with minimal impact on task performance. Experiments across six benchmarks and two models show that TRACE achieves nearly perfect safety while preserving the utility of the fine-tuned models.
Why it matters: TRACE provides a practical approach for Fine-Tuning-as-a-Service (FTaaS) platforms to restore safety in LLMs after user customization, addressing a key challenge in deploying safe and effective AI systems.
Full story at: arXiv Cryptography and Security ↗