SPA-VL: A Comprehensive Safety Preference Alignment Dataset for Vision Language Models

Yongting Zhang, Luyao Chen, Guodong Zheng, Yifeng Gao, Rui Zheng +8 more
6/1/2024

No official code found. Best implementation: EchoseChen/SPA-VL-RLHF (48 stars, MIT, last commit 2024-06-24).

Abstract

The emergence of Vision Language Models (VLMs) has brought unprecedented advances in understanding multi-modal information. The combination of textual and visual semantics in VLMs is highly complex and diverse, making the safety alignment of these models challenging. Furthermore, due to the limited study on the safety alignment of VLMs, there is a lack of large-scale, high-quality datasets. To address these limitations, we propose a Safety Preference Alignment dataset for Vision Language Models named SPA-VL. In terms of breadth, SPA-VL covers 6 harmfulness domains, 13 categories, and 53 subcategories, and contains 100,788 samples of the quadruple (question, image, chosen response, rejected response). In terms of depth, the responses are collected from 12 open-source (e.g., QwenVL) and closed-source (e.g., Gemini) VLMs to ensure diversity. The construction of preference data is fully automated, and the experimental results indicate that models trained with alignment techniques on the SPA-VL dataset exhibit substantial improvements in harmlessness and helpfulness while maintaining core capabilities. SPA-VL, as a large-scale, high-quality, and diverse dataset, represents a significant milestone in ensuring that VLMs achieve both harmlessness and helpfulness.

View on arXivView PDF

Code Implementations(1)

MIT

The reinforcement learning codes for dataset SPA-VL

480Shell, PythonJun 6, 20242 years agoMIT

Cite this paper

@article{zhang2024spavl,
  title  = {SPA-VL: A Comprehensive Safety Preference Alignment Dataset for Vision Language Models},
  author = {Yongting Zhang and Luyao Chen and Guodong Zheng and Yifeng Gao and Rui Zheng and Jinlan Fu and Zhen-fei Yin and Senjie Jin and Yu Qiao and Xuanjing Huang and Feng Zhao and Tao Gui and Jing Shao},
  year   = {2024},
  eprint = {2406.12030},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2406.12030},
  journal = {CVPR 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.12030.svg)](https://codeofpaper.com/paper/2406.12030)

Discussion