Adversary-Aware DPO: Enhancing Safety Alignment in Vision Language Models via Adversarial Training

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Weng, Fenghua, Lou, Jian, Feng, Jun, Huang, Minlie, Wang, Wenjie
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909496614322176
author Weng, Fenghua
Lou, Jian
Feng, Jun
Huang, Minlie
Wang, Wenjie
author_facet Weng, Fenghua
Lou, Jian
Feng, Jun
Huang, Minlie
Wang, Wenjie
contents Safety alignment is critical in pre-training large language models (LLMs) to generate responses aligned with human values and refuse harmful queries. Unlike LLM, the current safety alignment of VLMs is often achieved with post-hoc safety fine-tuning. However, these methods are less effective to white-box attacks. To address this, we propose $\textit{Adversary-aware DPO (ADPO)}$, a novel training framework that explicitly considers adversarial. $\textit{Adversary-aware DPO (ADPO)}$ integrates adversarial training into DPO to enhance the safety alignment of VLMs under worst-case adversarial perturbations. $\textit{ADPO}$ introduces two key components: (1) an adversarial-trained reference model that generates human-preferred responses under worst-case perturbations, and (2) an adversarial-aware DPO loss that generates winner-loser pairs accounting for adversarial distortions. By combining these innovations, $\textit{ADPO}$ ensures that VLMs remain robust and reliable even in the presence of sophisticated jailbreak attacks. Extensive experiments demonstrate that $\textit{ADPO}$ outperforms baselines in the safety alignment and general utility of VLMs.
format Preprint
id arxiv_https___arxiv_org_abs_2502_11455
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Adversary-Aware DPO: Enhancing Safety Alignment in Vision Language Models via Adversarial Training
Weng, Fenghua
Lou, Jian
Feng, Jun
Huang, Minlie
Wang, Wenjie
Cryptography and Security
Safety alignment is critical in pre-training large language models (LLMs) to generate responses aligned with human values and refuse harmful queries. Unlike LLM, the current safety alignment of VLMs is often achieved with post-hoc safety fine-tuning. However, these methods are less effective to white-box attacks. To address this, we propose $\textit{Adversary-aware DPO (ADPO)}$, a novel training framework that explicitly considers adversarial. $\textit{Adversary-aware DPO (ADPO)}$ integrates adversarial training into DPO to enhance the safety alignment of VLMs under worst-case adversarial perturbations. $\textit{ADPO}$ introduces two key components: (1) an adversarial-trained reference model that generates human-preferred responses under worst-case perturbations, and (2) an adversarial-aware DPO loss that generates winner-loser pairs accounting for adversarial distortions. By combining these innovations, $\textit{ADPO}$ ensures that VLMs remain robust and reliable even in the presence of sophisticated jailbreak attacks. Extensive experiments demonstrate that $\textit{ADPO}$ outperforms baselines in the safety alignment and general utility of VLMs.
title Adversary-Aware DPO: Enhancing Safety Alignment in Vision Language Models via Adversarial Training
topic Cryptography and Security
url https://arxiv.org/abs/2502.11455