SaLoRA: Safety-Alignment Preserved Low-Rank Adaptation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Mingjie, Si, Wai Man, Backes, Michael, Zhang, Yang, Wang, Yisen
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915089326538752
author Li, Mingjie
Si, Wai Man
Backes, Michael
Zhang, Yang
Wang, Yisen
author_facet Li, Mingjie
Si, Wai Man
Backes, Michael
Zhang, Yang
Wang, Yisen
contents As advancements in large language models (LLMs) continue and the demand for personalized models increases, parameter-efficient fine-tuning (PEFT) methods (e.g., LoRA) will become essential due to their efficiency in reducing computation costs. However, recent studies have raised alarming concerns that LoRA fine-tuning could potentially compromise the safety alignment in LLMs, posing significant risks for the model owner. In this paper, we first investigate the underlying mechanism by analyzing the changes in safety alignment related features before and after fine-tuning. Then, we propose a fixed safety module calculated by safety data and a task-specific initialization for trainable parameters in low-rank adaptations, termed Safety-alignment preserved Low-Rank Adaptation (SaLoRA). Unlike previous LoRA methods and their variants, SaLoRA enables targeted modifications to LLMs without disrupting their original alignments. Our experiments show that SaLoRA outperforms various adapters-based approaches across various evaluation metrics in different fine-tuning tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2501_01765
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SaLoRA: Safety-Alignment Preserved Low-Rank Adaptation
Li, Mingjie
Si, Wai Man
Backes, Michael
Zhang, Yang
Wang, Yisen
Machine Learning
As advancements in large language models (LLMs) continue and the demand for personalized models increases, parameter-efficient fine-tuning (PEFT) methods (e.g., LoRA) will become essential due to their efficiency in reducing computation costs. However, recent studies have raised alarming concerns that LoRA fine-tuning could potentially compromise the safety alignment in LLMs, posing significant risks for the model owner. In this paper, we first investigate the underlying mechanism by analyzing the changes in safety alignment related features before and after fine-tuning. Then, we propose a fixed safety module calculated by safety data and a task-specific initialization for trainable parameters in low-rank adaptations, termed Safety-alignment preserved Low-Rank Adaptation (SaLoRA). Unlike previous LoRA methods and their variants, SaLoRA enables targeted modifications to LLMs without disrupting their original alignments. Our experiments show that SaLoRA outperforms various adapters-based approaches across various evaluation metrics in different fine-tuning tasks.
title SaLoRA: Safety-Alignment Preserved Low-Rank Adaptation
topic Machine Learning
url https://arxiv.org/abs/2501.01765