Saved in:
Bibliographic Details
Main Authors: Yang, Wenhan, Stice, Spencer, Payani, Ali, Mirzasoleiman, Baharan
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2505.24208
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913866837917696
author Yang, Wenhan
Stice, Spencer
Payani, Ali
Mirzasoleiman, Baharan
author_facet Yang, Wenhan
Stice, Spencer
Payani, Ali
Mirzasoleiman, Baharan
contents Ensuring Vision-Language Models (VLMs) generate safe outputs is crucial for their reliable deployment. However, LVLMs suffer from drastic safety degradation compared to their LLM backbone. Even blank or irrelevant images can trigger LVLMs to generate harmful responses to prompts that would otherwise be refused in text-only contexts. The modality gap between image and text representations has been recently hypothesized to contribute to safety degradation of LVLMs. However, if and how the amount of modality gap affects LVLMs' safety is not studied. In this work, we show that the amount of modality gap is highly inversely correlated with VLMs' safety. Then, we show that this modality gap is introduced during pretraining LVLMs and persists through fine-tuning. Inspired by this observation, we propose a regularization to reduce the modality gap during pretraining. Our extensive experiments on LLaVA v1.5, ShareGPT4V, and MiniGPT-4 show that our method substantially improves safety alignment of LVLMs, reducing unsafe rate by up to 16.3% without compromising performance, and can further boost existing defenses by up to 18.2%.
format Preprint
id arxiv_https___arxiv_org_abs_2505_24208
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Bootstrapping LLM Robustness for VLM Safety via Reducing the Pretraining Modality Gap
Yang, Wenhan
Stice, Spencer
Payani, Ali
Mirzasoleiman, Baharan
Artificial Intelligence
Ensuring Vision-Language Models (VLMs) generate safe outputs is crucial for their reliable deployment. However, LVLMs suffer from drastic safety degradation compared to their LLM backbone. Even blank or irrelevant images can trigger LVLMs to generate harmful responses to prompts that would otherwise be refused in text-only contexts. The modality gap between image and text representations has been recently hypothesized to contribute to safety degradation of LVLMs. However, if and how the amount of modality gap affects LVLMs' safety is not studied. In this work, we show that the amount of modality gap is highly inversely correlated with VLMs' safety. Then, we show that this modality gap is introduced during pretraining LVLMs and persists through fine-tuning. Inspired by this observation, we propose a regularization to reduce the modality gap during pretraining. Our extensive experiments on LLaVA v1.5, ShareGPT4V, and MiniGPT-4 show that our method substantially improves safety alignment of LVLMs, reducing unsafe rate by up to 16.3% without compromising performance, and can further boost existing defenses by up to 18.2%.
title Bootstrapping LLM Robustness for VLM Safety via Reducing the Pretraining Modality Gap
topic Artificial Intelligence
url https://arxiv.org/abs/2505.24208