Concept-Based Dictionary Learning for Inference-Time Safety in Vision Language Action Models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Wen, Siqi, Yang, Shu, Fu, Shaopeng, Zhang, Jingfeng, Hu, Lijie, Wang, Di
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911537112809472
author Wen, Siqi
Yang, Shu
Fu, Shaopeng
Zhang, Jingfeng
Hu, Lijie
Wang, Di
author_facet Wen, Siqi
Yang, Shu
Fu, Shaopeng
Zhang, Jingfeng
Hu, Lijie
Wang, Di
contents Vision Language Action (VLA) models close the perception action loop by translating multimodal instructions into executable behaviors, but this very capability magnifies safety risks: jailbreaks that merely yield toxic text in LLMs can trigger unsafe physical actions in embodied systems. Existing defenses alignment, filtering, or prompt hardening intervene too late or at the wrong modality, leaving fused representations exploitable. We introduce a concept based dictionary learning framework for inference time safety control. By learning sparse, interpretable dictionaries from hidden activations, our method identifies harmful concept directions and attenuates risky components when the estimated risk exceeds a threshold. Experiments on Libero-Harm, BadRobot, RoboPair, and IS-Bench show that our approach achieves state-of-the-art defense performance, cutting attack success rates by over 70\% while maintaining task success. Crucially, the framework is plug-in and model-agnostic, requiring no retraining and integrating seamlessly with diverse VLAs. To our knowledge, this is the first inference time concept based safety method for embodied systems, advancing both interpretability and safe deployment of VLA models.
format Preprint
id arxiv_https___arxiv_org_abs_2602_01834
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Concept-Based Dictionary Learning for Inference-Time Safety in Vision Language Action Models
Wen, Siqi
Yang, Shu
Fu, Shaopeng
Zhang, Jingfeng
Hu, Lijie
Wang, Di
Robotics
Vision Language Action (VLA) models close the perception action loop by translating multimodal instructions into executable behaviors, but this very capability magnifies safety risks: jailbreaks that merely yield toxic text in LLMs can trigger unsafe physical actions in embodied systems. Existing defenses alignment, filtering, or prompt hardening intervene too late or at the wrong modality, leaving fused representations exploitable. We introduce a concept based dictionary learning framework for inference time safety control. By learning sparse, interpretable dictionaries from hidden activations, our method identifies harmful concept directions and attenuates risky components when the estimated risk exceeds a threshold. Experiments on Libero-Harm, BadRobot, RoboPair, and IS-Bench show that our approach achieves state-of-the-art defense performance, cutting attack success rates by over 70\% while maintaining task success. Crucially, the framework is plug-in and model-agnostic, requiring no retraining and integrating seamlessly with diverse VLAs. To our knowledge, this is the first inference time concept based safety method for embodied systems, advancing both interpretability and safe deployment of VLA models.
title Concept-Based Dictionary Learning for Inference-Time Safety in Vision Language Action Models
topic Robotics
url https://arxiv.org/abs/2602.01834