arXiv:2608.25572cs.ROcs.AI2026-08

通过置信度引导选择数据,提升机器人世界模型在新场景下的预测精度。

ConfAL-WM: Confidence-Guided Active Learning for Action-Conditioned World Models

论文配图:ConfAL-WM: Confidence-Guided Active Learning for Action-Conditioned World Models
图 1 · 摘自论文原文
  • 基于潜空间置信度图,实现局部区域精准采样。
  • 在RoboTwin2.0上,轨迹一致性提升17.3%,预测误差降低22.6%。
  • 适合需要高可靠性的机器人仿真与规划任务开发者。

动作条件世界模型是具身预测、规划和合成数据生成的重要基础,但其在新任务与场景分布下的错误常集中于局部时空区域,如机械臂、操作物体、接触区及遮挡物体。本文提出ConfAL-WM,一种面向后训练具身世界模型的置信度引导主动学习框架。基于EVAC,在UNet解码器特征上附加轻量级置信度探针,预测潜空间中的密集置信度图。这些图被聚合为任务级、帧级和块级得分,支持高效数据选择与局部训练增强。流程首先用少量目标域数据微调置信度探针并预热EVAC,再进行任务级预筛选以分配采样预算,最后对选定数据进行重训练,可选帧级或块级加权增强。在RoboTwin2.0上的实验表明,置信度引导选择显著提升后训练效率,而帧级与块级加权进一步优于标量奖励、进度与判断基线,预测质量与具身轨迹一致性分别提升17.3%与22.6%。

原文摘要 · Abstract (English)

Action-conditioned world models have become an important foundation for embodied prediction, planning, and synthetic data generation, but their errors under new task and scene distributions are often concentrated in localized spatiotemporal regions such as robot arms, manipulated objects, contact areas, and occluded objects. This paper presents ConfAL-WM, a confidence-guided active learning framework for post-training embodied world models. Built upon EVAC, we attach a lightweight confidence probe to UNet decoder features and predict dense confidence maps in the latent space. These maps are aggregated into task-, frame-, and patch-level scores, enabling both efficient data selection and localized training enhancement. Our pipeline first retrains the confidence probe and warms up EVAC with a small subset of target-domain data, then performs task-level prescreening to allocate sampling budgets, and finally applies selected-data retraining with optional frame or patch weighted data enhancement. Experiments on RoboTwin2.0 show that confidence-guided selection improves post-training efficiency, while dense frame and patch weighting further enhances prediction quality and embodied trajectory consistency compared with scalar reward, progress, and judge-based scoring baselines. A quick visual overview of this work is available at https://ConfAL-WM.github.io.

世界模型主动学习机器人置信度

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。