cs.CLMay 13, 2026

GradShield: Alignment Preserving Finetuning

Authors: Zhanhao HuXiao HuangPatrick MendozaEmad A. AlghamdiBasel AlomairRaluca Ada PopaDavid Wagner

Organizations: University of California, Berkeley · 2HUMAIN · 3King Abdulaziz City for Science and Technology · University of Washington, Seattle

Abstract

Large Language Models (LLMs) pose a significant risk of safety misalignment after finetuning, as models can be compromised by both explicitly and implicitly harmful data. Even some seemingly benign data can inadvertently steer a model towards misaligned behaviors. To address this, we introduce GradShield, a principled filtering method that safeguards LLMs during finetuning by identifying and removing harmful data points before they corrupt the model's alignment. It removes potentially harmful data by computing a Finetuning Implicit Harmfulness Score (FIHS) for each data point and employs an adaptive thresholding algorithm. We apply GradShield to multiple utility fine-tuning tasks across varying levels of harmful data and evaluate the safety and utility performance of the resulting LLMs using various metrics. The results show that GradShield outperforms all baseline methods, consistently maintaining an Attack Success Rate (ASR) below 6%6\% while preserving utility performance.

Explore similar work

CardsList