← 返回 2026-11-05

DRAP:面向物理 AI 持续学习的动力学风险自适应可塑性 DRAP: Dynamics-Risk Adaptive Plasticity for Continual Learning in Physical AI

Oswaldo Ludwig 📅 2026-11-03 👍 0 2026-11-11 18:30
持续学习 机器人操作 模仿学习 灾难性遗忘 视觉-语言-动作模型

免回放持续学习:以专家轨迹扰动放大风险加权张量重要性,逐张量缩放 AdamW 学习率抑制遗忘

前置知识

灾难性遗忘与持续学习

神经网络按任务序列 A→B 顺序训练时,拟合新任务会覆写旧任务所需权重,导致旧技能性能骤降,即灾难性遗忘。持续学习研究如何在无旧数据或极少数据的约束下顺序学习多个任务并保持旧能力,主要策略包括经验回放、参数正则化、架构扩展/路由和可塑性控制。

本文整个问题设定就是 VLA 模型的免回放持续学习,DRAP 属于其中'基于重要性的可塑性控制'分支,理解这一图谱才能定位它的贡献。

视觉-语言-动作模型(VLA)与 OpenVLA

VLA 把摄像头图像、语言指令与低层机器人控制统一在单个预训练策略里。OpenVLA 是开源 7B 模型,将连续动作离散化为动作 token,用预训练视觉-语言骨干逐 token 预测;后续变体 OpenVLA-OFT 引入连续动作、动作分块与优化解码,把 LIBERO 平均成功率推到 97.1%。

DRAP 的实验载体是 OpenVLA 的一个固定子网,理解动作 token 表示才能读懂重要性估计里动作位置集合 $Q_i$ 的含义。

模仿学习的复合误差与协变量偏移

行为克隆只在专家轨迹经过的状态上训练,策略一旦出错就会到达专家数据未覆盖的状态,执行分布与训练分布错位,即协变量偏移。闭环执行中每步小误差改变后续观测,误差沿轨迹复合放大直至任务失败,Ross 与 Bagnell 的 DAgger 归约理论给出了这一放大机制的形式刻画。

这是 DRAP 动力学风险信号的理论出发点:物理 AI 里'哪里容易放大误差'决定了哪里更要牢牢记住,不能只会背梯度大小。

奇异值与李雅普诺夫间接方法

矩阵的最大奇异值 $\sigma_{\max}(A)$ 度量线性映射对向量能量的最大拉伸倍数。李雅普诺夫间接法在轨迹附近将非线性系统局部线性化,用线性化矩阵的谱结构判断局部稳定性:若扰动满足 $\delta s_{t+1}\approx A\delta s_t$,则扰动能量至多放大 $\sigma_{\max}(A)^2$ 倍。

DRAP 的风险分数 $\rho_t=\sigma_{\max}(A_t)$ 正是把这一经典思路用于演示轨迹的滑窗线性化,给每个样本标注'单步扰动放大倍数'。

EWC 与基于重要性的正则化

EWC 用对角 Fisher 信息估计每个参数对旧任务的重要度,在损失里加二次惩罚阻止重要参数偏离旧最优 $\theta^\star$;SI、MAS、RWalk 是同族变体,Benzing 证明它们与 Fisher 类量高度相关。这类方法需要存参考权重,且新任务开始时位移为零、保护为零,随偏移才逐渐生效。

EWC 是本文直接对照基线,理解其'参考锚定+位移惩罚'机制,才能看出 DRAP'即时生效、无需参考副本'的本质差异。

KappaTune 与条件数

条件数 $\kappa(W)=\sigma_{\max}/\sigma_{\min}$(正则化后)度量矩阵奇异方向的分化程度。KappaTune 假设大条件数张量的奇异方向分化强、更可能编码预训练专有信息,应予保留;因此在视觉、语言、动作头各组内选出固定预算的低 $\kappa$ 张量作为可训练子网,其余参数全部冻结。

论文消融证明去掉 KappaTune 后 DRAP 在第三个任务上崩溃(0.11/0.08),两者是'在哪适应'与'留多少可塑性'的互补分工。

AdamW 与逐张量学习率分组

AdamW 是带解耦权重衰减的 Adam 优化器,其参数组(param group)机制允许为不同张量指定不同学习率。DRAP 不修改损失函数也不用自定义优化器,只为每个被选张量建一个参数组,把基础学习率 $\eta$ 乘上可塑性乘子 $s_T=(1+\beta\bar{I}_T)^{-1/p}$,从优化器接口直接压低重要张量的步长。

这是 DRAP 全部的'保护动作'——理解参数组机制就知道该方法实现有多轻量,也明白为什么它说'抑制的是真实优化步长而非原始梯度'。

研究动机

物理 AI 的持续学习与常规持续学习有本质区别:策略是闭环执行的,一步动作错误会改变下一时刻的物理状态与观测,把策略推到训练分布之外,后续误差沿轨迹复合放大,最终即使每步预测误差很小也会导致任务失败——这正是模仿学习中协变量偏移与 DAgger 理论刻画的问题。现有对策各有硬伤:回放是最强基线但要存储旧任务演示,Liu 等人证明预训练 VLA 配合小回放几乎不遗忘,可零回放设定下其 LIBERO-Object 与 LIBERO-10 的 Sequential 和 EWC 基线仍出现大幅负向迁移;CLARE、Stellar VLA 等扩展与路由方法靠增加适配器模块或知识空间专家路由来隔离任务,改变了模型架构;EWC、SI、MAS、RWalk 等参数正则化方法用 Fisher 类重要性惩罚对旧最优解的位移,但保护随位移才逐渐生效,且完全不感知行为误差的物理后果。本文实验中,标准顺序微调在 LIBERO 上训练完 Goal 后,Spatial 成功率从 0.77 跌到 0.00,Object 从 0.83 跌到 0.07,遗忘触目惊心。

本文的目标是本文的目标是构建一种免回放的持续学习方法,让单个 OpenVLA 检查点在只见到当前套件数据的约束下,按 Spatial→Object→Goal 顺序适应 LIBERO 三个操作套件,同时最大限度保留此前习得的操作技能并保持对新套件的获取能力。核心主张是:物理 AI 的持续学习必须同时考虑模型敏感度与行为误差的物理后果——在动力学敏感的轨迹区域支撑正确行为的参数,在后续任务中应当更不可塑。量化目标上,DRAP 在 Object 阶段后把平均成功率做到 0.77、Spatial 遗忘压到 0.02;在 Goal 阶段后达到平均成功率 0.55、平均遗忘 0.31,全面优于标准微调的 0.26/0.77 与 KappaTune+EWC 的 0.32/0.63。整个方法不存任何旧套件演示,也不存参考参数副本,任务之间只需保存一张张量级累积重要性表。

与已有工作不同的是,本文的独特切入是把李雅普诺夫间接法与收缩分析的局部线性化思想引入重要性估计:不再把'梯度大'当作重要,而是先从专家演示本身估计'哪里容易放大扰动'。具体做法是在每条演示轨迹的滑窗上用岭回归拟合局部线性动力学 $\tilde{s}_{j+1}=A_t\tilde{s}_j+B_t\tilde{a}_j+c_t$,用状态转移矩阵的最大奇异值 $\rho_t=\sigma_{\max}(A_t)$ 作为策略无关的单步扰动放大分数,再用它加权样本级动作梯度责任。与 EWC 类参考锚定方法的本质区别在保护机制:DRAP 不惩罚对旧检查点的位移、不存参考权重,历史重要性从新任务第一次更新起就通过逐张量 AdamW 学习率直接抑制真实优化步长;与同作者的 CLASSP 共享'调整抑制'哲学,但重要性信号从通用梯度累积换成物理风险加权的行为责任,接口从自定义优化器换成标准 AdamW 参数组。

核心方法

直觉上,机器人在哪些演示样本上犯错最危险?答案是局部动力学天然放大状态扰动的轨迹段——比如接触瞬间、精细对准阶段。DRAP 把这一直觉变成三段流水线。第一段,KappaTune 按正则化条件数 $\kappa(W)=\sigma_{\max}(W)/(\sigma_{\min}(W)+\epsilon_\kappa)$ 在视觉、语言、动作头各组内选出固定预算的低 $\kappa$ 张量构成可训练子网,其余冻结——决定'在哪适应'。第二段,离线从专家演示的滑窗岭回归中提取逐样本动力学风险 $r_t$,是 CPU 级轻量计算。第三段是在线循环:任务 $k$ 用标准 AdamW 模仿学习(动作 token 位置交叉熵)训练;任务结束后用 batch size=1 的确定性样本子集计算风险加权重要性 $\hat{I}^{(k),new}_T=\frac{1}{N_k}\sum_i r_i^2\|\nabla_T\ell_i\|_F^2/n_T$,经中位数归一并入累积记忆 $I^{(k)}_T=\gamma I^{(k-1)}_T+\bar{I}^{(k),new}_T$(默认 $\gamma=1$);下一任务开始前把重要性转成可塑性乘子 $s^{(k)}_T=(1+\beta\bar{I}^{(k)}_T)^{-1/p}$,逐张量有效学习率 $\eta_T=\eta\cdot s_T$。全程无回放、无参考参数副本:KappaTune 圈定场地,DRAP 分配场地内的可塑性。

核心创新是给重要性加权注入物理后果。论文证明,由于风险分数 $r_i$ 由演示轨迹预先算出、与模型参数无关,风险加权样本损失对张量 $T$ 的梯度范数满足 $\|\nabla_T(r_i\ell_i)\|_F^2=r_i^2\|\nabla_T\ell_i\|_F^2$,因此把 $r_i^2$ 乘进平方梯度即可实现'越危险的样本,支撑它的张量越重要'。风险分数本身有李雅普诺夫式上界支撑:由 $\delta s_{t+1}\approx A_t\delta s_t$ 与扰动能量 $V_s(\delta)=\delta^\top\delta$ 可得 $V_s(\delta_{t+1})\le\sigma_{\max}^2(A_t)V_s(\delta_t)$,故取 $\rho_t=\sigma_{\max}(A_t)$,再按套件中位数归一并裁剪到 $[0.25,4]$ 得 $r_t$。与已有方法的分界线:EWC 用对角 Fisher 惩罚对 $\theta^\star$ 的位移、保护从零渐进生效且需存参考权重;CLASSP 用通用梯度/更新量累积并靠阈值做稀疏化;SI/MAS/RWalk 经 Benzing 证明都与 Fisher 同源。DRAP 是第一个用演示轨迹动力学放大风险替代这些通用统计信号的工作,且保护在第一个优化步就全额生效。

方法步骤详情

第 0 步:KappaTune 一次性选出固定张量集 $\mathcal{T}$,初始化 $I^{(0)}_T=0$,组外参数冻结。第 1 步(离线):把专家演示的状态与动作按套件级稳健统计标准化,在每条 episode 的滑窗 $W_t$ 上用岭回归拟合 $\tilde{s}_{j+1}=A_t\tilde{s}_j+B_t\tilde{a}_j+c_t+\epsilon_j$,计算 $\rho_t=\sigma_{\max}(A_t)$,再经 $r_t=\mathrm{clip}(\rho_t/(\mathrm{median}_j\rho_j+\epsilon_r),\,c_{\min},\,c_{\max})$ 归一裁剪(实验取 $[0.25,4]$),得到与训练样本对齐的风险权重。第 2 步:训练任务 $k$,AdamW 交叉熵损失 $\ell_i=\frac{1}{|Q_i|}\sum_{q\in Q_i}\ell_{i,q}$ 只对动作 token 位置 $Q_i$ 求平均,各张量学习率不同。第 3 步:任务结束后用 $N_k$ 个样本、batch size=1 估计新任务重要性,含三个归约:动作 token 位置平均、张量 $n_T$ 个元素平均、$N_k$ 样本平均;除以 $n_T$ 防止大张量凭参数多占优,batch=1 防止不同样本与动作方向的平方梯度互相抵消。第 4 步:中位数归一 $\bar{I}^{(k),new}=\mathcal{N}(\hat{I}^{(k),new})$ 并入累积记忆 $I^{(k)}_T=\gamma I^{(k-1)}_T+\bar{I}^{(k),new}_T$。第 5 步:算 $s^{(k)}_T=(1+\beta\bar{I}^{(k)}_T)^{-1/p}$,为下一任务建逐张量学习率组 $\eta_T=\eta\cdot s^{(k)}_T$,循环至任务流结束,每阶段保存检查点、张量集与重要性表。

技术新颖性

技术新颖性可从五个角度看。其一,信号层面:这是首个把轨迹级扰动放大风险作为持续学习重要性权重的工作,Table I 的逐维对比显示'物理 grounding'是 DRAP 与 CLASSP/EWC 唯一的维度级差异;其二,保护机制层面:不改损失(对比 EWC 的正则项)、不写自定义优化器(对比 CLASSP),只用 AdamW 的逐张量参数组加一个标量乘子,工程实现近乎零侵入,且抑制的是真实优化步长而非原始梯度;其三,存储层面:只需保存一张张量级累积重要性表,比 EWC 的 Fisher 对角矩阵加参考参数更省,且天然免回放;其四,估计器设计层面:batch size=1 的平方梯度避免跨样本抵消、除以 $n_T$ 消除张量规模偏置、中位数归一去除任务全局尺度只留相对分布,三处细节显示作者对重要性偏差来源的系统性处理;其五,组合层面:KappaTune(结构性圈定适应范围)与 DRAP(按历史责任分配可塑性)形成互补分工,消融证明缺一即崩,这种'选址+配速'的双层控制结构干净且各自可独立替换。

Dynamics-risk computation from expert demonstrations. Local linear dynamics are estimated over sliding windows of logged state-action transitions. The largest singular value σmax(At) defines a local one-step perturbation-amplification score, which is robustly normalized and stored as a sample-level risk weight aligned with the training dataset.
Fig. 1: Dynamics-risk computation from expert demonstrations. Local linear dynamics are estimated over sliding windows of logged state-action transitions. The largest singular value σmax(At) defines a local one-step perturbation-amplification score, which is robustly normalized and stored as a sample-level risk weight aligned with the training dataset.
Overview of DRAP. Expert trajectories provide a local dynamics-risk signal. Sample-level action gradients identify tensor responsibility at risky samples. Cumulative importance directly scales future AdamW learning rates; no previous tensor reference or displacement penalty is required.
Fig. 2: Overview of DRAP. Expert trajectories provide a local dynamics-risk signal. Sample-level action gradients identify tensor responsibility at risky samples. Cumulative importance directly scales future AdamW learning rates; no previous tensor reference or displacement penalty is required.

实验结果

实验在 LIBERO 上按 Spatial→Object→Goal 顺序训练单个 OpenVLA 检查点,五配置对比,每阶段池化 6 次 Gym 评估×50 episode。标准微调呈强新近效应:Spatial 0.77→Object 后 0.11→Goal 后 0.00,Object 0.83→0.07,Goal 后平均成功率仅 0.26、平均遗忘 0.77。KappaTune+EWC 只有短期保护:Object 后 Spatial 0.41,Goal 后跌到 0.03/0.30/0.64,平均遗忘 0.63。KappaTune+DRAP 全面最优:Object 后 0.75/0.80(Spatial 遗忘仅 0.02、平均成功率 0.77),Goal 后 0.34/0.61/0.71,平均成功率 0.55、平均遗忘 0.31,两项汇总指标均为最佳。均匀风险消融($r_i\equiv1$,保留其余全部机制)Goal 后 0.29/0.57/0.66、平均遗忘 0.36,证明张量级重要性加逐张量学习率贡献大头,动力学风险加权再额外改善旧技能保持且不损害新套件获取(Goal 0.71 对 0.66)。去 KappaTune 消融:Object 阶段尚可(0.71/0.82),Goal 后崩至 0.11/0.08/0.66、平均遗忘 0.71,说明更大可塑空间短期获取强、长期干扰路径更多。参照系:OpenVLA-OFT 四套件独立微调平均 97.1%、单策略联合训练 96.8%,DRAP 的免回放顺序学习与之有绝对差距但问题设定本质不同。

任务指标本文基线提升
LIBERO 顺序适应(Spatial→Object→Goal,Goal 阶段后汇总) 平均成功率 AvgSuccess 0.55(KappaTune+DRAP) 0.26(标准顺序微调);0.32(KappaTune+EWC) 较标准微调 +0.29(约 2.1 倍),较 EWC +0.23
LIBERO 顺序适应(Goal 阶段后汇总) 平均遗忘 AvgForget 0.31(KappaTune+DRAP) 0.77(标准微调);0.63(KappaTune+EWC);0.36(均匀风险消融) 较标准微调降低 0.46,较 EWC 降低 0.32,较均匀风险消融降低 0.05
Object 阶段后 Spatial 保持 Spatial 成功率 / Spatial 遗忘 0.75 / 0.02 0.11 / 0.66(标准微调);0.41 / 0.35(KappaTune+EWC) 成功率提升 0.64,遗忘降低 0.64,几乎零遗忘
Goal 套件新任务获取(Goal 阶段后) Goal 成功率 0.71(KappaTune+DRAP) 0.71(标准微调);0.64(EWC);0.66(均匀风险消融) 与标准微调持平且高于 EWC,说明保持旧技能未以牺牲新任务学习为代价
消融:去掉 KappaTune(DRAP 直接用于全模型) Goal 阶段后平均成功率 / 平均遗忘 0.28 / 0.71(DRAP without KappaTune) 0.55 / 0.31(完整 KappaTune+DRAP) 反向对照:证明结构性子网选择是方法成立的必要条件

局限与改进

作者明确承认的局限:$r_t$ 只是相对的、策略无关的指示器,不是闭环稳定性的形式化证明——它只含直接状态敏感项 $A_t$,忽略策略反馈项 $B_tJ_{\pi,t}$(闭环 Jacobian 近似 $J_{CL,t}\approx A_t+B_tJ_{\pi,t}$),也不涵盖高阶非线性、接触模式切换、观测混叠、局部模型估计误差以及偏离专家轨迹的情形;重要张量仍可能通过多次小幅更新累积漂移,论文自己在 Table I 里把这一条列为 DRAP 的 primary risk。我的补充观察:其一,评估只覆盖三个套件的单一顺序、每格 300 episode 池化,但未报告方差或置信区间,0.34/0.61 这类格子的置信度存疑;其二,未与回放基线正面对比,而 Liu 等人已证明小回放对预训练 VLA 非常有效,DRAP 相对'小回放+正则化'的净优势未知;其三,未测 LIBERO-Long 长视野任务,而长序列恰是遗忘累积机制的试金石;其四,$\gamma$、$\beta$、$p$ 未做敏感性分析,裁剪界 $[0.25,4]$ 自称 arbitrarily set;其五,风险估计依赖记录的机器人状态 $s_t$,纯视觉或遥操作数据可能不满足;其六,仅在 OpenVLA 单一架构上验证。

独立分析的弱点

独立分析的弱点:第一,遗忘只是被缓解而非消除——Goal 后 Spatial 剩 0.34(从 0.77 损失 0.43)、平均遗忘 0.31,按此速率更长任务序列下旧技能仍会趋近归零,改进方向是引入 CLASSP 式梯度阈值做稀疏补充,或在检测到遗忘加速时动态调大 $\beta$。第二,策略无关风险的盲区:$\sigma_{\max}(A_t)$ 完全不看策略自身的纠偏能力,一个对扰动鲁棒的策略在敏感区的错误也许可自愈,DRAP 却会过度保护,改进方向是在线估计 $B_tJ_{\pi,t}$ 或改用多步放大增益。第三,重要性估计的代表性偏差:每任务结束时的单次 batch=1 快照可能高估与末期数据相关的方向,改进是任务内多个时点累积或对 $N_k$ 子集按任务阶段分层采样。第四,对 KappaTune 的强依赖:没有它第三任务即崩(0.11/0.08),说明方法的鲁棒窗口窄,改进是让累积重要性反过来在线调整可训练集本身,形成闭环。第五,统计报告薄弱:无置信区间、无多种子训练重复,6 次评估的池化只能压 rollout 方差、压不掉训练随机性,结论的统计强度有限。

未来方向

作者提出的方向:把动力学风险估计器从状态转移项扩展到控制通路敏感度(含 $B_tJ_{\pi,t}$)、在线估计、多步扰动传播以及接触感知或可学习的局部动力学;引入 CLASSP 启发的更新阈值;在更长的持续学习序列上验证。基于本文成果可延伸的方向:其一,用数据驱动的小型动力学模型替代滑窗岭回归,让风险分数适应接触丰富的操作场景;其二,把 DRAP 与小容量回放做系统组合实验,量化免回放正则与经验回放的互补边界——Liu 等人的结果提示两者可能高度互补;其三,在 OpenVLA-OFT(连续动作、动作分块)上重估 $r_t$ 信号,连续动作下 $\|\nabla_T\ell_i\|_F^2$ 的统计性质与离散 token 不同,值得检验;其四,理论侧建立重要性加权强度与收缩率、李雅普诺夫指数的正式联系,回答'该保护多少才够';其五,把协议扩展到 10+ 任务的跨套件长序列与真实机器人,检验 $\gamma$ 折扣在长程下的表现;其六,探索把动力学风险用于数据加权采样或课程排序,而不仅是可塑性控制。

复现评估

复现评估:代码已在 GitHub 开源(oswaldoludwig/DRAP-Dynamics-Risk-Adaptive-Plasticity),LIBERO 基准与 OpenVLA 7B 权重均公开,数据门槛低。算力需求中高:需顺序微调 7B VLA 三次、跑五种配置、每阶段做 6×50 次闭环 Gym 评估,评估端的闭环 rollout 是主要成本,估计需要多卡 80GB 级 GPU 运行数天;相比之下风险估计环节(滑窗岭回归+奇异值+平方梯度累积)是 CPU 级轻量计算,实现半天可完成。整体难度中等偏上:要复现完整表格需搭建 LIBERO 仿真环境、跑通 OpenVLA 训练栈并组织五组顺序训练与池化评估。最大复现风险是超参透明度:正文只给出裁剪界 $[0.25,4]$ 与 $\gamma=1$,而 $\beta$、$p$、KappaTune 各组预算、基础学习率、$N_k$ 与滑窗长度均未在文中给出数值,必须依赖开源代码;评估协议(池化 6×50、每阶段全套件评估)描述清晰,这点对复现友好。单作者完成全部实验也说明工程复杂度可控。

相似论文推荐