SaFeR-ToolKit: Structured Reasoning via Virtual Tool Calling for Multimodal Safety

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Xu, Zixuan, He, Tiancheng, Yi, Huahui, Wang, Kun, Chen, Xi, Xi, Gongli, Li, Qiankun, Li, Kang, Liu, Yang, Zeng, Zhigang
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866908861768663040
author Xu, Zixuan
He, Tiancheng
Yi, Huahui
Wang, Kun
Chen, Xi
Xi, Gongli
Li, Qiankun
Li, Kang
Liu, Yang
Zeng, Zhigang
author_facet Xu, Zixuan
He, Tiancheng
Yi, Huahui
Wang, Kun
Chen, Xi
Xi, Gongli
Li, Qiankun
Li, Kang
Liu, Yang
Zeng, Zhigang
contents Vision-language models remain susceptible to multimodal jailbreaks and over-refusal because safety hinges on both visual evidence and user intent, while many alignment pipelines supervise only the final response. To address this, we present SaFeR-ToolKit, which formalizes safety decision-making as a checkable protocol. Concretely, a planner specifies a persona, a Perception $\to$ Reasoning $\to$ Decision tool set, and a constrained transition graph, while a responder outputs a typed key-value tool trace before the final answer. To make the protocol reliably followed in practice, we train a single policy with a three-stage curriculum (SFT $\to$ DPO $\to$ GRPO), where GRPO directly supervises tool usage beyond answer-level feedback. Our contributions are two-fold: I. Dataset. The first tool-based safety reasoning dataset, comprising 31,654 examples (SFT 6k, DPO 18.6k, GRPO 6k) plus 1k held-out evaluation. II. Experiments. On Qwen2.5-VL, SaFeR-ToolKit significantly improves Safety/Helpfulness/Reasoning Rigor on 3B (29.39/45.04/4.98 $\to$ 84.40/71.13/78.87) and 7B (53.21/52.92/19.26 $\to$ 86.34/80.79/85.34), while preserving general capabilities (3B: 58.67 $\to$ 59.21; 7B: 66.39 $\to$ 66.81). Codes are available at https://github.com/Duebassx/SaFeR_ToolKit.
format Preprint
id arxiv_https___arxiv_org_abs_2603_02635
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle SaFeR-ToolKit: Structured Reasoning via Virtual Tool Calling for Multimodal Safety
Xu, Zixuan
He, Tiancheng
Yi, Huahui
Wang, Kun
Chen, Xi
Xi, Gongli
Li, Qiankun
Li, Kang
Liu, Yang
Zeng, Zhigang
Machine Learning
Vision-language models remain susceptible to multimodal jailbreaks and over-refusal because safety hinges on both visual evidence and user intent, while many alignment pipelines supervise only the final response. To address this, we present SaFeR-ToolKit, which formalizes safety decision-making as a checkable protocol. Concretely, a planner specifies a persona, a Perception $\to$ Reasoning $\to$ Decision tool set, and a constrained transition graph, while a responder outputs a typed key-value tool trace before the final answer. To make the protocol reliably followed in practice, we train a single policy with a three-stage curriculum (SFT $\to$ DPO $\to$ GRPO), where GRPO directly supervises tool usage beyond answer-level feedback. Our contributions are two-fold: I. Dataset. The first tool-based safety reasoning dataset, comprising 31,654 examples (SFT 6k, DPO 18.6k, GRPO 6k) plus 1k held-out evaluation. II. Experiments. On Qwen2.5-VL, SaFeR-ToolKit significantly improves Safety/Helpfulness/Reasoning Rigor on 3B (29.39/45.04/4.98 $\to$ 84.40/71.13/78.87) and 7B (53.21/52.92/19.26 $\to$ 86.34/80.79/85.34), while preserving general capabilities (3B: 58.67 $\to$ 59.21; 7B: 66.39 $\to$ 66.81). Codes are available at https://github.com/Duebassx/SaFeR_ToolKit.
title SaFeR-ToolKit: Structured Reasoning via Virtual Tool Calling for Multimodal Safety
topic Machine Learning
url https://arxiv.org/abs/2603.02635