← 返回 2026-10-02

智能体先验引导的策略学习 Agent Priors-guided Policy Learning

Puming Jiang, Tianrun Hu, Haozhe Du, Yibo Li, Zhiwei Xue, Xinhu Li, Harold Soh 📅 2026-09-28 👍 50 2026-11-11 18:30
LLM智能体 分布外泛化 扩散策略 技能组合 机器人操作 模仿学习

用技能的结构先验同时塑形训练泛化并指导运行时技能选择组合

前置知识

扩散策略(Diffusion Policy)

将机器人策略表示为条件去噪扩散模型:给定观测历史,从高斯噪声出发迭代去噪,直接生成一段未来动作序列(action chunk)。训练时对演示动作加噪并预测噪声(epsilon-prediction),通常用一维 U-Net 实现,DDPM 训练、DDIM 采样。它能表达多峰动作分布、拟合精细接触操作,是近年少样本模仿学习的主流骨干。

APPL 中所有技能策略与两条基线(B0、SinglePrior)都以条件 1D U-Net 扩散策略为骨干(Exp.1 每模型 20,000 步更新、预测 16 步执行前 4 步),先验正是通过改表示、改动作坐标系、加辅助损失作用在这一骨干上,不懂扩散策略就读不懂先验的"训练实现"。

结构先验(归纳偏置)

对"行为依赖哪些关系"的显式假设,例如抓取只依赖夹爪相对物体的位姿,与物体在世界坐标系的绝对位置无关。实现上可把动作或观测变换到物体坐标系 $x_{loc}=R_t^\top(x_{world}-p_t)$,也可设计等变网络或辅助损失。先验决定策略在演示之外的泛化方向:猜对先验,少样本也能外推;猜错则被错误对称性束缚。

本文全部贡献建立在"同一个先验既塑形训练又描述接口"的双重角色上,先验概念是理解其动机、方法与消融实验的核心词汇。

技能泛化与组合泛化

技能泛化指单个技能策略从演示未覆盖的起始状态(物体移位、前一技能留下的中间状态)仍能达成子目标,文中用成功区域 $W(\pi_i)$(成功概率至少 $1-\epsilon$ 的起始观测集合)刻画;组合泛化指通过跳过、重排或新拼接已学技能去解演示中不存在的任务。两者耦合:新组合会产生技能泛化必须覆盖的入口状态。

论文的问题定义、两类 OOD 评测(motion-level / task-level)与 composition 划分、以及"接口要忠实且有信息量"的形式化($A_i\subseteq W(\pi_i)$ 且 $A_i\approx W(\pi_i)$)都建立在这对概念上。

任务与运动规划(TAMP)的符号接口

TAMP 把任务层(离散符号规划)与技能层(连续控制)分离,用符号算子的前置条件与效果连接,例如"抽屉已关才能执行开抽屉"。符号可人工编写,也可从数据共同发明,如 SymSkill 联合学习谓词、算子与相对系动力学系统技能。符号接口精确但抽象,丢失了技能在物理上真正可行的连续区域信息。

APPL 的出发点正是"符号/语言接口看不到策略的泛化结构"这一信息缺口,理解 TAMP 才能明白它替代与补充了什么,以及与 SymSkill 的三点区别。

工具增强的 LLM 智能体

以大语言模型为决策核、通过工具调用(读写代码与文档、运行检查、提交产物、观测状态)完成多阶段任务的系统,通常配合固定提示词与冻结工具集。本文有两个:离线构建智能体负责分段、提出先验、实现策略、分析验证结果;在线运行时智能体每回合读取接口文档、选择策略、绑定类型化参数并设定停止条件,全程不改权重。

APPL 的构建与部署由这两个 LLM 智能体驱动(Exp.1 设计用 GPT-6 Astra、Exp.2 构建与运行用 GPT-5.6 Sol),附录 C 逐字公开全部提示词,模型与提示词直接影响结果与可复现性。

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

在海量机器人数据上预训练的单体模型,把摄像头图像与语言指令直接映射为动作序列(如 π0.5 有 4.14B 参数),靠数据广度而非显式结构获得泛化,但对布局与物体位姿变化的鲁棒性仍受限;也可作为技能库被高层智能体按名字调用。

Exp.2 的 Agent+VLA 基线把 π0.5 在与 APPL 相同的 204 个技能段(112,761 帧、17 条指令)上微调,由同一运行时智能体调用,用来检验"语言名字接口"与"先验文档接口"的差别。

研究动机

机器人只能从昂贵的少量完整演示学习(Exp.1 每任务仅 2–20 条、Exp.2 每任务 12 条),部署却要面对新目标、新操作顺序与移位的物体。这需要两类互相依赖的泛化,但任务层与技能层之间信息丢失:技能在哪里可行由其策略训练所用的结构决定,组合层却只能通过名字、指令或符号算子这种"另一套描述"看到技能。后果有具体数据支撑:朴素 Diffusion Policy 在六个 MetaWorld 任务上平均 OOD 成功率仅 37.86%(2 条演示时 28.96%);ManiSkill 五个长程任务上全任务 DP 移位物体后成功率仅 10.0%。VLA 靠数据广度但 LIBERO-Plus 等鲁棒性基准显示其对布局与位姿变化脆弱;TAMP 的符号前置条件要么手工编写、要么从预定义族学习(最接近的 SymSkill 被固定在相对坐标系+动力学系统控制器这一族);LLM 智能体调用技能工具时只知名字与可行性估计,技能间交接是常见失败来源。

本文的目标是在只给定 $N$ 条完整无分割演示 $\mathcal{D}_N$ 与任务知识 $K$ 的条件下,离线构建一个冻结系统 $S=(C,L)$:策略库 $L$ 允许同一技能保留多个策略实现,组合器 $C$ 在运行时只通过接口 $I_i$ 看到并依次调用策略,最大化测试目标分布上的期望成功率 $J(S)=\mathbb{E}_{g,o_0}[Succ(g,o_T)]$。更精确地说,要求学到的技能策略其成功区域 $W(\pi_i)$ 覆盖新组合可能产生的入口状态,同时每个接口声明的适用区域 $A_i$ 满足两条性质:忠实($A_i\subseteq W(\pi_i)$,不把策略调到会失败的地方)且有信息量($A_i\approx W(\pi_i)$,不放弃本可成功的调用)。构建过程不得使用测试分布信息,部署后库与智能体全部冻结、每回合记忆重置。

与已有工作不同的是,本文的独特切入是把每个技能的"结构先验"提升为组合层与技能层之间的接口:同一先验既作为训练时的归纳偏置实现进表示、动作坐标系或辅助损失,塑造策略的泛化范围;又用自然语言连同交接条件、训练支持与验证证据写进运行时文档,让组合层知道每个实现"为何可信、何处适用"。与最近邻工作的差别明确:LGA/KALM 让模型选表示先验,但每任务只产出一个设计且只用于训练;SymSkill 的结构被固定在相对系+动力学系统这一族,每个算子绑定单一策略;Harness VLA、BATON 等 agentic 系统在策略存在之后才从执行记录学习操作范围。APPL 则让构建智能体按技能从开放设计空间提议多个先验、保留备选策略,并使先验信息在第一次执行之前就可被运行时智能体读取。

核心方法

直觉是:在物体相对坐标系中训练的抓取策略天然能迁移到新的物体位置——但只有把这一点写成文档告诉任务层,任务层才会在物体移位时主动选它。APPL 据此分两阶段。离线构建:构建智能体把 $N$ 条完整演示分割成可复用技能(相邻技能段在交接处有意重叠),为每个技能提出约 3 个预期失败模式互补的结构先验,逐先验实现并训练一个条件扩散策略,再从演示入口状态验证每个策略、据反馈增补第 4 个先验,最后为每个策略写接口 $I_i=(d_i,h_i,s_i,v_i)$(先验描述、交接条件、训练支持、验证证据)并冻结库。在线组合:运行时智能体读取目标谓词、交互历史与回合笔记本,可检视任一策略的完整接口,选择策略、绑定类型化参数、设定时长(至多 300 步)与数值停止条件;执行器以新鲜观测驱动策略并逐步检查条件,触发即交还控制权,任务目标达成时立即中断调用。

核心创新是结构先验的双重角色及其对接口性质的耦合。训练侧(coverage),先验 $\rho_i$ 限定策略类 $\Theta_{\rho_i}$ 并可加辅助损失:$\theta_i=\arg\min_{\theta\in\Theta_{\rho_i}}\mathbb{E}_{\mathcal{D}_{k(i)}}[\ell_{BC}(\theta)+\lambda_i\,\ell^{aux}_{\rho_i}(\theta)]$,由此塑造策略演示之外的行为,即成功区域 $W(\pi_i)$。描述侧(selection),同一先验给出适用性假设 $A_i=\{o:\phi_{\rho_i}(o)\in R_i\}$,其中 $\phi_{\rho_i}$ 是先验定义的观测变换、$R_i$ 是变换后演示覆盖的区域;接口通过先验描述、实测训练支持与交接条件把这一假设传达给运行时智能体。因为声明的适用性建立在塑造策略的同一假设上,$A_i\subseteq W(\pi_i)$ 的忠实性获得机制性支撑(虽不保证成立)。这把"选什么策略结构"从一次性设计决策变成可提议、可实现、可在运行时复选的按技能假设,并让同一技能的多个先验版本互为备选。

方法步骤详情

第一步,技能分段:构建智能体逐条轨迹提议边界,把执行同一可复用操作的段归为技能数据集 $\mathcal{D}_k$;相邻技能包含前任结尾与后继开头的重叠动作监督,用原演示的转移拓宽交接处的训练支持而不新增数据。第二步,提议先验:每技能从不同族提出约 3 个"预期失败模式互补"的先验,至少一个针对被操纵物体的位移。第三步,实现与训练:按固定接口写代码——构建因果观测历史、在先验指定坐标系编码动作并解码回原生命令,可加辅助损失;例如物体系先验用 $x_{loc}=R_t^\top(x_{world}-p_t)$、$\Delta x_{loc}=R_t^\top\Delta x_{world}$、$\Delta x_{world}=R_t\Delta x_{loc}$ 变换末端目标,经共享逆运动学解算为关节指令,每个先验独立训练一个条件扩散策略。第四步,验证:每个策略从其训练段的演示入口状态启动、运行 1.5 倍段长、按预定义退出判据在末态评分;智能体只读这些结果写报告,并为每技能增补第 4 个先验 h04 再验证一次。第五步,写接口 $I_i=(d_i,h_i,s_i,v_i)$ 并声明类型化参数(操作对象、目的地等)。第六步,运行时组合:$(i_j,u_j,m_{j+1})\sim R_\alpha(\cdot\,|\,h_{t_j},g,m_j,\mathcal{L})$ 选出策略调用与参数,执行器逐步检查数值停止条件(如夹爪宽度小于 3 cm 且物体抬升超过 4 cm),任务目标达成即中断;智能体随后可继续当前技能、换实现、换技能或结束任务。

技术新颖性

与最近邻方法的本质区别有三点。其一,对 SymSkill:其结构被固定在相对系谓词+动力学系统控制器这一预定义族,每个算子绑定单一技能策略,规划器读符号前置/后置条件;APPL 让构建智能体按技能从开放设计空间(表示、动作坐标系、编码器、辅助损失)提议不同族的先验,同一技能保留多个策略由运行时智能体选择,接口是自然语言先验+交接重叠+实测训练支持+验证证据。其二,对 LGA/KALM/ATK:它们也用(视觉)语言模型选表示先验,但每任务一个设计、先验只作为策略输入用于训练;APPL 的先验同时写进训练目标与运行时文档,这是作者明确主张的真正新意。其三,对 Harness VLA、BATON、RoboHarness 等 agentic 调度系统:它们在策略存在后从执行记录学习入口条件与操作范围,APPL 的文档来自构建时的先验与训练数据的交接重叠,首次执行前就可用;Agentic Skill Discovery 虽保留每技能多策略变体但把选择留作未来工作,APPL 正是用先验补上了这一选择环节。

APPL builds prior-specific policies offline and composes them online. The construction agent segments demonstrations, proposes several priors per skill, trains one documented policy per prior, and verifies each policy from demonstrated entry states. The runtime agent reads these documents to choose a policy, its arguments, a duration, and stop conditions. Orange marks the prior's two roles.
Figure 1: APPL builds prior-specific policies offline and composes them online. The construction agent segments demonstrations, proposes several priors per skill, trains one documented policy per prior, and verifies each policy from demonstrated entry states. The runtime agent reads these documents to choose a policy, its arguments, a duration, and stop conditions. Orange marks the prior's two roles.

实验结果

Exp.1(MetaWorld 六任务、嵌套演示数 $N\in\{2,5,10,20\}$、共 144 个模型、隐藏测试):APPL q4 的宏平均 OOD 成功率 92.40%,对比朴素扩散策略 B0 的 37.86% 与固定关系先验 B1 的 47.29%;$N=2$ 时 89.58% vs 28.96%/37.92%,对 B1 的领先在四档 $N$ 下分别为 51.67/50.83/41.04/36.88 个百分点。流程分解:首次提议 q1 已超 B1 18.1–22.5 点;三选一开发集选择 q3 再加 10.2–24.0 点;验证反馈轮 q4 又加 2.9–12.1 点(24 个条件中 15 个选中 A4,测试上 10 升、13 平、1 降)。$N=20$ 时外推区 E 成功 B0 25.42%、B1 36.25%、q4 88.33%。典型修订:drawer/$N{=}2$ 用可逆残差动作编码 $z_{xyz}=L_\theta a_{xyz}+4P_{xy}e$ 从 75.00% 升至 100.00%;assembly/$N{=}5$ 移除目标位置捷径从 41.25% 升至 93.75%;也有负例 door/$N{=}20$ 从 100.00% 降至 96.25%。Exp.2(ManiSkill 五个长程任务、每任务 12 条演示):移位物体 OOD 上 APPL 50.0%,DP 与 SinglePrior 均 10.0%(McNemar $p<0.001$),Agent+VLA 32.5%;任务级 OOD 92.5%(重入 18/20、前缀目标 19/20);16 个未见组合解出 8 个(Agent+VLA 6/16)。消融证明接口信息本身关键:同一策略库隐藏全部接口信息后三指标跌至 20.0%/65.0%/2-16;仅恢复验证分数与退出判据(w/o prior information)升至 45.0%/72.5%/3-16;用固定规则替换智能体为 22.5%/52.5%/1-16。

Exp. 1 hidden-test success (%), averaged equally over six tasks, with N demonstrations per task. IID uses fresh states within the demonstrated factor ranges; OOD averages recombined (C) and extrapolated (E) states equally. Mean averages OOD over the four values of N. Bold marks the best entry in each column. q1 is the first proposal (A1); q3 and q4 are selected by development OOD success from A1–A3 and from A1–A4.
Table 1: Exp. 1 hidden-test success (%), averaged equally over six tasks, with N demonstrations per task. IID uses fresh states within the demonstrated factor ranges; OOD averages recombined (C) and extrapolated (E) states equally. Mean averages OOD over the four values of N. Bold marks the best entry in each column. q1 is the first proposal (A1); q3 and q4 are selected by development OOD success from A1–A3 and from A1–A4.
Exp. 2 first-attainment success. Each task has eight motion-OOD (M) and eight task-level-OOD (T) cases, with one episode per case; totals average over five tasks. Comp. reports successes on 16 new compositions. DP and SinglePrior have no goal input (–). Bold marks the best entry per column.
Table 2: Exp. 2 first-attainment success. Each task has eight motion-OOD (M) and eight task-level-OOD (T) cases, with one episode per case; totals average over five tasks. Comp. reports successes on 16 new compositions. DP and SinglePrior have no goal input (–). Bold marks the best entry per column.
Verification successes from the twelve demonstrated entry states of each skill policy (h01–h03 original priors, h04 added after verification). The last row per task starts the full-task policies from the demonstrated initial states.
Table 8: Verification successes from the twelve demonstrated entry states of each skill policy (h01–h03 original priors, h04 added after verification). The last row per task starts the full-task policies from the demonstrated initial states.
Paired comparisons with full APPL on the same saved cases. Wins are cases only APPL solves; losses are cases only the comparator solves. p is the two-sided exact McNemar value.
Table 9: Paired comparisons with full APPL on the same saved cases. Wins are cases only APPL solves; losses are cases only the comparator solves. p is the two-sided exact McNemar value.
Two roles of task structure across related families.
Table 10: Two roles of task structure across related families.
Design comparison with the closest methods. “Reads” is what the high-level decision maker sees about each low-level policy before or during invocation.
Table 11: Design comparison with the closest methods. “Reads” is what the high-level decision maker sees about each low-level policy before or during invocation.
Five long-horizon tasks at their demonstrated start states. The robot exchanges blocks between a drawer and a pad, swaps two blocks through a buffer, stores a block from a low tunnel in a drawer, uncovers and inserts a peg, and pours particles into a bowl before returning the container.
Figure 2: Five long-horizon tasks at their demonstrated start states. The robot exchanges blocks between a drawer and a pad, swaps two blocks through a buffer, stores a block from a low tunnel in a drawer, uncovers and inserts a peg, and pours particles into a bowl before returning the container.
Demonstrated and evaluation factor layouts. Support points are the 20 demonstrations per task; test states were not available during design.
Figure 3: Demonstrated and evaluation factor layouts. Support points are the 20 demonstrations per task; test states were not available during design.
Test OOD success for each task and system (C and E weighted equally). Designs can differ across N, so the curves show the design procedure across data sizes, not the scaling of one architecture.
Figure 4: Test OOD success for each task and system (C and E weighted equally). Designs can differ across N, so the curves show the design procedure across data sizes, not the scaling of one architecture.
查看结构化数据
任务指标本文基线提升
MetaWorld 技能泛化(六任务宏平均) 隐藏测试 OOD 成功率,N=2 条演示 89.58%(APPL q4) 28.96%(vanilla DP);37.92%(固定关系先验 B1) +60.62 / +51.66 个百分点
MetaWorld 技能泛化(六任务宏平均) 隐藏测试 OOD 成功率,四档 N 平均 92.40%(APPL q4) 37.86%(B0);47.29%(B1);67.92%(q1 首次提议) +54.54 / +45.11 / +24.48 个百分点
MetaWorld 外推区(E) N=20 时的外推成功率 88.33%(q4) 25.42%(B0);36.25%(B1) +62.91 / +52.08 个百分点
ManiSkill 五任务移位物体 OOD 首达成功率(8 布局×5 任务) 50.0%(APPL) 10.0%(全任务 DP);10.0%(SinglePrior);32.5%(Agent+VLA) +40.0 / +40.0 / +17.5 个百分点(对 DP/SinglePrior p<0.001)
ManiSkill 任务级 OOD(中途重入+前缀目标) 首达成功率(8 案例×5 任务) 92.5%(APPL) 70.0%(Agent+VLA);65.0%(隐藏接口信息消融) +22.5 / +27.5 个百分点
ManiSkill 未见技能组合 16 个新组合的解出数 8/16(APPL) 6/16(Agent+VLA);2/16(隐藏接口信息消融) +2 / +6 个案例

局限与改进

作者承认的局限:结构先验只描述"意图上的泛化",不保证能力,策略仍可能在演示支持之外失败;库中缺少恢复类技能;验证协议只允许演示入口状态、按固定时长末态评分,可能误导选择——覆盖销装配中 h03 的 0/12 分数掩盖了它在第 347–384 步间曾瞬时满足交接判据(570 步才评分),带着该报告的完整 APPL 在该任务上仅 1/8,反而低于不做验证的 8/8。我自己的观察:其一,成功按"首次达成目标"计并有执行器辅助停止,未评估目标状态的持续性——案例 C-N3 中方块以约 1.66 m/s 的平均速度下落途中被记成功;其二,motion-OOD 划分是在看过更难划分的结果之后把位移减半选定的,四个 drawer 组合案例也在结果出来后才被重分类,评测协议存在事后选择成分;其三,构建与运行都依赖 GPT-6 Astra(max reasoning effort)与 GPT-5.6 Sol 这类前沿闭源模型,运行时每回合最多 128 次请求且仿真暂停等待推理;其四,全部实验是结构化状态输入的仿真操作(仅 VLA 基线用图像),视觉观测与真机迁移未验证。

独立分析的弱点

弱点一:验证与部署分布错配。只在演示入口状态末态评分,既发现不了"瞬时满足交接条件"的策略(h03 案例),也可能在分布移位下误排策略(覆盖销装配 8/8→1/8 的反转);改进方向是把验证对象定义为"策略+入口分布+交接规则"的组合,报告判据满足的时间区间而非仅末态。弱点二:无恢复能力。演示不含失败轨迹,物体掉落、抓取把抽屉碰关后库中无技能可用;可增补恢复技能族,或在运行时检测到先验文档声明的"预期失败签名"时触发性地采集定向数据。弱点三:忠实性无保证。$A_i\subseteq W(\pi_i)$ 只是假设,先验错误或网络未实现先验时智能体会系统性选错;可引入廉价的运行时前置检查或 SymSkill 式反应式恢复作为回退。弱点四:对前沿 LLM 依赖重、推理成本高(每回合至多 128 次请求、仿真暂停等待推理),真机实时性差;可把运行时决策蒸馏为小模型或缓存常见情形。弱点五:每技能 3–4 个策略加验证轮带来约 4 倍训练与评测开销,而对先验差异无意义的技能(如各先验都 12/12 的 close_drawer)纯属浪费,成本应按技能自适应分配。

未来方向

作者提出的方向:改进验证使其与部署时的入口分布和交接规则对齐(在真机上无法做 OOD 试验时尤为关键);通过定向数据采集或数据增强扩展技能的支持范围;集成运动规划器等互补工具以弥补学习策略的失败模式。基于本文成果可延伸的:把 SymSkill 的已学谓词与反应式恢复嵌入 APPL 接口,作为执行检查与回退行为,形成互补;让运行时的调用记录反哺构建——检测到 $A_i$ 与实测行为不符时自动更新文档或触发再训练,构成闭环;把接口文档从自然语言推进为可执行的前置条件检查;在视觉观测与语言指令端到端的设定下检验先验接口是否仍有效;研究并行技能、动态障碍等更丰富组合类型下的组合泛化边界;以及系统比较"多先验备选+运行时选择"与"更强单先验"的算力-性能权衡。

复现评估

文档与代码基础属上乘:作者承诺发布实验代码、智能体生成的全部策略实现、固定提示词、评估用例与回合结果;附录 A/B 详列任务改造、数据划分、训练超参(1D U-Net、AdamW 学习率 $10^{-4}$、Exp.1 每模型 20,000 步、Exp.2 中 APPL 策略 20,000 步而全任务策略 60,000 步、DDPM 100 步)与评估协议,附录 C 逐字公开设计、分段、验证、运行时四类智能体的提示词。硬门槛有三:其一,构建/设计需要 GPT-6 Astra(reasoning effort max)、运行时与使用说明需要 GPT-5.6 Sol 等特定闭源模型,重跑会产生不同设计与决策,数值不会精确一致;其二,算力可观——Exp.1 训练 144 个扩散策略,Exp.2 每技能 4 个先验策略外加微调 4.14B 参数的 π0.5(112,761 帧、20,136 步、bf16 混合精度),需要多卡 GPU 数天量级;其三,流水线环节多(分段、先验实现、验证、文档、运行时),工程复刻成本高。总体评级:中高难度,适合有仿真操作与 LLM 智能体工程经验的团队。

相似论文推荐