SAE weight editing steers robot behavior at inference time, IIIT Hyderabad paper lands IJCAI Safe Physical AI workshop

ponguru · x · 2026-10-18

A paper from IIIT Hyderabad, Controlling Robot Behavior at Inference Time: Weights based SAE steering in Vision Language Action models, was accepted at the IJCAI Safe Physical AI Workshop. The project began in the authors' Spring 2026 Responsible & Safe AI course.

The method works in three stages: training a Matryoshka SAE on the FFN of layer 31 of a VLA model to get a sparse latent space; projecting FFN vectors into vocabulary space via the LM head to select SAE features tied to concepts like "task", "slow", "push" and "destroy"; and applying signed feature amplification at inference time, reconstructing the steered FFN weights as a temporary weight edit.

In simulated manipulation experiments, steering reduced execution time and improved task success rate, offering a way to control robot behavior safely without retraining. A 5-minute podcast discussion of the paper is also available.

Original post →

More from Embodied

Embodied channel →