Papers

SAE 特征替换可劫持 π0.5 机器人模型的目标指令

Editing the Goal: Per-Token SAE Steering and Its Depth Limits in $π_{0.5}$

Why we picked it

改掉一个词的 SAE 特征就能让 π0.5 机器人改做别的任务,20/20 全成功,还标出了哪几层改得动。

研究在机器人基础模型 π0.5 的 LIBERO 基准上测试稀疏自编码器(SAE)特征的因果操控。在第 0 层将单个目标词的 BatchTopK 特征替换为另一目标词的特征,最多在 20/20 次rollout 中改变机器人执行的任务,而随机特征替换无一成功。效果随层级衰减:目标词编辑对物体位置有效到第 6 层、对物体身份有效到第 2 层,若同时覆盖相机 token 则两者都有效到第 13 层。不依赖 SAE 也行:加入两倍自然幅度的均值差向量即可重定向 8-10/10 次rollout。

Original · arXiv cs.AI

Editing the Goal: Per-Token SAE Steering and Its Depth Limits in $π_{0.5}$

Sparse autoencoders (SAEs) have recovered interpretable, causally relevant features in robot foundation models (RFMs) such as $π_{0.5}$ (arXiv:2603.19183, arXiv:2603.19233). SAE features on individual instruction tokens are less interpretable than timestep-pooled ones (arXiv:2603.19183), and whether they can redirect a robot's goal remains untested. We show that replacing the SAE features of a single goal word with those of an alternative goal word, leaving the instruction text unchanged, steers $π_{0.5}$-LIBERO into completing the alternative task. At layer 0, BatchTopK feature swaps redirect behavior in up to 20/20 rollouts in a scene where random-feature swaps redirect none, with success varying across six scenes in two LIBERO suites and across layers. The effect depends on SAE architecture: word-selected Matryoshka edits largely do not outperform random ones. SAEs are also not required. Adding a fixed difference-in-means vector at twice its natural magnitude redirects 8-10/10 rollouts. Layer-wise activation patching shows that goal-word edits remain effective up to layer 6 for object location and layer 2 for object identity, and up to layer 13 in both cases when patching also covers camera tokens. These results are proof of existence in an RFM in simulation and an initial map of how deep goal-redirecting edits remain effective. Visit https://doubleblind-sae-pi05.github.io for video rollouts.