A GPU-Parallel Framework for Heterogeneous Multi-Task Reinforcement Learning
发表时间:2026-10-06
arXiv · HF 讨论 · alphaXiv① 一句话定位
本文处在 GPU 并行机器人强化学习、多任务强化学习与演示引导学习的交叉点,做出两项相互耦合的贡献。其一,Hebero 是一个异构 GPU 并行 Isaac Lab 基准,把 LIBERO 的 40 个任务(Goal、Object、Spatial、Long 四套件各十项)编译进同一个仿真进程,提供统一的状态/视觉观测接口、演示轨迹与标准化评测。其二,DGPO 是一个演示引导的多任务 PPO 训练框架,并以 IW-ABC 为参考配方:复用演示构造稠密跟踪奖励与非对称价值函数,再用一个轻量的逐任务成功率 EMA 信号同时协调自适应行为克隆与面向滞后任务的重要性加权。论文的核心立场是:GPU 吞吐量本身不能解决稀疏探索与多任务不平衡,现有基准很少同时具备异构 GPU 并行、演示与标准化多任务评测;在固定墙钟预算下扩展并行副本能提升成功率,且 IW-ABC 在同预算下大幅超过 FAMO-ABC,并有零更新真机迁移证据。
② 背景与动机
背景可从三条线理解。第一,基准演进:早期 CPU 基准如 RLBench 提供百余任务、演示与多任务设置,Meta-World 定义联合多任务与元 RL 评测,LIBERO 组织富含演示的目标/物体/空间/长程套件;但它们的原始执行模型不支持单个 GPU 并行进程中的数千异构环境。后续 GPU 系统如 ManiSkill3 提供异构物理与渲染、RoboVerse 统一仿真与数据接口、MTBench 把 Meta-World 移植到 Isaac Gym 做状态 MTRL,另有 MO-Playground 的多目标 RL 与 MMBench/Newt 的大规模多任务方向。Hebero 试图把异构 GPU 并行、多任务与演示三者合并。第二,演示引导 RL:DAPG 用演示初始化并正则化策略,RLPD 把离线数据并入在线离策略学习,Rainbow-DemoRL 改进演示增强 RL,Data-Efficient Multitask DAgger 按任务进度分配额外专家数据,RFCL 用演示状态构造反向重置课程;参考跟踪提供稠密模仿奖励,若干工作学习可复用稠密奖励或残差/可解释奖励。DGPO 通过共享栈把演示的不同用法组织为可对比接口。第三,多任务优化:梯度幅度均衡、PCGrad 的梯度几何修改、FAMO 的损失动态加权、PolicyGradEx 的任务亲和度分组等处理任务进展不均与冲突更新。IW-ABC 则用 rollout 内逐任务成功率统计做重要性加权,与奖励尺度无关、与演示接口正交。
③ 方法机制
【问题与基准编译】采用标准 MTRL 形式:任务 T_k 是有限视界 MDP (S_k,A,P_k,R_k,H,γ),p(T) 在 K=40 上均匀;学习单一任务条件策略 π_θ(a_t|o_t),o_t 含任务编码 z_k。Hebero 把任务语义与仿真执行解耦:LIBERO BDDL 任务在训练前编译为缓存 JSON 描述符(指令、场景、资产、生成区、目标谓词与阈值),目标谓词从实时仿真状态求值,任务成功为各谓词合取,兼作稀疏奖励事件与评测结果。MuJoCo MJCF 资产离线转成 USD 缓存,保留几何、外观、动力学、关节与限位。异构向量化仿真中,环境 i 被指派任务 i mod K(总环境数 N=K·g);一个仿真器/渲染器、一个 rollout buffer 与一次策略更新服务全部任务。每个资产原型只保留一个批量视图,覆盖实例化它的环境子集,维护全局环境 ID 与局部视图索引的缓存映射;任务绑定保留物体初始位姿与语义角色。 【交互接口】动作 a_t∈[-1,1]^7:前三/后三维经操作空间控制器产生末端相对平移与旋转 Δp=s_p·a_{1:3}、ΔR=Exp(s_R·a_{4:6}),第七维符号控制夹爪。状态观测 o^state=[a_{t-1},z_k,o^proprio,o^object-target] 共 300 维:上一动作 7 维、z_k 36 维多热编码(6 个共享子任务指示 + 30 个任务 ID 位)、本体感觉 23 维、物体-目标位姿缓冲 234 维(26 实体槽×9 特征,未用槽置零)。视觉观测用冻结 ViT 池化特征替换位姿缓冲,共 450 维。任务奖励 r^task=r^goal−r^reg−r^safe:稀疏成功奖励 w_succ·1[g_k(s_t)],减去动作平滑/动作范数/关节速度正则与关节位置/速度软限位惩罚。评测在 50 条演示起点框定的 x、y 范围内均匀重采样物体位姿并做碰撞拒绝,测 Mean SR、Long SR、任务覆盖(≥80% 成功)与 SR-AUC。 【DGPO 共享组件】命令流统一管理各并行环境的任务身份、活动演示与时间步游标,并从共享轨迹库缓存机器人/物体/夹爪/动作引用,供奖励、critic 与 learner 接口复用。演示跟踪奖励同时跟踪机器人运动与世界状态:r^demo=Σ_{m∈E_k} w_m·exp(−‖e_t^m‖/σ_{r,m}),E_k 含末端位姿、臂关节位置/速度、夹爪误差(机器人侧)以及刚体姿态、关节角误差(世界侧);稠密项既奖励机器人怎么动,也奖励动作在世界上产生的结果,并加入任务奖励。非对称 critic 在 actor 输入上追加跟踪误差与归一化演示游标 φ_t=m_e(t)/max(T_ξ−1,1),其中 m_e(t)=min(r_e+t,T_ξ−1),提供部署时不可得的特权监督;视觉 critic 用独立的可训练注意力池查询同一批冻结 token。 【参考 learner 与演示接口】共享栈固定,仅改变演示接口:PPO 无额外机制;BC→PPO 离线 actor 初始化;DAPG 离线似然正则;ABC 在线时间对齐动作目标;RFCL 反向重置课程(从演示后段状态反向推进前沿,精通后回退,全部归零后切到完整起点分布)。 【IW-ABC 核心机制】用同一个轻量学习进度信号协调模仿与任务优先级。(1) 进度信号:任务 k 的 τ_k 用新完成回合的二值成功均值初始化,之后 τ_k←(1−α_ema)τ_k+α_ema·SR̄_k(α_ema=0.05),直接从 rollout 计算,无需额外评测。(2) 自适应动作引导(ABC):β_{k,u}=β_min+(β_max−β_min)[(1−τ_k)(1−ρ_u)]^κ,ρ_u=min(u/U_ann,1),κ=2,β_max=1,β_min=0.1,U_ann=200,000 步;已解决任务可提前降低模仿压力,时间退火防止持续困难任务永远停在最大引导。损失 L_ABC=c_BC·E[β_{k,u}‖μ_θ(o_t)−a*_t‖²]。目标 a* 是时间索引动作 a^demo_{ξ(e),m_e(t)},游标每控制步推进一次,到演示末尾截断并重置;演示起点重置建立初始对齐,稠密跟踪奖励维持接近参考,系数随成功/时间释放。(3) 任务均衡(IW):τ̄ 为已初始化任务均值,p_k^IW=σ(s_IW(τ_k−τ̄))(s_IW=10),w_k=w_max(1−p)+w_min·p(w_max=2,w_min=0.5),低于均值的任务获得更大权重;小批量内归一化 w~_k=w_k/mean(w) 保持平均损失尺度。综合目标 L^{IW-ABC}=E[w~_k(ℓ^π+c_V·ℓ^V−c_H·H)]+L_ABC,IW 同时缩放策略、价值与熵,单次反向传播、无逐任务梯度。直觉:该信号与奖励尺度无关、与演示接口正交,故可叠加到 PPO 参考实现上。(4) FAMO-ABC 对照:只在 critic 侧用 FAMO 损失进度加权,D_k=L_k+ε、ω=softmax(q),目标 c·Σ sg(ω) log D_k(c=sg(Σ ω/D)),更新后重算 D_k 得 δ=log D_before−log D_after、g_k=ω(δ−Σωδ) 更新 logits;策略代理与熵保持均匀,ABC 保留。 【训练与数据细节】全部 learner 在 40 任务上 30,000 次在线 PPO 迭代,每任务 50 条演示;离线预处理数据集含 336,575 条状态-观测-动作记录。PPO:T_rollout=16 步、每 rollout 5 epoch、4 个小批量;Adam 学习率 2e-4,折扣 0.99,GAE 0.95,clip 0.15,c_V=1.0,c_H=0.005,自适应 LR 的 KL 目标 0.005,梯度裁剪 1.0。状态 learner 共享 [512,256,128] ELU MLP(前馈、观测 running 归一化、初始 σ=0.8),actor 约 0.32M 参数。视觉侧用冻结的 Theia 式 Tiny ViT(patch 16、维度 192、深度 12、224×224→196 token),双视图沿特征维拼接成 196×384,经可训练单查询注意力池 softmax(q^T Z/√(2D_v)) 得到 384 维 h^visual,critic 用独立查询 q_V;PPO 只训练注意力池与 MLP,部署总参数约 5.92M(含冻结编码器)。训练用 3 个种子、每 run 8 块 L20 GPU、每卡 2,000 并行环境,单 run 约两天。
④ 实验与证据
实验部分分若干条线展开。首先,基准吞吐与扩展性:把 Hebero 与 CPU 上的 MuJoCo 配置对比仿真吞吐与显存占用,并给出多卡端到端训练的规模;在固定墙钟预算下,增加每任务并行副本提高成功率并更快达到同等奖励水平。其次,在 DGPO 共享栈内做受控 learner 对比:覆盖纯 PPO、离线行为克隆初始化、在线自适应动作引导、DAPG 式正则、RFCL 式反向重置课程,以及 FAMO 式任务加权;结果显示在线自适应引导优于离线初始化与重置课程,基于成功率的任务加权在同等预算下胜过基于损失动态的 FAMO,视觉版本进一步改善,长程任务覆盖更广。再次,学习动力学:滞后的目标与长程任务获得更高重要性权重,动作引导系数随任务成功与训练时间退火,而任务优先级在退火后仍然保留。最后,真机部署:单条多任务状态策略在若干 Piper 任务上零权重更新部署,配合外部感知取得较高总成功率。此外还有消融验证任务加权应同时作用于 actor 与 critic,以及逐任务系数动力学分析。所有具体数值见随附的 claim-quote 对。
-
Hebero 在单块 L20 GPU、一千六百个并行环境、全部四十个任务上达到约七千五百步/秒,吞吐量是九十六核 MuJoCo CPU 基线的三点四倍。
Across 40 tasks and 1,600 environments, Hebero delivers approximately 7.5k simulation steps per second on one L20 GPU, 3.4 times the throughput of our 96-core MuJoCo baseline.
-
在固定墙钟预算下,单块 48 GB L20 GPU 上增加每任务并行副本提高成功率,并以更少时间达到相同奖励水平。
on a single 48 GB L20 GPU, more replicas per task yield higher success within the same wall-clock budget ( fig. 3 (a)) and reach matched reward levels sooner ( fig. 3 (b)).
-
IW-ABC 在状态输入下达到百分之九十点一的平均成功率,以零点三二兆参数的 actor 超过最强基线 FAMO-ABC(百分之八十二点三)七点八个百分点。
IW-ABC achieves the highest state-input Mean SR of 90.1% with a 0.32M-parameter actor, outperforming the strongest baseline, FAMO-ABC (82.3%), by 7.8 percentage points.
-
相对最强状态基线 FAMO-ABC,IW-ABC 将平均成功率从百分之八十二点三提升到百分之九十点一,将长程成功率从百分之五十三点三提升到百分之七十点零。
Compared with FAMO-ABC, the strongest state-input baseline, it improves Mean SR from 82.3% to 90.1% and Long SR from 53.3% to 70.0%.
-
视觉策略达到百分之九十三点五的平均成功率,总部署参数量五百九十二万,含冻结编码器。
The visual policy reaches 93.5% with 5.92M total deployed parameters, including its frozen encoder.
-
全部 learner 在四十个任务上联合训练三万次在线 PPO 迭代,每任务五十条演示,并采用统一评测协议。
All learners in Table 2 jointly train on 40 tasks for 30,000 online PPO iterations with the same 50 demonstrations per task and undergo unified evaluation protocol ( section 3.4 ).
-
每个配置训练三个种子,每 run 使用八块 NVIDIA L20 GPU、每块 GPU 两千个并行环境,耗时约两天。
We train each configuration with three training seeds; each run uses eight NVIDIA L20 GPUs with 2,000 parallel environments per GPU and takes approximately two days.
-
在 IW-ABC 中同时加权 actor 与 critic 相比只加权 actor,平均成功率提高十一点九个百分点,长程成功率提高二十点零个百分点,任务覆盖从四十个中的三十一个增至三十五个。
Weighting both actor and critic improves Mean SR by 11.9 percentage points and Long SR by 20.0 points over actor-only weighting, while increasing task coverage from 31 to 35 out of 40 ( table 5 ).
-
ABC、IW-ABC 与视觉 IW-ABC 分别覆盖十个 Long 任务中的四个、七个和九个,视觉版本长程技能覆盖最广。
ABC, IW-ABC, and Vis IW-ABC cover 4, 7, and 9 of the ten Long tasks, demonstrating broader coverage of long-horizon skills.
-
在真机上每任务二十次物理试次,联合训练的多任务策略达到百分之八十二点五的总成功率。
Across 20 physical trials per task, the policy achieves an overall success rate of 82.5%.
-
在固定任务重要性加权下,IW-ABC 在全部四项指标上超过 IW-DAPG 与 IW-RFCL,说明在线自适应动作引导优于离线正则化与反向重置课程。
With task IW fixed, IW-ABC outperforms IW-DAPG and IW-RFCL across all four metrics, supporting adaptive matched-action guidance over offline demonstration regularization and reverse-reset curricula for demonstration-guided joint learning.
⑤ 审稿人视角
Novelty:新颖性中等。基准层面,Hebero 的新意主要是系统集成而非新任务:把 LIBERO 的异构场景接入单个 Isaac Lab GPU 并行进程,用资产原型视图与环境-任务映射处理异构性,这是对 MTBench 同构移植的重要工程扩展;标准化评测(重采样起点、Mean/Long SR、任务覆盖、SR-AUC)有方法学价值。算法层面,稠密跟踪奖励、非对称 critic、BC 初始化、DAPG、RFCL 均非原创;ABC 是自适应退火模仿,IW 是按成功率 EMA 的加权,二者分别有先例。真正的新组合在于用同一个轻量任务进度信号同时调节模仿压力与 PPO 任务权重,并把不同演示接口放进共享 DGPO 栈做受控比较。整体属于系统加组合式贡献,单点算法创新有限。 Soundness:整体可信但有保留。优点:受控比较严格(同一 PPO 骨干、奖励、观测、训练预算、种子与评测协议),消融能分离 IW 放置位置与 learner 接口;动力学分析用系数轨迹支撑机制解释。顾虑:多数主实验仅 3 个种子,而 Long SR 方差很大(如 IW-ABC 70.0±10.0、视觉版 81.9±11.5),未报告显著性检验;真机验证用状态策略加外部感知,未证明视觉 sim-to-real;评测在同一 40 任务内、物体在演示起点框内重采样,缺 held-out 任务与更宽 OOD 放置;FAMO 是重实现(FAMO-ABC 仅 critic 加权、多两次前向),比较公平性依赖实现保真;单 run 8 卡两天的成本限制重复与社区复现。综上,内部一致性较好,结论的推广性与统计稳健性中等。 Significance:对 GPU 并行机器人 RL 社区有实际意义:Hebero 填补了异构多任务 GPU 基准、演示与标准化 MTRL 评测的组合空缺;DGPO 的共享栈降低演示引导方法的对比成本;IW-ABC 的轻量成功率 EMA 机制实现简单、可与 PPO 参考组合,易于被采纳;真机零更新迁移增加了应用说服力。但意义受范围限制:任务来自 LIBERO 抓放与关节操作、仅 PPO 骨干、单机器人;尚无向新任务、新形态或更长期限扩展的证据,算法优势能否泛化到其他基准未知。若社区沿用其评测协议并扩展到更大任务集,影响力会进一步放大。
⑥ 局限与保留意见
局限主要有以下几点。其一,时间索引动作目标在策略偏离演示轨迹时会失配,论文自己也承认这一点;尽管有稠密跟踪奖励与演示起点重置,长时间漂移仍可能使 ABC 目标变成错误监督,需要在线状态匹配、相位估计或最优传输对齐。其二,基准与训练范围被限制在 LIBERO 40 任务与 RoboTwin 四个任务,缺少更大任务集、跨形态、双臂或移动操作验证,无法回答方法随任务数扩展的泛化能力。其三,真机实验只用状态输入策略,依赖 FoundationPose、颜色分割与深度反投影等外部感知流水线,未展示视觉策略的 sim-to-real,弱化了视觉结果的实际意义。其四,评测仍在同一 40 任务内、物体在 50 条演示起点框定的范围内重采样,无 held-out 任务,OOD 位姿覆盖有限,测的是放置鲁棒性而非任务泛化。其五,方法依赖每任务 50 条演示与稠密跟踪奖励,演示质量、覆盖与数量敏感;稀疏奖励下的独立学习效果未充分展示。其六,框架只支持 PPO 类 on-policy learner,FAMO 是适应 DGPO 的重实现,算法比较的公平性与结论外推有限。其七,Long SR 方差较大且种子少,多卡、两日级的单 run 成本不利于大规模种子复现;吞吐对比与 CPU 基线未完全归一化。其八,视觉编码器冻结、只训练注意力池与 MLP,可能限制视觉表征的学习上限。其九,IW 依赖二值成功信号,训练早期成功稀疏时需要预热,成功谓词带噪时权重可能误导;τ 的 EMA 速率与 s_IW、w_max/w_min 等敏感超参的鲁棒性未系统考察。
⑦ 值得跟进
后续工作可从以下方向展开。第一,把时间索引目标替换为漂移鲁棒的演示对齐:在线状态匹配、相位估计或最优传输,减少策略偏离后的错误模仿监督。第二,扩大基准规模与多样性:超出 LIBERO 的更多任务、更长期限、跨形态与双臂设置,并把 DGPO 与 IW-ABC 迁移到更大任务集,检验随任务数扩展的稳定性。第三,强化评测协议:引入 held-out 任务、更宽随机化、组合泛化与持续学习协议,减少对演示起点框内重采样的依赖。第四,在共享栈内研究任务均衡方法的组合,例如 IW 与 FAMO 损失动态、PCGrad 梯度几何的叠加与消融,厘清不同信号在何种失败模式下互补。第五,降低演示依赖:少量演示、RLPD 式离线数据混合、逆 RL 与可学习稠密奖励,在 DGPO 接口下与 ABC/IW 组合。第六,视觉 sim-to-real:训练视觉 IW-ABC 并端到端部署,验证冻结 ViT 表征在真实场景的迁移,而非依赖外部感知。第七,形式化分析 τ 信号与奖励尺度的独立性,比较成功率 EMA、基于奖励、基于 critic 损失的进度代理,给出敏感超参的鲁棒性研究。第八,提升效率与可复现性:单卡规模化、异构视图的显存优化、标准化吞吐基准,公开种子与配置并建立 Mean SR/Long SR/覆盖/SR-AUC 的排行榜。