SafeBranch:用分支对把安全信号钉进具身智能体的决策点

快速导读:SafeBranch 通过环境回滚从智能体自身的不安全轨迹中构造同状态分支对,并用 BranchPO 将安全信号内化到 VLM 具身规划器中,实现无 critic 部署。

SafeBranch 针对 VLM 具身智能体的交互式安全问题,提出了一种新的监督形式——分支对(branch pair)。其核心主张是:安全违规由稀疏的安全关键步决定,因此安全信号应当集中在这类决策点上,而不是分散到整条轨迹或外挂到推理时模块。SafeBranch 通过环境回滚从智能体自身的不安全 rollout 中构造同状态分支对,并用 BranchPO 将 critic 的安全判断内化到智能体内部,实现无 critic 部署。

文章将按以下顺序展开:先说明交互式安全的特殊性和现有方法的缺口,再详细拆解 SafeBranch 的数据构造流程与 BranchPO 目标,然后梳理主要实验结果与跨模拟器迁移证据,最后讨论数据构造效率、过滤阶段的作用、局限性与相关工作。

英文题目:SafeBranch: Branch-Pair Safety Alignment for Embodied Agents

论文出处:arXiv 每日论文精选 · arXiv:2608.19729

原始论文:PDF / 论文页面

这篇论文解决什么问题?

交互式安全与任务成功是不同粒度的目标。任务成功关注整条轨迹是否达成目标,而安全往往由某一个决策点决定:在某个安全关键步上选错动作,整条轨迹就变成危险的。这种稀疏性意味着,安全监督必须能够定位到具体的决策点,而不是笼统地评价整条轨迹。

现有推理时安全模块,如 HomeGuard、C-MCTS、RoboMonkey 等,在部署时对每一步或关键步调用外部 critic 或验证器。这类方法的问题在于:部署开销随步数线性增长,且智能体本身并没有学会安全行为,一旦移除护栏,安全性立即消失。

训练时方法中,SFT 模仿安全轨迹,但缺少不安全对比项,智能体无法从错误中学习;轨迹级 DPO 对比安全与不安全轨迹,但安全信号被分散到整条轨迹的所有 token 上,无法告诉智能体到底是哪一步出了问题。D2PO、GRAPE 等轨迹级偏好方法都存在这个粒度错配。

SafeBranch 的研究缺口由此明确:需要一种监督形式,能够在安全关键步上提供集中的步级安全信号,同时不增加部署时的推理开销。

核心创新

  • 提出分支对(branch pair)监督形式:同一安全关键上下文下仅动作不同的安全/不安全对比,直接传递步级安全信号。
  • 提出 SafeBranch 数据构造框架:通过环境回滚将单条不安全轨迹转化为同状态分支对,效率比自然 best-of-N 采样高约 5.2 倍。
  • 提出 BranchPO 目标:在共享锚点上累积安全-不安全 log-probability 边际,将 critic 的安全判断内化到智能体。
  • 构建 OOD-ObjectShift 与 OOD-TaskShift 两个受控分布外基准,分离感知扰动与目标替换对安全泛化的影响。

方法概览

SafeBranch 从智能体自身的不安全 rollout 出发,由安全 critic 识别违规约束并给出反馈,将环境回滚到安全关键锚点步,在相同上下文下采样修复动作,形成仅在该步分叉的分支对;经 judge 过滤与跨温度去重后,用 BranchPO(DPO 形式、共享锚点的步级对比)对齐智能体,部署时无 critic、无回滚、无反馈。

  • 分支对的定义:在同一安全关键上下文下,两个输出共享完全相同的锚点前缀,仅在动作选择上分叉,一个安全、一个不安全。这种形式使安全决定动作变得显式,如图 1 所示。
  • 数据构造起点:让智能体自己跑出不安全 rollout,由安全 critic 识别违规并给出反馈。critic 的触发分为前瞻触发(动作执行前识别违规)和回顾触发(轨迹完成后识别残留危险并定位责任步)。
  • 环境回滚:将模拟器状态恢复到安全关键锚点步,在相同上下文下采样修复动作。这一步是 SafeBranch 区别于自然 best-of-N 采样的关键——它保证分支对共享锚点,而不是两条独立轨迹。
  • 过滤与去重:原始分支对经过 judge 过滤(判断修复动作是否真正安全且合理)和跨温度去重,从 753 对降到 475 对最终训练对。过滤阶段的人类可用率逐步提升,说明过滤确实提高了数据质量。
  • BranchPO 目标:在共享锚点上累积安全-不安全输出的 log-probability 边际,是 DPO 的步级变体。与轨迹级 DPO 的区别在于,对比发生在锚点之后的分支部分,而非整条轨迹。
  • 训练流程:先做简短的 SFT 预热,再进行 BranchPO。训练提示中移除 critic 反馈,确保智能体在部署时不依赖外部信号。
  • 部署形态:测试时零 critic 调用、零回滚、零反馈,智能体直接输出安全感知的动作序列。

逐图理解论文

一个具体失败

一个具体失败
Table 1: Qualitative comparison of supervision signals on a hygiene task. Task: put a peach from the fridge onto a soiled plate. Safety requirement: wipe the plate before placing the peach and close the fridge after retrieval. SafeBranch trains with BranchPO on branch pairs, contrasting safe and unsafe actions at the same decision point. Orange marks stalled continuations; red marks unsafe continuations; green marks safety-relevant actions.

表 1 用卫生任务定性对比不同监督信号。注意橙色标记的停滞延续、红色标记的不安全延续和绿色标记的安全相关动作,SafeBranch 的分支对在相同决策点上直接对比安全与不安全动作。

研究缺口

研究缺口
Figure 1 illustrates this branching view of inter- active safety. A branch pair places the actor in the same situation and contrasts two possible actions at a safety-critical step: one safe and one unsafe. By making the two branches differ only at that step, the branch pair makes the safety-determining choice explicit and provides the actor with a direct step-level safety signal.

这是图 1 的说明性变体,强调分支对使安全决定选择变得显式。对比两个分支的动作差异,可以直观理解为什么步级对比比轨迹级对比更能传递安全信号。

SafeBranch 的贡献

SafeBranch 的贡献
Figure 2: Overview of SafeBranch. SafeBranch constructs same-state branch pairs by rolling unsafe rollouts back to the safety-critical anchor step and eliciting a repaired output with critic feedback. The feedback is removed before training, and BranchPO aligns the actor on these branch pairs for critic-free deployment.

图 2 是 SafeBranch 的完整流程概览。关注从不安全 rollout 到环境回滚、再到修复动作采样和 BranchPO 训练的闭环,以及 critic 反馈在训练前被移除的关键设计。

证据与结论

证据与结论
Table 2: Main results on IS-Bench and the two OOD benchmarks constructed from it (ObjectShift injects distractors; TaskShift substitutes target objects). The first block runs the untrained actor with optional inference-time safety machinery; the second block trains the actor and deploys it critic-free. For each column, bold marks the best and underline the second-best.

表 2 是主结果表。第一块是未训练智能体加推理时安全机制,第二块是训练后无 critic 部署。对比 BranchPO 与 Trajectory DPO 和 SFT 的 SSR 与 SRec,可以验证步级信号的优势。

意义与局限

意义与局限
Figure 3: Analysis of branch pair construction. (a) Data generation efficiency. Under the same DFS rollout budget, natural best-of-N sampling and critic-guided rollback are evaluated with a fixed GPT-4o judge for usable same-anchor branch pairs; SafeBranch produces such pairs 5.2× faster. (b) Pair quality through filtering. Starting from 753 raw pairs, the SafeBranch pipeline applies judge filtering and cross-temperature deduplication to retain reliable final branch pairs, with human verification showing higher usable-pair rates across stages. (c) Downstream effect of filtering. Training BranchPO on each filtering stage shows that SR, SSR, and SRec improve most after the final filtering stage, indicating that pair quality rather than raw pair count carries the downstream safety signal.

图 3 分析分支对构造。子图 (a) 对比自然 best-of-N 与 critic 引导回滚的效率;子图 (b) 展示过滤漏斗;子图 (c) 显示过滤阶段对下游 SR/SSR/SRec 的影响,说明数据质量比数量更重要。

实验如何设计?

  • 主评测集为 IS-Bench,包含 161 个任务,覆盖多种交互式安全约束。
  • 作者构造了两个受控分布外基准:OOD-ObjectShift(147 任务,注入干扰物)和 OOD-TaskShift(138 任务,替换目标物体),分别测试感知扰动和目标替换对安全泛化的影响。
  • 跨模拟器迁移在 SafetyALFRED 上测试,包含 222 条危险轨迹、617 个危险轮次、五类风险。
  • 对比基线包括推理时方法(Self-Verification、Lookahead)和训练方法(SFT、Trajectory DPO、success-matched DPO)。
  • 消融实验包括:移除训练提示中的 critic 反馈(Ours vs +FB)、SFT 预热的作用(SafeBranch vs BranchPO-only)、以及推理时全 critic 的运行时成本对比。
  • 数据构造效率实验对比自然 best-of-N DFS 采样与 critic 引导回滚在相同 rollout 预算下产生可用分支对的速度。

关键结果与论文证据

  • IS-Bench 主结果(第 6 页):BranchPO 将 SSR 从 0.031 提升到 0.281,SRec 从 0.273 提升到 0.467,且部署时无 critic。
  • OOD 泛化(第 6 页):OOD-ObjectShift 上 SRec 提升 34.6 个百分点,OOD-TaskShift 上提升 50.0 个百分点,说明步级安全信号对感知扰动和目标替换都有强泛化。
  • 跨模拟器迁移(第 7 页):SafetyALFRED 上危险准确率从 0.274 提升到 0.438;Property Damage 提升 39.4 个百分点,Appliance Misuse 提升 15.4 个百分点。
  • 数据构造效率(第 8 页):SafeBranch 比自然 best-of-N DFS 采样快约 5.2 倍,说明 critic 引导回滚是更高效的分支对来源。
  • 过滤阶段的作用(第 8 页):训练在最终过滤阶段后的分支对上,SR、SSR、SRec 提升最大,说明数据质量而非原始数量决定下游安全信号。
  • 反馈消融(第 21 页):移除训练提示中的 critic 反馈(Ours)与保留反馈(+FB)在无 critic 评估下表现相当,说明 BranchPO 已将安全信号内化,不依赖提示中的反馈。
  • 运行时成本(第 23 页):推理时全 critic 在 30 步预算下需要 9510 次额外 GPT-4o 调用,而 SafeBranch 测试时零 critic 调用。
  • 局限类别(第 7 页):SafetyALFRED 上 Fall/Trip Hazard 与 Spoilage 类别无法合成训练对,对应类别无提升;Unsanitary 类别被基础 VLM 预解决(约 35%),基线本身较高。

阅读时需要注意

  • 训练阶段仍需安全 critic,SafeBranch 只是将 critic 成本从部署转移到数据构造,并未完全消除对 critic 的依赖。
  • 依赖支持环境回滚的模拟器;扩展到物理系统需要近似世界模型或人工重置,实际部署受限。
  • SafetyALFRED 上部分风险类别无法合成训练对,导致对应类别无提升,说明方法对 critic 可识别且可回滚的违规类型更有效。
  • OOD 基准由作者从 IS-Bench 构造,非独立第三方基准,泛化结论的外部效度有限。
  • 报告结果来自固定评估协议,未报告多种子误差条,统计稳健性有待进一步验证。

关联工作

  • IS-Bench 是本文的主要评测集,定义了交互式安全的评估协议;SafetyALFRED 提供跨模拟器的五类风险分类。
  • 推理时安全护栏方法(HomeGuard、Safety Guardrails for LLM-Enabled Robots)在部署时调用外部模块,与 SafeBranch 的无 critic 部署形成对比。
  • C-MCTS 和 RoboMonkey 分别代表基于搜索和测试时采样的安全规划方法,都在推理时消耗大量计算。
  • D2PO 和 GRAPE 是轨迹级偏好优化方法,将安全作为多目标之一,但缺乏步级信号;TCPO 是步级偏好优化,但用于决策推理而非具身安全。
  • CHOP 使用反事实人类偏好标签进行导航避障,与分支对的构造思路有相似之处,但依赖人工标注而非环境回滚。
  • SafeVLA 是 VLA 安全对齐的约束学习方法,与 SafeBranch 在目标上相关,但 SafeBranch 专注于 VLM 规划器而非底层控制。

发表评论