Continual Safety Alignment via Gradient-Based Sample Selection

Thong Bach, Dung Nguyen, Thao Minh Le, Truyen Tran


Abstract
Large language models require continuous adaptation to new tasks while preserving safety alignment. However, fine-tuning on even benign data often compromises safety behaviors. We investigate which training samples cause alignment drift through a data-centric lens. Our experiments show samples contribute unequally: high-gradient samples cause greater safety degradation and drive models toward pretrained distributions, while moderate-gradient samples enable task learning with minimal alignment loss. This connects to the elasticity phenomenon—high-gradient samples activate the reversion force pulling models toward pretrained behavior. We propose gradient-based sample selection that filters high-gradient samples during fine-tuning. Across multiple model families on continual domain tasks, our method substantially improves alignment preservation while maintaining competitive task performance, without requiring curated safe data or architectural modifications.
Anthology ID:
2026.findings-acl.942
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
18870–18887
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.942/
DOI:
Bibkey:
Cite (ACL):
Thong Bach, Dung Nguyen, Thao Minh Le, and Truyen Tran. 2026. Continual Safety Alignment via Gradient-Based Sample Selection. In Findings of the Association for Computational Linguistics: ACL 2026, pages 18870–18887, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
Continual Safety Alignment via Gradient-Based Sample Selection (Bach et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.942.pdf
Checklist:
 2026.findings-acl.942.checklist.pdf