PocketDP3: Efficient Pocket-Scale 3D Visuomotor Policy

Read full text
Authors: Jinhao Zhang Yunquan Zhang, Zijun Zhou, Huizhe Li, Yichen Lai, Wenlong Xia, Haoming Song, Yi Gong, Jie Me
Cited by: 0

Abstract

Recently, 3D vision-based diffusion policies have shown strong capability in learning complex robotic manipulation skills. However, a common architectural mismatch exists in these models: a tiny yet efficient point-cloud encoder is often paired with a massive decoder. Given a compact scene representation, we argue that this may lead to substantial parameter waste in the decoder. Motivated by this observation, we propose PocketDP3, a pocket-scale 3D diffusion policy that replaces the heavy conditional U-Net decoder used in prior methods with a lightweight Diffusion Mixer (DiM) built on MLP-Mixer blocks. This architecture enables efficient fusion across temporal and channel dimensions, significantly reducing model size. Notably, without any additional consistency distillation techniques, our method supports two-step inference without sacrificing performance, improving practicality for real-time deployment. Across three simulation benchmarks--RoboTwin2.0, Adroit, and MetaWorld--PocketDP3 achieves state-of-the-art performance with fewer than 1% of the parameters of prior methods, while also accelerating inference. Real-world experiments further demonstrate the practicality and transferability of our method in real-world settings. Code will be released.

Fields of Study & Subjects

Robot Manipulation and Learning Reinforcement Learning in Robotics 3D Shape Modeling and Analysis

Publication details

Journal / source
arXiv (Cornell University)
Publication date
29 January 2026
Document type
Preprint