Search papers, labs, and topics across Lattice.
This paper introduces feature-observability and feature-controllability as concepts for mechanistic interpretability in Vision-Language-Action Models (VLAs). Feature-observability is achieved by training linear classifiers to detect features in the representation space, while feature-controllability uses linear interventions based on optimal control to steer the VLA's output. Experiments on $\pi_{0.5}$ and OpenVLA show that these lightweight interventions can reliably steer robot behavior in simulation without fine-tuning, enabling real-time adaptation.
Forget retraining: you can steer a robot's behavior in real-time by nudging its internal representations with lightweight, targeted interventions.
Vision-Language-Action Models (VLAs) have shown remarkable progress towards embodied intelligence. While their architecture partially resembles that of Large Language Models (LLMs), VLAs exhibit higher complexity due to their multi-modal inputs/outputs and often hybrid nature of transformer and diffusion heads. This is part of the reason why insights from mechanistic interpretability in LLMs, which explain how the internal model representations relate to their output behavior, do not trivially transfer to VLA counterparts. In this work, we propose to close this gap by introducing and analyzing two main concepts: feature-observability and feature-controllability. In particular, we first study features that are linearly encoded in representation space, and show how they can be observed by means of a linear classifier. Then, we use a minimal linear intervention grounded in optimal control to accurately place internal representations and steer the VLA's output towards a desired region. Our results show that targeted, lightweight interventions can reliably steer a robot's behavior while preserving closed-loop capabilities. We demonstrate on different VLA architectures ($\pi_{0.5}$ and OpenVLA) through simulation experiments that VLAs possess interpretable internal structure amenable to online adaptation without fine-tuning, enabling real-time alignment with user preferences and task requirements.
