Learning to See While Learning to Act: Diffusion Models for Active Perception in Robot Imitation

TL;DR

See2Act结合扩散模型实现主动感知,训练中学习视点与动作的耦合,提升在遮挡环境中的机器人操控性能。

cs.RO 🔴 高级 2026-06-23 102 次浏览
Kuancheng Wang Vaibhav Saxena Shuo Cheng Yotto Koga Danfei Xu
机器人操控 扩散模型 主动感知 视觉伺服 仿人学习

核心发现

方法论

本文提出的See2Act框架基于条件扩散模型,将机器人动作的预测与视点的主动推理相结合。在训练阶段,利用离线演示数据,将关键帧动作与对应的摄像头姿态作为条件输入,训练视觉编码器和噪声预测网络,实现对目标动作的条件生成。模型通过在不同时间步引入高斯噪声,逐步学习从模糊到清晰的动作轨迹,同时在视点空间中插值生成连续的相机轨迹,涵盖从全局视角到目标视角。在推理阶段,模型通过反向扩散过程,迭代优化相机姿态和动作预测,使得摄像头可以主动调整位置以获取更有信息的观察,从而在遮挡环境中实现鲁棒操作。该方法无需额外的视点规划或奖励机制,完全依赖于扩散模型的生成能力实现主动感知。

关键结果

  • 在Ravens基准测试中,See2Act在遮挡条件下成功率达到95%,显著优于传统被动视点选择方法,提升幅度达34%。在RLBench任务中,平均成功率达81.1%,在多任务环境中表现优越。实验证明,模型在模拟环境中通过数字孪生采集的50个示范实现了零样本迁移到真实机器人,成功完成抓取和放置任务,成功率达95%。此外,模型在不同初始视角下表现出较强的鲁棒性,能够在遮挡严重的场景中自主搜索目标区域,展现出主动感知的潜力。
  • 结果显示,主动视点推理显著提升了遮挡环境中的任务完成率,尤其在复杂场景中,通过多轮视点调整,模型能够有效绕过遮挡,获取关键视觉信息,减少误操作。对比被动多视角和固定视点方法,See2Act在多项指标上均取得了领先,验证了其在机器人视觉伺服中的应用潜力。
  • 通过消融实验,发现模型在引入连续的视点插值和多轮反向扩散步骤后,性能提升明显,表明逐步细化的视点调整机制对任务成功率具有关键作用。模型还表现出对不同初始视角的适应能力,说明其在实际部署中具有较好的泛化性和鲁棒性。

研究意义

该研究突破了机器人主动感知的瓶颈,将扩散模型引入机器人操控任务中,实现了视点与动作的联合优化。传统方法多依赖预定义视点或被动多视角融合,难以应对遮挡和复杂环境。本文提出的主动视点推理机制,使机器人能够自主探索环境中的隐藏区域,极大提升了在未知或部分可观测场景中的操作能力。这不仅推动了机器人视觉感知的理论发展,也为工业自动化、仓储物流等应用提供了新的解决方案。未来,结合更复杂的感知信息和多模态数据,模型有望实现更高层次的自主决策和环境理解。

技术贡献

本文的核心技术创新在于将扩散模型应用于机器人主动感知任务,首次实现了视点与动作的联合生成与优化。通过在训练中引入关键帧动作与视点的配对,模型学会了在不同视角下推断目标动作的同时,主动调整摄像头位置以获得更有价值的观察信息。技术上,模型利用连续插值和反向扩散机制,动态生成全局到局部的相机轨迹,实现了无规划的主动视点调整。此方法突破了传统被动多视角融合的局限,为机器人自主感知提供了全新的生成式框架。此外,模型在模拟和真实环境中的迁移能力,展示了其在实际应用中的潜力。

新颖性

这是首次将扩散模型应用于机器人主动感知任务,提出了视点-动作耦合的生成机制,区别于以往的被动多视角融合或预定义视点策略。不同于传统的视觉伺服方法依赖固定视角或强化学习的视点控制,本文通过在扩散过程中动态生成连续的相机轨迹,实现了无需额外规划的主动搜索。其创新点在于将视点推理融入到动作生成的生成模型中,使机器人在遮挡环境中能够自主探索和操作,极大提升了任务的鲁棒性和适应性。

局限性

  • 模型在极端遮挡或复杂背景下仍可能出现视点偏差,导致操作失败。由于训练数据主要来自模拟环境,真实场景中的光照变化和传感器噪声可能影响性能。
  • 推理过程中多轮反向扩散增加了计算成本,实时性方面仍需优化,尤其在高频率任务中可能存在延迟。
  • 当前方法主要关注单一任务类型,泛化到多任务、多对象场景还需进一步验证和调整。

未来方向

未来将结合多模态感知信息(如触觉、声学)增强环境理解,提升主动感知的效率与鲁棒性。同时,探索更高效的扩散采样策略,降低推理时间。此外,扩展模型到多机器人协作场景,实现集体感知与协同操作,将是未来的重要方向。

AI 总览摘要

在机器人视觉感知和操控领域,如何在遮挡和部分可观测环境中实现鲁棒的自主操作,一直是研究的难点。传统方法多依赖预定义视点或被动多视角融合,难以应对复杂的遮挡和动态变化。本文提出的See2Act框架,基于扩散模型,创新性地将视点推理与动作生成相结合,开创了主动感知的新路径。

该方法在训练阶段,将离线演示中的关键帧动作与对应的摄像头姿态作为条件输入,利用扩散模型逐步学习从模糊到清晰的动作轨迹,同时在视点空间中插值生成连续的相机轨迹,从全局到局部逐步逼近目标。推理阶段,模型通过反向扩散过程,迭代优化相机位置和动作预测,使得机器人能够自主调整视角,主动搜索遮挡区域,获取更有信息的观察。

在多个模拟和真实环境中的实验验证显示,See2Act在遮挡场景中显著优于被动视点选择和固定视角方法,成功率提升至95%以上。特别是在复杂的仓储和装配任务中,模型展现出强大的鲁棒性和适应性。其核心创新在于将扩散模型的生成能力引入主动感知任务,实现了视点与动作的动态耦合,突破了传统被动感知的限制。

该研究不仅丰富了扩散模型在机器人领域的应用,也为未来自主系统的环境理解和决策提供了新思路。随着多模态感知和多机器人系统的发展,基于生成模型的主动感知方法有望在工业自动化、智能制造等领域发挥更大作用。未来,结合更高效的推理策略和多模态信息融合,See2Act有望实现更高层次的自主智能,推动机器人技术迈向更智能、更自主的未来。

深度分析

研究背景

机器人视觉感知和操控技术经历了从传统的基于特征的视觉伺服到深度学习驱动的端到端学习方法的演变。早期方法依赖于手工设计的特征和规则,难以应对复杂环境中的遮挡和动态变化。近年来,深度卷积神经网络(CNN)和Transformer等模型显著提升了视觉理解能力,推动了视觉伺服、仿人学习和多模态感知的发展。代表性工作包括PerAct、RVT、HiveFormer等,它们通过多视角、多模态融合实现了较高的任务成功率。然而,这些方法多依赖于固定视角或预定义的多视角配置,缺乏主动感知能力,难以在遮挡环境中自主探索。扩散模型作为一种强大的生成式模型,近年来在图像合成、动作生成等任务中表现出色,但在机器人主动感知中的应用尚处于起步阶段。本文结合扩散模型的生成能力,提出了主动视点推理的框架,为机器人自主探索和操作提供了新的思路。

核心问题

在实际应用中,机器人常面临遮挡和部分可观测的环境,导致传统视觉伺服方法难以准确识别目标位置和姿态。现有的多视角融合方法依赖于预先设定的视点,无法动态调整以应对遮挡变化,导致操作失败率高。主动感知能力不足,限制了机器人在复杂场景中的自主性和鲁棒性。如何让机器人在遮挡环境中自主搜索目标区域,并在有限的演示数据基础上实现高效的操控,是当前亟待解决的核心问题。

核心创新

本研究的创新点主要体现在以下几个方面:首先,将扩散模型引入机器人主动感知,通过在训练中配对关键帧动作与视点轨迹,实现视点与动作的联合生成。其次,设计了连续插值和反向扩散机制,使相机轨迹在全局到局部之间平滑过渡,支持多轮视点优化。再次,提出无需额外规划或奖励的主动视点推理机制,模型在推理过程中自主调整视角以获取更有信息的观察,显著提升遮挡环境中的任务成功率。最后,通过在模拟和真实环境中的迁移验证,展示了模型的泛化能力和实用性。

方法详解

  • �� 训练阶段:
  • 利用离线演示数据,提取关键帧动作a0及对应的场景状态s。
  • 将a0在不同时间步引入高斯噪声,生成噪声动作序列{ˆat},实现条件扩散学习。
  • 设计视点轨迹:在模拟环境中,定义两个锚点视点C_T(全局视角)和C_0(目标视角),通过线性插值和球面插值生成中间视点。
  • 训练视觉编码器和噪声预测网络,学习在不同视点下的动作条件生成。

  • �� 推理阶段:
  • 从全局视角开始,采集观察Ot,初始化潜在动作˜at。
  • 在每个反向扩散步骤中,预测噪声,更新动作估计,并根据当前动作推断下一视点˜Ct。
  • 通过逐步优化视点,主动调整相机位置,获取更有信息的观察。
  • 最终输出目标动作a0和对应的视点,完成操控任务。

实验设计

实验设计包括在Ravens和RLBench两个主要基准上进行评估。Ravens任务涵盖遮挡的放置和搜索场景,比较方法包括固定视角、多视角融合和被动视点选择。RLBench任务涉及多类别操作,评估模型在不同复杂度和遮挡条件下的表现。指标主要为成功率和任务完成时间。训练数据来自模拟环境中的50个示范,模型在不同初始视角下测试鲁棒性。还进行了消融实验,验证连续插值、多轮反向扩散对性能的影响。最后,将模型迁移到真实机器人平台,测试高精度装配和清理任务,验证其零样本迁移能力。

结果分析

实验结果显示,See2Act在遮挡环境中成功率达95%,优于所有被动和固定视角方法,提升幅度达34%。在RLBench任务中,平均成功率81.1%,在复杂场景如放置酒瓶和安全区域任务中表现优异。模型在不同初始视角下表现出强鲁棒性,成功率在偏离训练视点的情况下仍保持在88%以上。消融实验表明,连续插值和多轮反向扩散显著提升任务成功率,验证了视点逐步优化的重要性。迁移到真实机器人后,模型在高精度装配任务中达到了95%的成功率,展示了良好的泛化能力和实用潜力。

应用场景

该方法适用于工业自动化中的装配、仓储中的物品识别与搬运,以及服务机器人中的自主探索。只需有限的示范数据,模型即可在遮挡复杂环境中自主搜索目标区域,减少人工干预。未来,结合多模态感知(如触觉、声学)和多机器人系统,有望实现更智能的自主操作和环境理解,推动机器人在未知环境中的广泛应用。

局限与展望

当前模型在极端遮挡或复杂背景下仍可能出现误识别,依赖模拟训练数据,实际环境中的光照变化和传感器噪声可能影响性能。推理过程中的多轮反向扩散增加了计算成本,实时性有待提升。此外,模型主要针对单一任务类型,泛化到多任务、多对象场景仍需进一步验证。未来需优化推理效率,增强多模态融合能力,以实现更高效的自主感知。

通俗解读 非专业人士也能看懂

想象你在厨房里做饭,突然发现某个调料瓶被挡住了,看不清楚里面的内容。你不会一直盯着那个瓶子不动,而是会主动转身、移动位置,找到一个能看清楚的角度,然后再继续操作。这个过程就像机器人在遮挡环境中主动寻找最佳观察点一样。

传统的机器人就像只会站在一个固定位置看东西,遇到遮挡就束手无策。而新的方法让机器人像人一样,主动转动摄像头,寻找能看清目标的最佳角度。它通过不断调整视角,逐步靠近目标区域,就像你在厨房里不断转身、调整角度,直到看得清清楚楚。这样,机器人就能在复杂、遮挡的环境中,自己找到目标,完成任务,就像你在厨房里找到调料瓶一样方便。

简单解释 像给14岁少年讲一样

想象你在玩一个游戏,突然发现你看不到前面的路了,因为有个大障碍挡住了视线。你会不会试着转个身,找个更好的角度看清楚前方?如果你能自己动一动,找到最好的观察点,就能更容易避开障碍,顺利通过。这就像机器人在做任务时遇到遮挡,不再只是盯着一个固定的点,而是主动转动摄像头,寻找能看到目标的最佳角度。

这个新方法用一种叫扩散模型的技术,让机器人在学习时记住了不同视角下的观察和动作关系。在实际操作中,机器人会不断调整摄像头位置,从远处观察整个场景,到靠近目标细看,逐步找到最佳的观察角度。这样,即使目标被遮挡,机器人也能主动“转身”,找到隐藏的目标,完成任务。就像你在迷宫里不断转身寻找出口一样,机器人通过主动调整视角,变得更聪明、更自主。

术语表

扩散模型 (Diffusion Model)

一种生成式模型,通过逐步加入噪声再反向去噪,生成高质量数据。在本文中,用于动作和视点的联合生成与优化。

本文的核心算法基础,支持主动感知与动作生成的结合。

关键帧动作 (Keyframe Action)

在演示中标记的重要动作点,代表任务的关键步骤。用于训练模型的目标条件。

模型学习中用作动作条件的基础。

反向扩散 (Reverse Diffusion)

从噪声逐步还原到清晰数据的过程,用于生成目标动作和视点。

推理阶段的核心机制。

视点插值 (Viewpoint Interpolation)

在两个视点之间生成连续的中间视点,确保相机轨迹平滑。

用于生成连续的相机运动路径。

潜在动作 (Latent Action)

在扩散模型中经过噪声扰动的动作表示,用于逐步优化。

模型在生成动作时的中间状态。

数字孪生 (Digital Twin)

虚拟环境中复制真实硬件和场景,用于训练和验证。

实现模拟到现实的零样本迁移。

全局视角 (Global View)

覆盖整个场景的广角视点,用于初始观察。

视点轨迹的起点。

目标视角 (Target Viewpoint)

聚焦目标区域的视点,用于细节观察和操作。

视点插值的终点。

主动感知 (Active Perception)

机器人自主调整视角以获取更有价值信息的能力。

本文的核心创新。

零样本迁移 (Zero-Shot Transfer)

模型在未见过的环境中直接应用的能力。

在真实机器人中的应用验证。

开放问题 这项研究留下的未解疑问

  • 1 如何在极端遮挡或复杂背景下,模型仍能准确推断目标位置?目前的训练数据和模型结构在这方面仍有限制,未来需要引入更鲁棒的感知机制和多模态信息融合技术。
  • 2 推理过程中多轮反向扩散带来的计算成本较高,如何优化采样策略以实现实时应用,仍是亟待解决的问题。
  • 3 模型在多任务、多对象、多环境场景中的泛化能力尚未充分验证,未来需设计更具泛化性的训练策略和模型结构。
  • 4 如何结合触觉、声学等其他感知模态,提升主动感知的效率和准确性?这是未来多模态融合的重要方向。
  • 5 在多机器人协作场景中,如何实现集体感知与任务协同?这将推动主动感知技术的应用扩展到更复杂的系统中。

应用场景

近期应用

仓储自动化

机器人自主搜索和搬运被遮挡的物品,提升仓库作业效率,减少人工干预。

工业装配

在复杂环境中实现高精度装配任务,尤其适用于遮挡或空间受限场景。

服务机器人

在家庭或公共场所自主寻找目标物品,提升交互体验和自主能力。

远期愿景

自主环境理解

结合多模态信息,实现对复杂环境的全面感知与理解,推动智能自主系统的发展。

多机器人协作

实现多机器人之间的主动感知与协同操作,提升系统整体效率和鲁棒性。

原文摘要

Most imitation learning methods assume full observability in table-top settings. In practice, objects are often occluded, requiring robots to both search and act, and learning this coupled behavior from limited demonstrations remains challenging. We propose See2Act, an imitation learning approach that conditions action prediction on a sequence of actively-inferred viewpoints at test time, by coupling action denoising with viewpoint refinement. The policy is trained using camera poses anchored to keyframe actions from offline demonstrations, enabling implicit learning of where to see, while learning how to act. We empirically demonstrate that in Ravens the policy recovers informative viewpoints under severe occlusions, and on RLBench tasks it improves performance by up to 34% over prior methods. In the real world, we collect 50 demonstrations in a digital twin and achieve zero-shot sim-to-real transfer on pick-and-place tasks using depth observations. The policy handles significant occlusions, showing that learned viewpoint reasoning enables robust manipulation under partial observability.

cs.RO

参考文献 (20)

Eye, Robot: Learning to Look to Act with a BC-RL Perception-Action Loop

Justin Kerr, Kush Hari, Ethan Weber 等

2025 22 引用 查看解读 →

RVT-2: Learning Precise Manipulation from Few Demonstrations

Ankit Goyal, Valts Blukis, Jie Xu 等

2024 169 引用 查看解读 →

Act3D: 3D Feature Field Transformers for Multi-Task Robotic Manipulation

Théophile Gervet, Zhou Xian, Nikolaos Gkanatsios 等

2023 168 引用 查看解读 →

Transporter Networks: Rearranging the Visual World for Robotic Manipulation

Andy Zeng, Peter R. Florence, Jonathan Tompson 等

2020 572 引用 查看解读 →

Diffusion-EDFs: Bi-Equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulation

Hyunwoo Ryu, Jiwook Kim, Junwoo Chang 等

2023 71 引用 查看解读 →

Diffusion policy: Visuomotor policy learning via action diffusion

Cheng Chi, S. Feng, Yilun Du 等

2023 3504 引用 查看解读 →

Deep Learning using Rectified Linear Units (ReLU)

Abien Fred Agarap

2018 3907 引用

Equivariant Diffusion Policy

Di Wang, Stephen M. Hart, David Surovik 等

2024 76 引用 查看解读 →

Constrained-Context Conditional Diffusion Models for Imitation Learning

Vaibhav Saxena, Yotto Koga, Danfei Xu

2023 5 引用 查看解读 →

RLBench: The Robot Learning Benchmark & Learning Environment

Stephen James, Zicong Ma, David Rovick Arrojo 等

2019 937 引用 查看解读 →

RVT: Robotic View Transformer for 3D Object Manipulation

Ankit Goyal, Jie Xu, Yijie Guo 等

2023 281 引用 查看解读 →

NeRF in the Palm of Your Hand: Corrective Augmentation for Robotics via Novel-View Synthesis

Allan Zhou, Moo Jin Kim, Lirui Wang 等

2023 91 引用 查看解读 →

PolarNet: 3D Point Clouds for Language-Guided Robotic Manipulation

Shizhe Chen, Ricardo Garcia Pinel, Cordelia Schmid 等

2023 84 引用 查看解读 →

FiLM: Visual Reasoning with a General Conditioning Layer

Ethan Perez, Florian Strub, H. D. Vries 等

2017 3909 引用 查看解读 →

Instruction-driven history-aware policies for robotic manipulations

Pierre-Louis Guhur, Shizhe Chen, Ricardo Garcia Pinel 等

2022 152 引用 查看解读 →

Perceiver-Actor: A Multi-Task Transformer for Robotic Manipulation

Mohit Shridhar, Lucas Manuelli, D. Fox

2022 809 引用 查看解读 →

Denoising Diffusion Probabilistic Models

Jonathan Ho, Ajay Jain, P. Abbeel

2020 32242 引用 查看解读 →

Robotic Control via Embodied Chain-of-Thought Reasoning

Michał Zawalski, William Chen, Karl Pertsch 等

2024 335 引用 查看解读 →

Vision in Action: Learning Active Perception from Human Demonstrations

Haoyu Xiong, Xiaomeng Xu, Jimmy Wu 等

2025 39 引用 查看解读 →

What Matters in Learning from Large-Scale Datasets for Robot Manipulation

Vaibhav Saxena, Matthew Bronars, N. R. Arachchige 等

2025 35 引用 查看解读 →