4. 预训练与微调基础

在视觉语言动作模型(VLA)的落地过程中,预训练与微调是两条紧密衔接的链路。预训练让模型在海量图文与动作数据中学习通用表征,微调则把这种通用能力迁移到具体机器人本体、具体任务和具体场景。理解迁移学习、领域适应与参数高效微调(PEFT)之间的关系,是后续动手微调与部署的前提。

一、核心原理:从通用表征到任务适配

迁移学习的基本假设是:源域上学到的特征对目标域仍有价值。VLA 模型通常在大规模视觉-语言数据与多形态机器人轨迹上预训练,已经具备“看懂物体、理解指令、生成动作”的初级能力。微调阶段,我们不再从随机权重开始,而是以预训练权重为起点,用目标机器人采集的少量演示数据继续训练。

领域适应关注的是分布差异:预训练数据来自仿真或不同相机、不同机械臂,而部署现场存在光照变化、相机内参差异、动作空间维度不同等问题。领域适应的目标不是重新学习全部知识,而是让模型在目标域上保持稳定。常见策略包括输入归一化、视觉增广、动作空间对齐、少量目标域数据校准等。

参数高效微调则解决“全量微调代价过高”的问题。VLA 模型参数量常达数亿甚至数十亿,全量微调需要保存优化器状态、梯度与多份权重,显存和存储开销巨大。PEFT 通过冻结大部分预训练参数,只训练少量新增或选择的参数,在接近全量微调效果的同时显著降低资源需求。

一句话理解:预训练提供“通用能力”,迁移学习决定“往哪迁移”,领域适应解决“分布不一致”,PEFT 解决“怎么低成本迁移”。

二、方法流程:微调一条 VLA 模型的典型路径

以机械臂抓取任务为例,微调流程通常包含以下步骤:

  1. 确定目标与数据:明确任务类型(抓取、放置、按压等),采集 50~500 条演示轨迹,包含 RGB 图像、语言指令、关节或末端位姿动作。
  2. 选择微调策略:数据量少、显存有限时优先 LoRA;需要插入轻量模块且不改动原结构时选 Adapter;追求极致参数效率时考虑 Prompt Tuning 或 Prefix Tuning。
  3. 配置训练参数:设置学习率、批大小、秩(rank)、缩放系数 alpha、目标模块等。
  4. 训练与验证:在验证集上观察动作误差与任务成功率,防止过拟合。
  5. 部署与回退:将 LoRA 权重合并或独立加载,在真实机器人上测试,必要时回退到预训练权重。

下面给出一个 LoRA 微调的配置示例,展示关键参数:

{
  "method": "lora",
  "r": 16,
  "lora_alpha": 32,
  "lora_dropout": 0.05,
  "target_modules": ["q_proj", "v_proj", "action_head"],
  "learning_rate": 1e-4,
  "batch_size": 8,
  "epochs": 30,
  "freeze_backbone": true
}

三、参数高效微调方法对比

不同 PEFT 方法在插入位置、训练参数比例和适用场景上差异明显。下表对 LoRA、Adapter、Prompt/Prefix Tuning 进行对比:

方法插入位置可训练参数占比优点局限
LoRA注意力/线性层旁路约 0.1%~1%不改变原结构,可合并权重,推理无额外延迟秩与目标模块需调参
AdapterTransformer 层间约 1%~5%模块化强,适合多任务切换增加推理层,略有延迟
Prompt Tuning输入前缀约 0.01%~0.1%参数极少,训练快对复杂动作任务容量有限
Prefix Tuning每层 KV 前缀约 0.1%~1%表达能力强于 Prompt实现较复杂,占用序列长度

四、案例与场景:LoRA 在机械臂抓取中的应用

假设我们有一个预训练的 VLA 模型,已在仿真环境中见过大量抓取任务,但真实机械臂的相机角度和夹爪行程不同。直接全量微调需要 4 张 A100,成本高。采用 LoRA 后,只训练注意力层的低秩矩阵和动作头,单张 24GB 显卡即可完成。

训练日志中可观察到:前 5 个 epoch 损失快速下降,第 10 个 epoch 后验证成功率趋于稳定。若验证损失开始上升而训练损失继续下降,说明出现过拟合,应降低秩、增大 dropout 或增加数据增广。

实践建议:首次微调时,先用小秩(r=8)和较高学习率快速验证流程;确认可行后再增大秩或扩展目标模块。领域差异大时,可先做一轮轻量领域适应,再接入 LoRA。

五、误区与复盘

  • 误区一:PEFT 一定不如全量微调。在数据量有限、目标域与源域接近时,PEFT 常能取得接近甚至更优的泛化效果,因为它减少了过拟合风险。
  • 误区二:只调动作头就够了。VLA 的视觉与语言表征同样需要适配,只调动作头可能无法应对新物体或新指令。
  • 误区三:忽略领域适应。相机内参、光照、动作尺度不一致时,直接微调效果会大打折扣,应先做输入与动作空间对齐。
  • 复盘要点:记录每次实验的秩、目标模块、学习率与数据量,建立可复现的配置档案,便于后续排障与迭代。
VLA 模型:预训练 → 迁移/领域适应 → PEFT 微调 → 部署 预训练权重 大规模图文 + 轨迹 通用视觉-语言-动作表征 迁移学习 复用通用特征 目标机器人演示数据 PEFT 微调 LoRA / Adapter 冻结主干,训练少量参数 部署 真实机器人 任务执行 领域适应:解决源域与目标域的分布差异 输入归一化 相机内参 / 光照 视觉增广 颜色 / 裁剪 / 噪声 动作空间对齐 关节 / 末端位姿 少量目标域校准 微调前快速适配 PEFT 方法选择参考 LoRA:旁路低秩矩阵 Adapter:层间轻量模块 Prompt:输入前缀 Prefix:KV 前缀

总体来看,预训练与微调不是对立关系,而是“通用能力”与“专用能力”的接力。迁移学习决定知识复用的方向,领域适应保证分布一致,PEFT 让微调在有限算力下可行。掌握这三者的配合方式,才能在真实机器人上稳定地完成 VLA 模型的落地。