Domain Arithmetic: One-Shot VLA Adaptation under Environmental Shifts¶
会议: ECCV 2026
arXiv: 2607.00666
代码: https://github.com/snumprlab/dart
领域: 多模态VLM / 机器人
关键词: VLA适配, 权重算术, 域迁移, one-shot学习, 子空间对齐
一句话总结¶
DART 提出一种基于权重空间类比运算的 VLA 模型 one-shot 域适配方法:用单个源域+目标域 demonstration 分别微调,通过子空间对齐过滤后相减提取纯域向量,再将其加回基座模型,在 LIBERO 多视角、视觉扰动、跨实体和真实机器人场景下,仅用 1 条 demonstrations 就显著超越现有适配基线。
研究背景与动机¶
VLA 模型(如 RT-2、OpenVLA、π 系列)在大规模机器人数据上预训练后展现出强大的多任务能力,但一旦部署到与训练环境存在域偏移(相机位姿变化、光照差异、传感器标定不同、甚至换成不同型号的机械臂)的新场景中,同一任务的性能会急剧下降。已有适配方法要么需要为目标域每个任务收集大量 demonstrations(全量微调成本极高),要么引入特定架构修改(如仅调视觉编码器、加 LoRA),限制了对不同 VLA 骨干的通用性。
这构成了一个尖锐矛盾:真实部署中域偏移无处不在,但逐任务采集 demonstrations 在家庭等场景根本不可行。于是作者把目标推向极限——one-shot VLA 适配:只用目标域中一个任务的一条 demonstration,就将基座模型在目标域上恢复出全部已学任务的执行能力。
直觉是:虽然 one-shot 微调后模型参数的变化(update-vector)主要由任务特异的方向主导(因而对 held-out 任务泛化差),但这些 update-vector 中同时存在一个更小但可复用的域特异方向——它编码了"目标域长什么样"的信息。作者通过子空间对齐分析发现:同一任务跨域的 update-vector 高度对齐,同一域跨任务的 update-vector 也有高于随机的对齐,并且 task/domain 方向在权重空间中近似可加性分解。这直接启发了 DART 的核心 insight:用类比运算(target update minus source update)消去任务方向,仅保留域方向,再将其注入基座模型——如同 "queen = king + woman - man" 在权重空间的翻版。
方法详解¶
整体框架¶
DART 的目标是:给定在源域 ε_src 上训练好的多任务 VLA 基座模型 θ₀,以及目标域 ε_tgt 中单个任务 T_m 的一条 demonstration D_{m,tgt},产出适配参数 θ 使 θ 在目标域全部任务上表现良好。整体流程分三步:
- 分别 one-shot 微调:用同一任务 T_m 的源域演示 D_{m,src} 和目标域演示 D_{m,tgt} 分别对 θ₀ 做行为克隆微调(各 1,000 步),得到 θ_{m,src} 和 θ_{m,tgt},进而算出两个 update-vector:Δ_{m,src} = θ_{m,src} − θ₀,Δ_{m,tgt} = θ_{m,tgt} − θ₀。
- 子空间增强的域向量提取:对每一层的 Δ_{m,src} 和 Δ_{m,tgt} 做 SVD 分解,计算子空间对齐分数 γ^(l),通过重叠能量筛选仅保留与目标子空间对齐的源基底(子空间过滤),再用 γ^(l) 对域向量做逐层降权(子空间缩放),得到精炼域向量 δ̃_tgt。
- 域向量注入:将域向量加回基座模型 θ* = θ₀ + α·δ̃_tgt,得到目标域适配模型,直接用于全部任务推理。
%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
A["基座模型 θ₀<br/>(源域多任务)"] --> B["源域 one-shot 微调<br/>θ_{m,src}"]
A --> C["目标域 one-shot 微调<br/>θ_{m,tgt}"]
B --> D["Δ_{m,src} = θ_{m,src} − θ₀"]
C --> E["Δ_{m,tgt} = θ_{m,tgt} − θ₀"]
D --> F["SVD + 子空间过滤<br/>仅保留与目标对齐的源基底"]
E --> F
F --> G["类比减法 + 子空间缩放<br/>δ̃_tgt = γ · (Δ_{m,tgt} − Δ̃_{m,src})"]
G --> H["θ* = θ₀ + α·δ̃_tgt<br/>适配完成,目标域多任务推理"]
关键设计¶
1. 类比减法提取域向量:通过源-目标 update-vector 对减消除任务方向
作者在第 4 节证明,one-shot update-vector Δ_{m,tgt} 中任务特异方向占主导(同一任务跨域 γ 高),但也存在微弱但一致的域方向(同域跨任务 γ 略高于跨域跨任务),且两者在权重空间近似线性可加(任务原型 + 域原型 − 全局原型 的合成方向与真实 update-vector 的 γ 最高)。基于此,DART 用一个巧妙的类比操作来解耦:设同一任务 T_m 在源域的 update-vector 为 Δ_{m,src},目标域为 Δ_{m,tgt},则域向量 δ_tgt = Δ_{m,tgt} − Δ_{m,src}。因为两个 update-vector 共享任务特定方向,相减后任务分量被抵消,残留的就是纯域迁移信息。这个设计的精妙之处在于:只需分别用源域和目标域的同一条任务演示各微调一次,无需任何多任务数据或额外标注。实操上,可以先从源训练集中选定 T_m,再到目标域采集同任务的演示,保证任务匹配。
2. 子空间过滤:只减去与目标子空间对齐的源基底向量,防止源域噪声污染域向量
直接减法有一个隐患:Δ_{m,src} 中可能含有源域特有的伪影(与任务无关的微调噪声),盲目减去会将这些噪声注入域向量。DART 的做法是只过滤源 update-vector,保留目标 update-vector 的完整信息(因为目标 update 的独特基底恰好编码了域信息)。具体步骤:对每层权重矩阵 Δ_{m,src} 和 Δ_{m,tgt} 分别 SVD,得到左奇异向量 U_src 和 U_tgt;构造交互矩阵 C = U_tgt^⊤ U_src,第 j 个源基底的重叠能量 e_j = ||C_{:,j}||²₂ 衡量该基底落在目标子空间中的比例;以子空间对齐分数 γ 为阈值,贪心地选取能量和达到 γ·Σe_j 的基底索引集合 J_l,构造过滤后的源 update-vector Δ̃{m,src} = U_src[:,J_l] · U_src[:,J_l]^⊤ · Δ。这与模型合并中的子空间对齐方法方向相反——后者保留每个 update-vector 的独特成分以最大化能力,DART 则是找共有的任务相关成分以精确消除。
3. 子空间缩放:对低对齐层降权,抑制信噪比低的域向量
即使过滤了不对齐的基底,如果两个 update-vector 从根本上就弱对齐(γ→0),减法产生的域向量中噪声占比仍然很高。DART 直接用对齐分数 γ^(l) 对域向量逐层缩放:δ̃tgt^(l) = γ^(l) · (Δ^(l))。直观上,γ 高的层说明 source 和 target 共享大量任务结构,减法效果好、域向量干净,保留完整强度;γ 低的层说明两域差异大或更新本身噪声多,缩放后降低其对最终模型的影响。消融实验表明,子空间缩放带来的增益主要来自高度失对齐的层(如 LLM 中的 MLP Up_proj 层),这些层的 γ 接近 0,其域向量若不加降权反而会伤害多任务能力。}^(l) − Δ̃_{m,src
损失函数 / 训练策略¶
One-shot 微调阶段使用标准行为克隆(BC)损失,在目标域演示 D_{m,tgt} 的动作上最小化负对数似然,优化器为 AdamW,学习率 5×10⁻⁵,无 warmup,固定训练 1,000 步,batch size 64。基座模型按照各 VLA 的官方协议预训练(如 LIBERO 上 π₀.₅ 用 openpi 发布的 checkpoint,π₀-FAST 从头训练 30,000 步)。域向量提取阶段不涉及额外训练,仅做权重空间算术;缩放系数 α 在 LIBERO Long 子集上用 10 rollouts 小搜索确定(α=0.8 用于视角迁移,α=0.6 用于视觉扰动),并在所有其他场景(含真实机器人)中复用同一值。SVD 计算可用随机化 SVD(rank r=256)将全量分解 15 分 35 秒降至 6 分 33 秒,性能几乎无损(79.1% vs 78.7%)。
实验关键数据¶
主实验¶
DART 在 π₀.₅ 上的 LIBERO 新视角迁移结果(40 任务,3 次随机 adaptation task 平均,每次 50 rollouts):
| 方法 | Small | Medium | Large | 平均 |
|---|---|---|---|---|
| Zero-shot | 88.3 | 63.9 | 11.3 | 54.5 |
| One-shot FT | 43.4 | 33.3 | 17.8 | 31.5 |
| RETAIN (ICLR 2026) | 87.4 | 72.4 | 48.9 | 69.6 |
| FLA (CVPR 2026) | 92.2 | 76.4 | 54.3 | 74.3 |
| DART (Ours) | 92.0 | 80.8 | 64.4 | 79.1 |
真实机器人 UR10e 五任务视角迁移结果(仅用一条 Stack Cube 演示适配,每任务 12 rollouts):
| 方法 | Eggplant | Lemon | Carrot | Stack Cube | Press Stapler | 平均 |
|---|---|---|---|---|---|---|
| Zero-shot | 50.0 | 33.3 | 41.7 | 16.7 | 75.0 | 43.3 |
| One-shot FT | 58.3 | 58.3 | 41.7 | 33.3 | 66.7 | 51.7 |
| RETAIN | 58.3 | 41.7 | 41.7 | 16.7 | 83.3 | 48.3 |
| FLA | 58.3 | 50.0 | 50.0 | 16.7 | 100.0 | 55.0 |
| DART | 91.7 | 91.7 | 83.3 | 41.7 | 100.0 | 81.7 |
其他主实验结果:跨实体迁移(Panda→UR5e, MimicGen),DART 平均成功率 69.4%(vs Zero-shot 62.0%);π₀-FAST 架构上 DART 平均 79.4%(vs FLA 76.6%);组合视觉扰动(View+Noise+Light)下 DART 平均 75.0%(vs FLA 71.5%)。
消融实验¶
LIBERO 三视角平均成功率消融(π₀.₅):
| 子空间过滤 | 子空间缩放 | 平均成功率 (%) |
|---|---|---|
| ✗ | ✗ | 78.1 |
| ✓ | ✗ | 78.8 |
| ✗ | ✓ | 78.5 |
| ✓ | ✓ | 79.1 |
相比 One-shot FT 的 31.5%,纯类比减法(无子空间组件)即达到 78.1%(+46.6pp),证明类比运算本身就能有效隔离域知识。子空间过滤进一步提升到 78.8%(+0.7pp),子空间缩放到 78.5%(+0.4pp),二者叠加达到 79.1%,说明过滤和缩放分别贡献于抑制源域噪声和降低失对齐层的干扰。
关键发现¶
- 类比运算是最核心的贡献:仅靠 Δ_{m,tgt} − Δ_{m,src} 就在三视角平均上获得 +46.6pp 收益(相对 One-shot FT),证明 one-shot update-vector 中 task/domain 方向的加性分解假说成立。
- 缩放系数 α 极其鲁棒:在 Medium 和 Large 视角下 α 从 0.2 到 1.4 的波动仅带来 0.8% 和 2.1% 的标准差,DART w/o SA 则波动更大,说明子空间过滤+缩放抑制了会被 α 放大的噪声成分。
- 微调步数宽容:One-shot FT 随步数增加因灾难性遗忘而退化,DART 却稳步上升(因任务方向随训练更显著,有利于提取纯净域向量),且 200 步即可达到竞争力性能。
- LLM 层适配是关键:在 Vis/LLM/Action 三层中,Vis+LLM 效果接近全层,单独适配 Action 几乎无效,且 LLM 中 MLP Up_proj 层的域向量幅值最大、子空间对齐最低——提示域知识主要编码在 LLM 的 MLP 层。
- 域向量可合并:将 Small/Medium/Large 三个视角的域向量合并为一个,用 TSV 合并后平均成功率 75.7%,可减少多域部署的分量存储开销。
亮点与洞察¶
- 用"类比"而非"合并"做域适配,问题定义精准:模型合并(如 TIES、Iso-C)的目标是组合不同模型的能力,类比(analogy)的目标是消去共性、提取差异——这正好是域适配所需要的。DART 是第一个在 VLA 中系统使用类比做域迁移的工作,实验也表明直接套用合并方法(如 TIES 合并 [Δ_{m,tgt}, −Δ_{m,src}])效果不如 DART(77.6% vs 79.1%)。
- 子空间对齐分数一石二鸟:γ 既用作基底选择的阈值(决定过滤多少),又用作域向量的缩放系数(决定信噪比低的层降权多少),没有引入额外超参数。这个设计简洁且高效,背后是对 update-vector 低秩结构和任务/域方向几何关系的深刻理解。
- 源域演示的获取假设实际可满足:作者明确指出可以先从源训练集选好 adaptation task,再到目标域采集同任务的演示——不是要求在庞大的目标域数据里找出和源域"同样的任务",而是反过来做,这个工程细节让假设在实际部署中可行。
- 方法对 VLA 架构不敏感:π₀.₅(flow matching 连续回归)和 π₀-FAST(自回归离散 token)均有效,说明 weight-space 的 task/domain 加性分解是 VLA 模型的通用结构属性,根植于其共享的视觉-语言预训练骨干。
局限与展望¶
- 作者承认大视角偏移(Large viewpoint)下所有 one-shot 方法性能都有限(DART 64.4% vs full-data FT 上界 90.9%),极端域偏移仍然是挑战,未来可探索更强的域向量提取机制或更好的基座模型微调策略。
- 缩放系数 α 虽在较宽范围内鲁棒,但仍需要一个小的搜索过程(10 rollouts),对真实部署中完全无搜索的场景还不够"即插即用";逐层自适应缩放可能是解法。
- 该方法要求源域和目标域的 adaptation task 相同(T_m = T_m')。当无法精确匹配时(如源数据集庞大且无清晰任务索引),使用相似任务(余弦相似度检索)可部分缓解但仍有明显退化(80.8% → 69.0%),说明域向量提取对任务匹配敏感;更强的跨域任务检索是一个有价值的改进方向。
- 当前实验集中在视觉域偏移和跨实体迁移,对动力学变化(如物体质量、摩擦力、关节阻尼等物理参数偏移)的效果未验证,这类偏移是否同样在权重空间形成线性可分的域方向值得探索。
相关工作与启发¶
- vs Task Arithmetic (ICLR 2023):TA 使用加法和类比做多任务能力的组合,但其类比仅做直接减法用于跨语言迁移和对齐迁移。DART 在 TA 的类比基础上引入了子空间过滤和缩放两个增强机制,并将类比重新定位为"域信息提取"而非"能力组合",这在 VLA 适配场景中获得显著收益。
- vs RETAIN (ICLR 2026):RETAIN 通过模块级插值权衡新旧任务能力,本质是参数空间线性插值,不显式建模域偏移方向。DART 的类比减法显式提取域向量,在相同 one-shot 数据预算下取得更高成功率(79.1% vs 69.6%),说明显式建模优于隐式权衡。
- vs FLA (CVPR 2026):FLA 仅在视觉编码器中插入 LoRA 进行参数高效适配,冻结 LLM 和 action expert。DART 通过权重算术同时适配所有模块,在数据极度匮乏(one-shot per scene vs task-wise)的设置下性能更好,且不引入任何架构修改,即插即用于任意 VLA。
- vs SCALE (ICML 2026, test-time adaptation):SCALE 在推理时适应但主要面向有限偏移。DART 在 π₀-FAST 上的三视角平均 79.4% vs SCALE 73.0%,说明在较大环境偏移下,test-time adaptation 的能力有限,对 VLA 策略参数的显式适配仍是必要的。
评分¶
- 新颖性: ⭐⭐⭐⭐⭐ 将 weight arithmetic 中的类比运算首次系统应用于 VLA 域适配,配合子空间对齐增强,问题定义和方法设计都很有新意
- 实验充分度: ⭐⭐⭐⭐⭐ 覆盖两个 VLA 架构、模拟+真实机器人、视觉偏移+跨实体、消融+超参分析+层选择+域向量合并+源域遗忘检查,实验设计周全
- 写作质量: ⭐⭐⭐⭐⭐ 从第 4 节实证分析(为什么 one-shot FT 失败)到第 5 节方法设计(基于分析发现提出 DART),逻辑链条紧密,动机充分自洽
- 价值: ⭐⭐⭐⭐⭐ one-shot VLA 适配是机器人部署的核心痛点,DART 数据效率极高(1 条 demo per environment)、架构无关、即插即用,实用价值突出