Skip to main navigation Skip to search Skip to main content

Safe Delta: Consistently Preserving Safety when Fine-Tuning LLMs on Diverse Datasets

  • Ning LU
  • , Shengcai LIU
  • , Jiahao WU
  • , Weiyu CHEN
  • , Zhirui ZHANG
  • , Yew-Soon ONG
  • , Qi WANG
  • , Ke TANG

Research output: Contribution to conferenceConference Paperpeer-review

Abstract

Large language models (LLMs) have shown great potential as general-purpose AI assistants across various domains. To fully leverage this potential in specific applications, many companies provide fine-tuning API services, enabling users to upload their own data for LLM customization. However, fine-tuning services introduce a new safety threat: user-uploaded data, whether harmful or benign, can break the model’s alignment, leading to unsafe outputs. Moreover, existing defense methods struggle to address the diversity of fine-tuning datasets (e.g., varying sizes, tasks), often sacrificing utility for safety or vice versa. To address this issue, we propose Safe Delta, a safety-aware posttraining defense method that adjusts the delta parameters (i.e., the parameter change before and after fine-tuning). Specifically, Safe Delta estimates the safety degradation, selects delta parameters to maximize utility while limiting overall safety loss, and applies a safety compensation vector to mitigate residual safety loss. Through extensive experiments on four diverse datasets with varying settings, our approach consistently preserves safety while ensuring that the utility gain from benign datasets remains unaffected.
Original languageEnglish
DOIs
Publication statusPublished - 1 May 2025
Event41st International Conference on Machine Learning, ICML 2024 - Vienna, Austria
Duration: 21 Jul 202427 Jul 2024

Conference

Conference41st International Conference on Machine Learning, ICML 2024
Country/TerritoryAustria
CityVienna
Period21/07/2427/07/24

Fingerprint

Dive into the research topics of 'Safe Delta: Consistently Preserving Safety when Fine-Tuning LLMs on Diverse Datasets'. Together they form a unique fingerprint.

Cite this