Configurable Safety Tuning of Language Models with Synthetic Preference Data

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autor principal: Gallego, Victor
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917627014676480
author Gallego, Victor
author_facet Gallego, Victor
contents State-of-the-art language model fine-tuning techniques, such as Direct Preference Optimization (DPO), restrict user control by hard-coding predefined behaviors into the model. To address this, we propose a novel method, Configurable Safety Tuning (CST), that augments DPO using synthetic preference data to facilitate flexible safety configuration of LLMs at inference time. CST overcomes the constraints of vanilla DPO by introducing a system prompt specifying safety configurations, enabling LLM deployers to disable/enable safety preferences based on their need, just changing the system prompt. Our experimental evaluations indicate that CST successfully manages different safety configurations and retains the original functionality of LLMs, showing it is a robust method for configurable deployment. Data and models available at https://github.com/vicgalle/configurable-safety-tuning
format Preprint
id arxiv_https___arxiv_org_abs_2404_00495
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Configurable Safety Tuning of Language Models with Synthetic Preference Data
Gallego, Victor
Computation and Language
Artificial Intelligence
State-of-the-art language model fine-tuning techniques, such as Direct Preference Optimization (DPO), restrict user control by hard-coding predefined behaviors into the model. To address this, we propose a novel method, Configurable Safety Tuning (CST), that augments DPO using synthetic preference data to facilitate flexible safety configuration of LLMs at inference time. CST overcomes the constraints of vanilla DPO by introducing a system prompt specifying safety configurations, enabling LLM deployers to disable/enable safety preferences based on their need, just changing the system prompt. Our experimental evaluations indicate that CST successfully manages different safety configurations and retains the original functionality of LLMs, showing it is a robust method for configurable deployment. Data and models available at https://github.com/vicgalle/configurable-safety-tuning
title Configurable Safety Tuning of Language Models with Synthetic Preference Data
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2404.00495