arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04115cs.ROcs.AIcs.CV
收藏

DNAct: 扩散引导的多任务3D策略学习

DNAct: Diffusion Guided Multi-Task 3D Policy Learning

UC San Diego(加州大学圣迭戈分校)

Ge Yan, Yueh-Hua Wu, Xiaolong Wang

AI总结:

本文提出语言条件化多任务3D策略框架DNAct,结合神经渲染预训练与扩散训练,仅用少量演示即可提升策略泛化性,成功率超现有NeRF类方法30%以上。

摘要:

本文提出DNAct,这是一种语言条件化的多任务策略框架,它整合了神经渲染预训练与扩散训练,以在动作序列空间中实现多模态学习。为了仅用少量演示就能学习具备泛化性的多任务策略,DNAct的预训练阶段利用神经渲染,将来自Stable Diffusion等基础模型的2D语义特征蒸馏到3D空间,从而提供对场景的全面语义理解。这使得该框架能够应用于需要丰富3D语义与精准几何信息的高难度机器人任务。此外,我们提出了一种利用扩散训练的新方法,用于学习能够封装多任务演示中固有多模态特性的视觉与语言特征。通过扩散过程重建来自不同任务的动作序列,该模型能够区分不同模态,进而提升所学表征的鲁棒性与泛化性。DNAct的成功率提升超过30%,显著超越了基于NeRF的最先进多任务操作方法。项目网站:dnact.github.io。

英文摘要:

This paper presents DNAct, a language-conditioned multi-task policy framework that integrates neural rendering pre-training and diffusion training to enforce multi-modality learning in action sequence spaces. To learn a generalizable multi-task policy with few demonstrations, the pre-training phase of DNAct leverages neural rendering to distill 2D semantic features from foundation models such as Stable Diffusion to a 3D space, which provides a comprehensive semantic understanding regarding the scene. Consequently, it allows various applications to challenging robotic tasks requiring rich 3D semantics and accurate geometry. Furthermore, we introduce a novel approach utilizing diffusion training to learn a vision and language feature that encapsulates the inherent multi-modality in the multi-task demonstrations. By reconstructing the action sequences from different tasks via the diffusion process, the model is capable of distinguishing different modalities and thus improving the robustness and the generalizability of the learned representation. DNAct significantly surpasses SOTA NeRF-based multi-task manipulation approaches with over 30% improvement in success rate. Project website: dnact.github.io.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑