CLARE:通过自适应适配器路由和扩展实现视觉-语言-动作模型的持续学习 CLARE: Continual Learning for Vision-Language-Action Models via Autonomous Adapter Routing and Expansion
CLARE通过轻量级适配器和自编码器路由实现机器人VLA模型的持续学习,避免灾难性遗忘
前置知识
视觉-语言-动作模型(VLA)
视觉-语言-动作模型是一类将视觉感知、语言理解和动作生成统一在单一框架中的机器人学习模型。这类模型通常基于预训练的视觉-语言模型(如CLIP、DINOv2)作为视觉和语言编码器,结合动作解码器(如扩散策略)来生成机器人控制指令。通过在大规模机器人演示数据上预训练,VLA能够获得广泛的视觉-语言-动作先验知识,但面对新任务时仍需要微调才能达到高性能。典型架构包括编码器-解码器结构,其中编码器处理多模态观测(图像、语言、本体状态),解码器生成动作块。
理解VLA是本文的基础,CLARE正是针对VLA的持续学习框架,需要了解VLA的架构特点和局限性才能理解本文的动机和方法设计
灾难性遗忘
灾难性遗忘是指神经网络在学习新任务时,由于参数更新导致之前学习任务性能急剧下降的现象。在持续学习场景中,当模型在新任务数据上进行标准微调时,会覆盖之前学到的表示,导致旧任务的成功率大幅下降。例如,在机器人操作中,学会'把杯子放在盘子上'后,再学习'关闭抽屉'可能会完全忘记如何完成第一个任务。这是持续学习的核心挑战,也是本文要解决的核心问题。
本文的核心目标就是解决VLA在持续学习中的灾难性遗忘问题,理解这一现象的严重性才能领会CLARE方法的价值
适配器(Adapter)
适配器是一种参数高效的微调技术,在预训练模型的特定层之间插入小型神经网络模块。这些模块通常采用瓶颈结构(先降维再升维),参数量远小于原始模型。在CLARE中,适配器采用编码器-解码器结构:输入特征$x_\ell$首先通过下投影矩阵$W_\ell^{down} \in \mathbb{R}^{r \times d_\ell}$降维到低秩空间,经过ReLU激活后,再通过上投影矩阵$W_\ell^{up} \in \mathbb{R}^{d_\ell \times r}$升维回原始维度,其中$r \ll d_\ell$是瓶颈维度。这种设计使得适配器只增加约2%的参数就能捕获任务特定知识。
CLARE的核心创新之一就是动态扩展的适配器模块,理解适配器的结构和工作原理是理解整个方法的关键
混合专家模型(MoE)
混合专家模型是一种架构范式,通过门控机制动态选择不同的专家子网络来处理不同的输入。在大型语言模型中,MoE层包含多个专家网络和一个路由器,路由器根据输入特征计算每个专家的权重,通常采用top-k选择策略。CLARE受到MoE启发,但有两个关键区别:1) MoE的专家数量固定,而CLARE的适配器数量随任务增加;2) MoE的路由器是训练得到的,而CLARE使用基于重建误差的自编码器路由机制。
CLARE的设计借鉴了MoE的思想,但针对持续学习场景进行了重要改进,理解MoE有助于理解CLARE的创新点
流匹配(Flow Matching)
流匹配是一种生成建模方法,通过学习向量场将简单分布(如高斯分布)转换为目标分布。在机器人策略学习中,流匹配用于学习动作块的生成过程:给定噪声动作$A_0 \sim \mathcal{N}(0, I)$和观测$o_t$,策略网络学习一个向量场$v_\theta(A_s, o_t, s)$,其中$s \in [0,1]$是时间步。训练目标是让向量场逼近从噪声到真实动作的线性插值:$\mathcal{L}(\theta_n) = \mathbb{E}_{s, (A_1, o), A_0} \| v_\theta(A_s, o, s) - (A_1 - A_0) \|^2$。推理时通过欧拉积分从噪声生成动作:$A_{s+\delta s} = A_s + \delta s \cdot v_\theta(A_s, o_t, s)$。
CLARE基于流匹配策略进行持续学习,理解流匹配的工作原理有助于理解CLARE的训练过程和损失函数设计
自编码器(Autoencoder)
自编码器是一种无监督学习模型,由编码器和解码器组成,目标是学习输入数据的压缩表示。在CLARE中,自编码器被用作判别器来评估输入特征与训练分布的匹配程度。每个判别器$D_j^\ell$接收输入特征$x_\ell$,尝试重建它,重建误差$e_j^\ell(x_\ell) = \| x_\ell - D_j^\ell(x_\ell) \|^2$用于衡量特征与训练分布的偏离程度。重建误差越小,说明该判别器对应的适配器越适合处理当前输入。
CLARE使用自编码器重建误差作为路由机制的核心,这是实现自主任务识别的关键技术
研究动机
在现实世界中,机器人必须在不断变化的环境中长期运行,例如家庭机器人可能需要操作新购买的电器,辅助机器人可能遇到行动能力不同的患者。这种长期适应性要求机器人能够持续学习新技能而不牺牲已有能力,即持续学习或终身学习。然而,现有的视觉-语言-动作模型(VLA)在持续学习中面临严重挑战。当前的主流方法是迭代式全参数微调(FFT),但这会导致灾难性遗忘:在LIBERO基准实验中,序列微调后的平均成功率仅为22.4%,而初始训练时成功率可达76.1%,遗忘率高达74.7%。经验回放(ER)虽然能缓解遗忘,但需要存储50%的历史数据,在存储或隐私受限的终身学习场景中不实用。更重要的是,ER在40个任务的长期序列中仍会出现显著的性能下降,某些任务(如T1和T7)的成功率大幅降低,产生23%的负向迁移。
本文的目标是本文的目标是开发一个通用的、参数高效的持续学习框架,使VLA能够在不存储历史数据(exemplar-free)、不需要任务标识符的情况下持续学习新任务。具体而言,CLARE需要满足以下要求:1) 保持预训练模型的通用表示能力;2) 为每个新任务添加少量参数(约2%);3) 在推理时能自主识别任务并激活相应的适配器;4) 避免灾难性遗忘,保持对旧任务的高成功率;5) 在仿真和真实世界环境中验证有效性。
与已有工作不同的是,现有持续学习方法存在三个关键缺陷:1) 经验回放类方法需要存储历史数据,在长期部署中不现实;2) 正则化类方法(如EWC、PackNet)受限于固定参数集,难以扩展到长任务序列;3) 架构扩展类方法通常需要任务标识符,无法实现自主操作。CLARE的独特切入角度是:将混合专家思想与动态架构扩展相结合,通过自编码器重建误差实现任务感知,无需任务标签即可路由到正确的适配器。同时,基于特征相似性的动态扩展策略只在必要时添加新适配器,避免了参数冗余。这种设计在保持模型紧凑性的同时,实现了任务无关的自主推理。
核心方法
CLARE的核心思想是在预训练VLA中注入轻量级适配器,并通过自编码器路由机制动态激活最相关的适配器。整体架构包含三个协同组件:首先,在选定层(主要是观察编码模块)中插入适配器作为并行分支,保持原始模型结构不变;其次,为每个适配器训练一个自编码器判别器,通过重建误差评估输入特征与训练分布的匹配程度;最后,在部署时,路由机制选择重建误差最小的判别器对应的适配器。这种设计使得模型能够:1) 通过冻结主干网络保持通用知识;2) 通过任务特定适配器学习新技能;3) 通过动态扩展避免参数冗余;4) 通过自主路由实现任务无关推理。
CLARE的核心创新在于两个关键设计:自适应适配器路由和动态扩展策略。与现有方法的本质区别在于:1) 与混合专家模型不同,CLARE的适配器数量随任务动态增长,而非固定;2) 与经验回放不同,CLARE无需存储历史数据,通过参数隔离避免遗忘;3) 与需要任务标识符的方法不同,CLARE使用基于重建误差的路由机制实现自主任务识别;4) 与简单的逐层扩展不同,CLARE基于特征相似性的z-score决定是否扩展,避免过度参数化。这种设计在保持模型紧凑性(每任务仅增加约2%参数)的同时,实现了接近零的负向迁移(NBT约-1.0%),甚至优于需要存储50%历史数据的经验回放方法。
方法步骤详情
CLARE的训练和推理包含以下步骤:\n\n**训练阶段(算法1):**\n1. 初始化:为每个可扩展层$\ell \in \mathcal{E}$初始化空适配器集$\mathcal{A}_\ell = \emptyset$和判别器集$\mathcal{D}_\ell = \emptyset$,计数器$k_\ell = 0$\n2. 对于每个新任务$\mathcal{T}_n$:\n a) 收集演示数据$\mathcal{D}_n$\n b) 动态扩展:对于每个可扩展层$\ell$:\n - 计算现有判别器的z-score:$z_j^\ell = \frac{1}{|\mathcal{D}_n|} \sum_{x_\ell \in \mathcal{D}_n} \frac{e_j^\ell(x_\ell) - \mu_j^\ell}{\sigma_j^\ell}$\n - 如果所有$z_j^\ell > \gamma$(阈值)或$n=1$:扩展新适配器$A_{k_\ell}^\ell$和判别器$D_n^\ell$,建立映射$\mathcal{B}_\ell(D_n^\ell) = A_{k_\ell}^\ell$\n - 否则:只添加判别器$D_n^\ell$,链接到最相关适配器:$\mathcal{B}_\ell(D_n^\ell) = A_i^\ell = \mathcal{B}_\ell(D_{j^*}^\ell)$,其中$j^* = \arg\min_j \mathbb{E}_{x_\ell \sim \mathcal{D}_n} e_j^\ell(x_\ell)$\n c) 两阶段训练:首先用流匹配损失训练新适配器;然后冻结其他参数,用重建损失训练新判别器\n\n**推理阶段(算法2):**\n对于输入特征$x_\ell$:\n1. 计算所有判别器的重建误差:$e_j^\ell(x_\ell) = \| x_\ell - D_j^\ell(x_\ell) \|^2$\n2. 选择最小误差对应的判别器:$j^* = \arg\min_j e_j^\ell(x_\ell)$\n3. 激活链接的适配器:$A_\ell^* = \mathcal{B}_\ell(D_{j^*}^\ell)$\n4. 计算输出:$M_\ell(x_\ell) = M_\ell(x_\ell)^{\text{pre}} + A_\ell^*(x_\ell)$
技术新颖性
CLARE的技术新颖性体现在四个方面:\n\n1. **自编码器路由机制**:不同于传统的门控网络或基于梯度的路由,CLARE使用重建误差作为任务相似性度量。这种设计无需任务标签,且能处理连续变化的任务分布。判别器通过最小化重建误差学习任务特定特征,路由选择重建误差最小的判别器对应的适配器。\n\n2. **动态扩展策略**:基于特征相似性的z-score决定是否扩展,而非盲目为每个任务添加适配器。z-score标准化考虑了不同判别器的统计特性,阈值$\gamma$控制扩展的积极程度。当$\gamma=0$时每个任务都扩展,当$\gamma=20$时仅扩展显著不同的任务。\n\n3. **参数高效设计**:适配器采用瓶颈结构($r \ll d_\ell$),每任务仅增加约2%参数。实验显示,在LIBERO-Long上,CLARE的AUC(75.1%)比需要存储50%数据的经验回放(60.5%)高14.6个百分点,而参数量仅为后者的约1/50。\n\n4. **两阶段训练策略**:先训练适配器(流匹配损失),再训练判别器(重建损失)。这种顺序确保判别器能在稳定的特征空间中学习,避免了适配器和判别器同时训练的不稳定性。\n\n与现有方法相比,CLARE在以下指标上表现突出:前向迁移(FWT)达75.0%,与全参数微调相当;负向迁移(NBT)仅1.9%,显著低于经验回放的22.7%;AUC达75.1%,比最好的基线高14.6个百分点。
实验结果
CLARE在多个基准和真实世界任务上表现出色,验证了其作为VLA持续学习框架的有效性。在LIBERO-Long基准上,CLARE的AUC达到75.1%,比经验回放(60.5%)高14.6个百分点,比序列微调(22.4%)高52.7个百分点。前向迁移(FWT)为75.0%,与全参数微调的76.1%相当,表明CLARE能有效学习新任务。负向迁移(NBT)仅1.9%,显著低于经验回放的22.7%,证明了其避免遗忘的能力。\n\n在LIBERO-Goal和LIBERO-Spatial基准上,CLARE分别达到89.3%和87.4%的AUC,比最佳基线高约11-13个百分点。特别值得注意的是,在LIBERO-Goal上,CLARE的NBT仅为0.3%,几乎完全避免了遗忘。\n\n在40个任务的长期序列实验(LIBERO-40)中,CLARE能够顺序学习并保持40个不同任务的成功率,而经验回放则出现了显著性能下降(NBT 23%),某些任务(如T1和T7)成功率大幅降低。\n\n真实世界实验在5个操作任务上验证了CLARE的实用性:AUC达63.3%,比经验回放高12个百分点,NBT为-2.9%(负值表示正向迁移)。在动态扩展实验中,当阈值$\gamma$从0增加到20时,适配器数量从60减少到16,AUC从75.1%降至约65%,但仍高于所有基线,证明了扩展策略的灵活性。\n\n计算开销方面,CLARE的推理时间增加低于3毫秒,GPU显存每任务仅增加约2%,远低于经验回放的数据存储需求。这种效率使得CLARE适合长期部署在资源受限的机器人系统中。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| LIBERO-Long(10个长期任务) | AUC(面积 under 成功率曲线) | 75.1±1.3% | ER: 60.5±0.2%, SeqFFT: 22.4±0.3%, SeqLoRA: 21.4±1.0% | 比最佳基线ER高14.6个百分点 |
| LIBERO-Long(10个长期任务) | FWT(前向迁移) | 75.0±1.4% | ER: 76.6±0.9%, SeqFFT: 76.1±1.0% | 与全参数微调相当 |
| LIBERO-Long(10个长期任务) | NBT(负向迁移,越低越好) | 1.9±0.4% | ER: 22.7±1.8%, SeqFFT: 74.7±1.1% | 比ER低20.8个百分点,接近零遗忘 |
| LIBERO-Goal(10个目标导向任务) | AUC | 89.3±1.1% | ER: 76.0±0.9%, DMPEL: 78±2% | 比最佳基线高11.3个百分点 |
| LIBERO-Spatial(10个空间任务) | AUC | 87.4±2.3% | ER: 77.6±0.8%, DMPEL: 70±3% | 比最佳基线高9.8个百分点 |
| 真实世界(5个操作任务) | AUC | 63.3% | ER: 51.1%, SeqFFT: 23.8% | 比ER高12.2个百分点 |
| LIBERO-40(40个任务长期序列) | NBT | 约0%(无显著遗忘) | ER: 23%(显著遗忘) | 避免了长期序列中的灾难性遗忘 |
| 计算效率 | 推理时间增加 | <3毫秒 | ER需要加载50%历史数据 | 几乎无推理开销 |
| 计算效率 | GPU显存增加 | 每任务约2% | ER需要存储历史数据 | 内存效率极高 |
局限与改进
尽管CLARE在持续学习中表现出色,但仍存在一些局限性:\n\n1. **预训练依赖**:CLARE假设VLA已在大规模多样化演示数据上预训练,覆盖常见的运动模式。如果预训练数据不足,观察编码模块可能无法提供足够的特征多样性,影响路由和扩展的效果。实验显示,扩展解码器模块的效果显著差于扩展编码器模块(AUC 41.8% vs 75.1%),表明当前设计对预训练质量有较高要求。\n\n2. **阈值敏感性**:扩展阈值$\gamma$需要根据任务重要性和内存约束权衡选择。实验显示,$\gamma$从0增加到20时,适配器数量从60减少到16,AUC从75.1%降至约65%。这种敏感性意味着在资源受限场景中,需要仔细调整超参数。\n\n3. **任务同质性假设**:CLARE的路由机制假设任务间存在可区分的特征分布。在真实世界实验中,DRAWER和LEGO任务涉及几乎相同的场景(相同工作台、相同抽屉),路由器必须依赖语言和本体特征的细微差异来区分。虽然实验显示路由器能正确处理(10次rollout一致),但在更复杂的场景中,这种细微差异可能不足。\n\n4. **扩展策略的保守性**:动态扩展策略基于特征相似性决定是否扩展,但'相似性'的定义可能过于保守。实验显示,当$\gamma=20$时,仅16个适配器处理40个任务,某些任务可能被迫使用不完全匹配的适配器。虽然AUC仍高于基线,但任务特定性能可能受影响。\n\n5. **评估范围**:虽然论文在LIBERO基准和5个真实任务上验证,但所有任务都涉及桌面操作,缺乏更复杂环境(如移动操作、双手协调)的验证。此外,真实世界实验仅5个任务,难以评估长期扩展性。\n\n6. **计算开销分析不完整**:论文报告了推理时间增加<3毫秒,但未详细分析训练时间。两阶段训练(适配器训练+判别器训练)可能增加总训练时间,特别是在任务数量多时。
独立分析的弱点
基于独立分析,CLARE存在以下可改进的弱点:\n\n1. **路由机制的可解释性不足**:虽然自编码器重建误差提供了任务相似性度量,但难以解释为什么某个适配器被选中。改进方向:引入注意力机制可视化路由决策,或开发基于规则的后处理来提高可解释性。\n\n2. **扩展策略的静态性**:阈值$\gamma$在训练前设定,无法根据任务难度动态调整。改进方向:开发自适应阈值机制,根据验证集性能自动调整$\gamma$,或使用元学习预测最优扩展策略。\n\n3. **适配器独立性假设**:每个适配器独立训练,缺乏跨任务知识共享。虽然这避免了干扰,但也可能错过任务间的共性。改进方向:引入分层适配器结构,底层共享通用特征,高层捕获任务特定知识。\n\n4. **判别器训练的不稳定性**:判别器在冻结其他参数后训练,可能无法充分利用新特征。改进方向:探索联合训练策略,或使用对抗训练提高判别器的区分能力。\n\n5. **缺乏失败案例分析**:论文未详细分析失败案例,特别是路由错误或扩展决策失误的情况。改进方向:系统分析失败模式,开发故障检测和恢复机制。
未来方向
基于CLARE的成果,未来研究可沿以下方向扩展:\n\n1. **大规模VLA适配**:将CLARE应用于更大的VLA模型(如OpenVLA、π0.5),验证在更复杂任务和更长任务序列中的有效性。预训练模型规模的增加可能改变扩展策略的最优性。\n\n2. **多模态任务定义**:当前任务仅基于语言指令定义,未来可探索结合视觉示例、力反馈等多模态任务定义,提高路由的鲁棒性。\n\n3. **在线学习扩展**:将CLARE从离线演示学习扩展到在线强化学习,使机器人能在真实交互中持续学习。这需要解决探索-利用平衡和样本效率问题。\n\n4. **分布式持续学习**:在多机器人系统中部署CLARE,实现跨机器人的知识共享和适应。需要解决通信效率和隐私保护问题。\n\n5. **安全约束整合**:在持续学习中加入安全约束,确保新技能的学习不会违反安全规则。这需要开发安全感知的扩展和路由策略。\n\n6. **理论分析**:深入分析CLARE的收敛性、泛化能力和遗忘边界,为方法设计提供理论指导。\n\n7. **硬件效率优化**:开发硬件友好的适配器结构,如量化适配器或稀疏适配器,进一步降低计算和内存开销。
复现评估
CLARE的复现性评估如下:\n\n**开源情况**:论文承诺代码、数据和视频将在网站tum-lsy.github.io/clare发布,但截至分析时可能尚未完全开源。开源内容包括CLARE框架代码、LIBERO基准评估脚本、真实世界实验数据。\n\n**数据要求**:CLARE需要预训练VLA模型和持续学习任务数据。预训练数据需要大规模多样化演示(如DROID数据集),持续学习数据需要任务特定的演示。在LIBERO基准上,每个任务有50个人类专家演示;真实世界实验使用1000个实验室演示和2000个DROID episode。\n\n**算力需求**:训练需要NVIDIA RTX 5090 GPU,每个仿真任务约1小时,每个真实世界任务约5小时。推理时间增加<3毫秒,GPU显存每任务增加约2%。这种需求对学术研究可行,但对资源受限的机构可能较高。\n\n**复现难度**:中等。需要:1) 预训练VLA模型(论文未明确提供预训练代码);2) LIBERO基准环境(开源);3) 真实世界硬件(FR3机械臂、相机等);4) 超参数调整(阈值$\gamma$、适配器结构等)。\n\n**关键挑战**:1) 预训练数据的质量和多样性对CLARE性能影响大,复现时需确保预训练质量;2) 真实世界实验需要精确的硬件设置和数据收集,增加了复现复杂性;3) 超参数(如$\gamma$)需要根据具体任务调整,论文提供了一些指导但非通用。\n\n**建议复现步骤**:1) 先使用LIBERO基准验证框架;2) 调整扩展阈值$\gamma$观察影响;3) 在简单真实任务上测试;4) 逐步扩展到更复杂场景。
论文图表
图7展示了CLARE的推理时间和内存开销。左侧显示推理时间随阶段增加的趋势(实线为实际测量,虚线为外推),中间显示GPU显存使用,右侧显示磁盘使用。适配器和判别器的开销远小于基础策略。
这张图量化了CLARE的效率优势,证明其适合长期部署。
表格I列出了CLARE的关键超参数:适配器和判别器的参数量(约0.38M-6.08M)、学习率(2e-4到5e-4)、批量大小(32-256)、训练步数(2000-20000)、扩展阈值$\gamma$(0-2.5)。适配器参数量远小于基础策略(约200M参数)。
这个表格提供了复现CLARE所需的具体参数,是理解实验设置的基础。
表格II展示了在LIBERO-Long上扩展不同层的性能对比。扩展编码器(线性投影)的AUC为75.1%,扩展解码器的AUC为41.8%,扩展编码器+解码器为66.6%。这表明观察编码模块最适合持续学习。
这个表格为CLARE的扩展策略提供了实证支持,解释了为什么选择观察编码模块。
表格III详细比较了CLARE与七个基线在三个LIBERO基准上的性能。CLARE在所有基准上取得最高AUC(75.1%、89.3%、87.4%),同时保持低NBT(1.9%、0.3%、0.9%)。基线包括SeqFFT、SeqLoRA、PackNet、ER、LOTUS、DMPEL、MLR。
这个表格是论文的核心结果,全面展示了CLARE相对于现有方法的优势。
表格IV展示了真实世界实验的总体性能:CLARE的AUC为63.3%,FWT为62.0%,NBT为-2.9%。与ER(AUC 51.1%)和SeqFFT(AUC 23.8%)相比,CLARE显著优越。负NBT表示正向迁移。
这个表格证明了CLARE在真实世界中的有效性,是验证方法实用性的关键。
表格V展示了五个真实世界任务在每个阶段的成功率变化。CLARE(右列)在大多数任务上保持稳定或提升,而ER(左列)在某些任务上出现下降。例如,BOWL任务CLARE保持100%成功率,ER从100%降至80%。
这个表格提供了详细的逐任务分析,展示了CLARE避免遗忘的能力。
表格VI比较了扩展3个和6个AdaLN模块的真实世界性能。扩展6个模块在大多数任务上表现更好,例如BOWL从90%提升到100%,DRAWER从60%提升到90%。这表明扩展更多模块可以提高性能。
这个表格为CLARE的模块选择提供了指导,展示了扩展更多模块的收益。