Effective contact-rich manipulation requires robots to synergistically leverage vision, force, and proprioception. However, Reinforcement Learning agents struggle to learn in such multisensory settings, especially amidst sensory noise and dynamic changes. We propose MultiSensory Dynamic Pretraining (MSDP), a novel framework for learning expressive multisensory representations tailored for task-oriented policy learning. MSDP is based on masked autoencoding and trains a transformer-based encoder by reconstructing multisensory observations from only a subset of sensor embeddings, leading to cross-modal prediction and sensor fusion. For downstream policy learning, we introduce a novel asymmetric architecture, where a cross-attention mechanism allows the critic to extract dynamic, task-specific features from the frozen embeddings, while the actor receives a stable pooled representation to guide its actions. Our method demonstrates accelerated learning and robust performance under diverse perturbations, including sensor noise, and changes in object dynamics. Evaluations in multiple challenging, contact-rich robot manipulation tasks in simulation and the real world showcase the effectiveness of MSDP. Our approach exhibits strong robustness to perturbations and achieves high success rates on the real robot with as few as 6,000 online interactions, offering a simple yet powerful solution for complex multisensory robotic control. Website: https://msdp-pearl.github.io/
翻译:有效的富含接触操作要求机器人协同利用视觉、力觉与本体感觉。然而,强化学习智能体在多感官环境中难以学习,尤其是在存在感官噪声和动态变化的情况下。我们提出多感官动态预训练(MSDP),一种新颖的框架,用于学习面向任务的策略学习所需的表达性多感官表征。MSDP基于掩码自编码,通过仅从部分传感器嵌入重建多感官观测来训练基于Transformer的编码器,实现跨模态预测与传感器融合。在后续策略学习中,我们引入了一种新颖的非对称架构:交叉注意力机制使评论家能够从冻结的嵌入中提取动态、任务特定的特征,而演员则接收稳定的池化表征来指导其动作。我们的方法在多种扰动(包括传感器噪声和物体动态变化)下展示了加速学习与鲁棒性能。在模拟与真实世界中多项具有挑战性的、富含接触的机器人操作任务上的评估,验证了MSDP的有效性。该方法对扰动表现出强鲁棒性,并且在实际机器人上仅需6000次在线交互即可实现高成功率,为复杂多感官机器人控制提供了一种简单而强大的解决方案。网站:https://msdp-pearl.github.io/