跳转至

ReMoT: Reinforcement Learning with Motion Contrast Triplets

会议: ICLR 2026
arXiv: 2603.00461
代码: 无
领域: 自动驾驶
关键词: 视觉语言模型, 时空推理, 运动对比三元组, GRPO, 强化学习

一句话总结

ReMoT 提出一个统一的训练范式,通过规则驱动的运动对比三元组数据集(ReMoT-16K)和 Group Relative Policy Optimization(GRPO)组合奖励优化,系统性地提升 VLM 在时空一致性推理上的能力,在时空推理任务上实现 25.1% 的性能跃升。

研究背景与动机

  1. 领域现状:视觉语言模型(VLM)已发展为通用感知系统,在 AIGC、具身智能和自动驾驶等领域广泛应用。这些任务本质上要求模型能够在空间和时间维度上理解物理场景的演变。
  2. 现有痛点:当前主流 VLM(包括 GPT-4o、Claude-Sonnet-4.5、Gemini-2.5-Pro 等)在时空一致性推理上存在根本缺陷——会混淆相机旋转与物体运动、误判机械臂状态、错误推断角色运动方向等。
  3. 核心矛盾:现有改进方法(架构修改、数据增强)都只是零散修补,缺乏从数据、训练、评估三个维度系统性解决问题的统一框架。训练数据以静态图文对为主,缺少对细粒度运动属性的显式建模。
  4. 本文要解决什么:系统性地解决 VLM 在时空一致性推理上的根本短板,涵盖数据构建、训练优化和评估基准三个层面。
  5. 切入角度:通过运动对比三元组(anchor-positive-negative)显式建模帧间运动属性,迫使模型学习细粒度运动判别而非依赖表面视觉模式。
  6. 核心 idea:利用视频元注释(相机位姿矩阵、机器人动作日志等)通过多专家协作流水线自动生成大规模运动对比三元组,结合 GRPO 强化学习和组合奖励进行训练。

方法详解

整体框架

输入视频元注释 → 多专家协作流水线生成运动对比三元组(ReMoT-16K)→ 基于 Qwen3-VL-4B-Thinking 进行 SFT/GRPO/混合训练 → 输出增强时空推理能力的 VLM。

关键设计

  1. 多专家协作数据构建流水线

    • 运动估计专家 \(g:(I_t, I_{t'}, \mathcal{A}) \to m\):从元数据中提取精确运动属性。导航专家从 SE(3) 位姿矩阵计算刚体变换,操作专家从机器人遥测数据提取末端执行器轨迹。
    • 三元组构建专家 \((\phi, \mathcal{N})\):通过属性特定的幅度阈值筛选显著正样本对:\(\phi(I_t, I_{t'}, m) = (I_{\text{anchor}}, I_{\text{pos}}, m)\)\(\|m\| \in \mathcal{T}_m\)。负样本通过属性条件合成生成:几何合成 \(\mathcal{T}_{\text{geo}}\) 模拟反向运动,检索 \(\mathcal{R}\) 搜索视觉相似但属性不匹配的帧。
    • VQA 公式化专家:为每个三元组设计多角度推理链问答(多选、判断、填空、比较推理等)。
    • 设计动机:直接用 VLM 生成会产生 55% 的格式错误和高昂 API 成本,多专家流水线可规模化、高质量地生成数据。
  2. GRPO 组合奖励训练

    • 核心目标函数:\(J(\theta) = \mathbb{E}_{q,\{o_i\}}\left[\frac{1}{G}\sum_{i=1}^{G}\min(r_i\hat{A}_i, \text{clip}(r_i, 1-\varepsilon, 1+\varepsilon)\hat{A}_i) - \beta D_{\text{KL}}(\pi_\theta \| \pi_{\text{ref}})\right]\)
    • CoT 长度正则化\(R_{\text{length}}(o_i) = -\max(0, |o_i^{\text{think}}| - L_{\text{target}})\),抑制过度冗长的推理链。
    • 逻辑一致性奖励\(R_{\text{logic}}(o) \in \{+1, 0, -1\}\),通过传递性检查识别逻辑矛盾(如 \(L_1 < L_2, L_2 < L_3\)\(L_3 < L_1\))。
    • 组合奖励\(R_i = R_{\text{task}} + \lambda_1 R_{\text{logic}} + \lambda_2 R_{\text{length}}\),权重比 3.5:3.5:1.3:1.7。
    • 设计动机:31.4% 的基线错误表现为逻辑不一致,解耦逻辑奖励可直接纠正此类问题。
  3. 混合优化策略

    • 顺序混合(SFT→GRPO):SFT 先提供稳定初始化,再切换到 GRPO 精调。
    • 交替混合(SFT↔GRPO):每隔几步交替执行 SFT 和 GRPO,让语言对齐和奖励对齐联合演化。交替策略取得最佳效果。

损失函数 / 训练策略

  • SFT 阶段仅在 <answer> 标签内的 token 上计算交叉熵损失
  • GRPO 采用 4 个 rollout、batch size 16、KL 正则系数 0.01
  • 交替混合策略在 2 个 epoch 后达到最佳表现(Overall 39.9%, Partial 67.7%)
  • 训练硬件:8× A800 GPU,混合精度

实验关键数据

主实验

模型 Overall Acc.(%) Partial Acc.(%) 备注
Qwen3-VL-CoT-4B (基线) 20.7 38.9 基础模型
GRPO 33.6 61.6 纯 RL
SFT→GRPO 35.0 63.3 顺序混合
SFT↔GRPO 38.0 64.0 交替混合,最佳
SFT↔GRPO (2 epochs) 39.9 67.7 更长训练
GPT-5-Chat 10.4 33.3 闭源
Gemini-2.5-Pro 26.4 49.1 闭源

消融实验

配置 Overall Acc.(%) Partial Acc.(%) 说明
无训练(基线) 20.7 38.9 Qwen3-VL-CoT-4B
仅操作数据 23.9 46.7 +3.2%
+ 导航数据 32.4 57.6 +8.4%,空间关系推理关键
+ 仿真数据 38.0 64.0 +5.6%
Binary 对比 vs Triplet 对比 19.4 vs 38.0 39.4 vs 64.0 三元组设计提升 +18.6%
GRPO w/o 逻辑奖励 68.6 (Ov.) 77.3 (Par.) 操作子集
GRPO w/ 逻辑奖励 78.0 (Ov.) 81.3 (Par.) +10.6%,逻辑一致性 99.3%

关键发现

  • 交替 SFT-GRPO 训练比纯 SFT 或纯 GRPO 更优,结合了语言流畅性和奖励对齐的联合优化
  • 三元组对比比二元对比提升 18.6%(Overall),联合对比监督是细粒度运动判别的关键
  • 多专家流水线数据质量远优于 VLM 生成数据,后者在扩展时波动大且上限约 0.49
  • 4B 模型即可在时空推理基准上超越 7.5× 大的 Qwen3-VL-30B-CoT
  • 纯 SFT 训练会导致训练崩溃,模型丧失 CoT 推理能力

亮点与洞察

  • 系统性地从数据、训练、评估三个维度解决 VLM 时空推理能力的不足,是一个完整框架
  • 多专家协作流水线巧妙利用视频元注释避免了昂贵的人工标注和不可靠的模型标注
  • 组合奖励设计(特别是逻辑一致性奖励)针对性强,直接解决推理链与答案脱钩的问题
  • 用 4B 模型匹配甚至超越 GPT-4o 级别的时空推理能力,性价比极高

局限性 / 可改进方向

  • ReMoT-16K 数据集规模相对有限(16.5K 三元组),扩大规模可能进一步提升
  • 闭源模型仅在约 40 个样本的 mini-benchmark 上评估,受 API 成本限制
  • 当前仅在 4B/8B 模型上验证,更大规模模型的效果有待探索
  • 运动对比三元组主要来自导航、操作和仿真,可扩展到更多领域(如体育、医疗)
  • 推理链忠实度虽从 60% 降到 12% 的错误率,但仍有提升空间

相关工作与启发

  • GeoNLF、BARF 等 pose-free NeRF 方法启发了从元数据提取运动信息的思路
  • GRPO 在视觉推理中的成功应用表明 RL 方法在多模态场景中的潜力
  • 运动对比三元组的设计方式可推广到视频理解、机器人动作识别等下游任务
  • 逻辑一致性奖励的设计思想可迁移到其他需要多步推理一致性的任务

评分

  • 新颖性: ⭐⭐⭐⭐
  • 实验充分度: ⭐⭐⭐⭐⭐
  • 写作质量: ⭐⭐⭐⭐
  • 价值: ⭐⭐⭐⭐

相关论文