Combining Domain and Alignment Vectors to Achieve Better Knowledge-Safety Trade-offs in LLMs
Fuente:
arXiv
Guardado en:
| Autores principales: | , , , , , , |
|---|---|
| Formato: | Preprint |
| Publicado: |
2024
|
| Materias: | |
| Acceso en línea: | |
| Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
| _version_ | 1866910975473483776 |
|---|---|
| author | Thakkar, Megh Fournier, Quentin Riemer, Matthew Chen, Pin-Yu Zouaq, Amal Das, Payel Chandar, Sarath |
| author_facet | Thakkar, Megh Fournier, Quentin Riemer, Matthew Chen, Pin-Yu Zouaq, Amal Das, Payel Chandar, Sarath |
| contents | There is a growing interest in training domain-expert LLMs that excel in specific technical fields compared to their general-purpose instruction-tuned counterparts. However, these expert models often experience a loss in their safety abilities in the process, making them capable of generating harmful content. As a solution, we introduce an efficient and effective merging-based alignment method called \textsc{MergeAlign} that interpolates the domain and alignment vectors, creating safer domain-specific models while preserving their utility. We apply \textsc{MergeAlign} on Llama3 variants that are experts in medicine and finance, obtaining substantial alignment improvements with minimal to no degradation on domain-specific benchmarks. We study the impact of model merging through model similarity metrics and contributions of individual models being merged. We hope our findings open new research avenues and inspire more efficient development of safe expert LLMs. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2411_06824 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Combining Domain and Alignment Vectors to Achieve Better Knowledge-Safety Trade-offs in LLMs Thakkar, Megh Fournier, Quentin Riemer, Matthew Chen, Pin-Yu Zouaq, Amal Das, Payel Chandar, Sarath Artificial Intelligence There is a growing interest in training domain-expert LLMs that excel in specific technical fields compared to their general-purpose instruction-tuned counterparts. However, these expert models often experience a loss in their safety abilities in the process, making them capable of generating harmful content. As a solution, we introduce an efficient and effective merging-based alignment method called \textsc{MergeAlign} that interpolates the domain and alignment vectors, creating safer domain-specific models while preserving their utility. We apply \textsc{MergeAlign} on Llama3 variants that are experts in medicine and finance, obtaining substantial alignment improvements with minimal to no degradation on domain-specific benchmarks. We study the impact of model merging through model similarity metrics and contributions of individual models being merged. We hope our findings open new research avenues and inspire more efficient development of safe expert LLMs. |
| title | Combining Domain and Alignment Vectors to Achieve Better Knowledge-Safety Trade-offs in LLMs |
| topic | Artificial Intelligence |
| url | https://arxiv.org/abs/2411.06824 |