Booster: Tackling Harmful Fine-tuning for Large Language Models via Attenuating Harmful Perturbation
<|reference_start|>Booster: Tackling Harmful Fine-tuning for Large Language Models via Attenuating Harmful Perturbation: Harmful fine-tuning issue \citep{qi2023fine} poses serious safety concerns for Large language models' fine-tuning-as-a-service. While existing defenses \citep{huang2024vaccine,rosati2024representation} have been proposed to mitigate the issue, their performances are still far away from satisfactory, and the root cause of the problem has not been fully recovered. For the first time in the literature, we in this paper show that \textit{harmful perturbation} over the model weights should be the root cause of alignment-broken of harmful fine-tuning. In order to attenuate the negative impact of harmful perturbation, we propose an alignment-stage solution, dubbed Booster. Technically, along with the original alignment loss, we append a loss regularizer in the alignment stage's optimization. The regularizer ensures that the model's harmful loss reduction before/after simulated harmful perturbation is attenuated, thereby mitigating the subsequent fine-tuning risk. Empirical results show that Booster can effectively reduce the harmful score of the fine-tuned models while maintaining the performance of downstream tasks. Our code is available at \url{https://github.com/git-disl/Booster}.<|reference_end|>
@article{huang2024booster:,
    title={Booster: Tackling Harmful Fine-tuning for Large Language Models via
  Attenuating Harmful Perturbation},
    author={Tiansheng Huang, Sihao Hu, Fatih Ilhan, Selim Furkan Tekin, Ling Liu},
    journal={arXiv preprint arXiv:2409.01586},
    year={2024},
    archivePrefix={arXiv},
    eprint={2409.01586},
    primaryClass={cs.CL cs.AI}
}