精读笔记

Problem Setting

这篇论文实际解决的是 VLA policy 的内部行为控制问题：给定同一个任务目标，不重新训练模型，也不改语言指令，而是通过干预内部表示来改变机器人执行任务的方式。这个问题比普通 policy conditioning 更窄，也更难，因为目标不是提升 success rate，而是在保持任务可完成的同时操控动作生成过程中的行为维度。

真正困难点在于 VLA 的“行为”不是一个干净的语义特征。它同时绑定视觉场景、物体几何、时间步、动作噪声、接触状态和语言目标。以前 LLM/VLM 中常用的 activation addition 默认存在一个稳定、全局、可加的行为方向；但在 VLA 中，这个方向即使能被 probe 找到，也未必是可写入的控制变量。关键矛盾是：表示空间里有行为信息，不等于沿该方向移动就能生成可执行的行为。

Motivation

已有路线不够的根源是它们把 steering 当成 linear feature editing：先构造正负样本差，得到一个方向，再在推理时加到 activation 上。这在语言模型里有时成立，因为很多行为偏好表现为 logits 或 residual stream 的近似线性偏置。但机器人动作生成是连续控制问题，内部表示必须落在能被 action expert 解码为可行动作的区域。

作者的核心观察是“linearly decodable but not linearly steerable”。这句话是整篇论文的动机核心：probe 能读出行为属性，只说明表示中存在相关信息；steering 要求干预后模型的后续动力学沿目标行为演化。二者不是同一件事。缺口因此不是再找更好的 linear direction，而是需要一种尊重 activation distribution 几何的写入机制。

Core Idea

DiMaS 的真正核心是把 steering 从“沿方向平移”改成“源行为分布到目标行为分布的 transport”。它不假设行为是一个一维线性轴，而是假设不同执行风格对应 action expert 表示空间中的不同条件分布。推理时，对当前 activation 做 distribution matching，使其更像目标行为下模型本来会产生的 activation。

这个改变引入了更合适的 inductive bias：控制不是修改一个 semantic scalar，而是把表示重新放回目标行为的可行流形。对于 flow-matching VLA，这尤其自然，因为模型本身就在通过连续流建模动作分布；DiMaS 相当于在 action generation 的 latent trajectory 上做分布对齐，而不是在输出动作或文本层面做后验修补。和 prior 的本质差异不是“用了 OT”这类工具细节，而是把 steerability 建模为分布几何问题。

Method

方法可以压缩成三个必要机制。

第一，收集带行为差异的 activation distributions。它解决的是 steering 监督从哪里来的问题：不是只用正负 prompt 差，而是用真实或离线轨迹中目标行为和非目标行为对应的内部表示。这样得到的是行为在 action expert 中的经验分布，而不是单个 contrastive direction。

第二，学习源分布到目标分布的 transport。它解决的是如何写入行为的问题。固定向量只能移动均值，容易把 activation 推出模型熟悉的动作表示区域；transport 试图保留样本依赖的几何关系，使每个状态下的 activation 被移动到更合理的目标区域。

第三，在 flow-matching VLA 的 action expert 内部应用 steering。它解决的是干预位置的问题。行为最终由动作生成专家决定，因此只改语言侧或最终动作后处理都太间接；在 action expert 表示中干预，等于改写动作流生成过程中的中间条件，而不是事后修正输出。

Key Insight / Why It Works

最重要的 insight 是：可解释性里的“可线性读出”不等于控制里的“可线性写入”。这对 VLA 特别关键，因为 action expert 的 hidden state 不是纯语义表征，而是面向连续控制的状态条件表示。一个 linear probe 可能只是利用了分布中某个方向的相关性；沿这个方向加 activation 可能破坏和视觉状态、动作时间、噪声条件之间的协方差结构。

DiMaS 可能有效的真正原因是 representation alignment，而不是简单 scaling。它用目标行为分布作为 anchor，把当前 activation 拉向模型已经见过、能被下游 action head 消化的区域。这比 activation addition 更像 manifold projection 或 conditional style transfer。核心贡献应是 distribution-level steering 这个 inductive bias；具体 transport 算法、层选择和超参数大概率是辅助。

但需要警惕：如果训练和测试任务高度相似，增益可能主要来自 data coverage 和局部插值，而不是泛化行为控制。所谓跨任务 steering 也可能只是共享对象布局、动作模板或 LIBERO-style benchmark overlap 下的隐式记忆复用。文中提到任务越不相似迁移越弱，这反而说明 DiMaS 的能力边界主要由 activation distribution support 决定，而不是形成了抽象、可组合的行为概念。

Relation To Prior Work

它最接近三条线：LLM/VLM representation engineering，activation transport / diffusion-model activation control，以及 VLA mechanistic interpretability。和 Panickssery、Turner、Zou、Arditi 这类 linear steering 的差异在于，DiMaS 不再把行为压成单方向；和 Rodriguez 等 activation transport 的关系更近，本质上是把 transport activation 的思想移植到 flow-matching VLA 的 action expert。

看似新的部分不是“用分布匹配控制模型”本身，而是把这个框架放到 VLA 行为控制里，并指出机器人控制中 linear decodability 与 steerability 的断裂。实质创新在于问题定位和机制匹配：VLA 的行为 steering 需要处理条件动作分布，而不是语言偏好方向。它属于 representation steering 从 linear feature editing 向 geometry-aware activation editing 演化的一步。

Dataset / Evaluation

从给定材料看，论文声称在两个 SOTA flow-matching VLA 上验证，并考察 learned steering 在任务相似度逐步降低时的迁移。引用中出现 LIBERO、OpenVLA、SmolVLA、π0.5 等相关系统，但正文片段没有给出完整实验配置，因此无法确认具体任务覆盖、是否包含真实世界真机、是否只在仿真 benchmark 上评估。

evaluation 是否支撑核心 claim，取决于两点：一是是否同时报告行为控制强度和任务成功率，避免只是牺牲成功率换行为变化；二是是否有跨对象、跨场景、跨语言指令、跨动力学的评估。仅证明在相似任务上能 steering，不足以支持强泛化 claim。文中提到随着任务 dissimilarity 增大控制减弱，这个分析很有价值，但也暴露了方法对分布重叠的依赖。给定材料不足以判断是否存在 benchmark leakage、任务模板重叠或 hidden supervision。

Limitation

DiMaS 的核心前提是目标行为的 activation distribution 可被采样、可被估计，并且测试时 activation 落在与训练分布足够接近的区域。这使它更像 supervised behavior editing，而不是开放式行为控制。新行为、组合行为、稀有接触模式和长时序策略都可能超出它的有效范围。

另一个上限是可组合性。distribution transport 通常是成对学习或估计的：A 行为到 B 行为可以成立，不代表 B+C、A 到 C、或跨任务抽象行为都成立。多行为冲突时，多个 transport 的叠加未必仍在可行动作流形上。

增益归因不清。它可能证明的是“分布级干预优于均值方向干预”，但不一定证明模型内部存在稳定、抽象、可迁移的行为因子。核心能力可能主要来自数据覆盖；所谓泛化可能依赖 benchmark overlap。文中未充分说明真实部署中的 safety envelope、干预强度选择、失败恢复，以及当目标行为和任务成功冲突时模型如何取舍。

Takeaway

• 第一，VLA steering 不能直接套用 LLM 的 linear direction 假设；机器人行为控制需要区分 decodability 和 steerability。
• 第二，distribution matching 是一个值得迁移的范式：当目标属性绑定上下文、时间和连续控制变量时，控制应优先考虑把 activation 放回目标分布，而不是做全局平移。
• 第三，这篇论文真正推动的是 VLA interpretability 从“读出行为特征”走向“写入可执行行为”。
• 未来更值得做的是可组合 transport、OOD support 检测、因果层定位，以及真实机器人上的安全约束 steering。

一句话总结

DiMaS 是 VLA 行为 steering 从线性 activation addition 转向分布几何对齐的一步，真正贡献在于指出并利用了“行为可读出但不可线性写入”这一 visuomotor representation 的关键断裂。