English

PointMapPolicy: Structured Point Cloud Processing for Multi-Modal Imitation Learning

Robotics 2026-01-27 v3 Machine Learning

Abstract

Robotic manipulation systems benefit from complementary sensing modalities, where each provides unique environmental information. Point clouds capture detailed geometric structure, while RGB images provide rich semantic context. Current point cloud methods struggle to capture fine-grained detail, especially for complex tasks, which RGB methods lack geometric awareness, which hinders their precision and generalization. We introduce PointMapPolicy, a novel approach that conditions diffusion policies on structured grids of points without downsampling. The resulting data type makes it easier to extract shape and spatial relationships from observations, and can be transformed between reference frames. Yet due to their structure in a regular grid, we enable the use of established computer vision techniques directly to 3D data. Using xLSTM as a backbone, our model efficiently fuses the point maps with RGB data for enhanced multi-modal perception. Through extensive experiments on the RoboCasa and CALVIN benchmarks and real robot evaluations, we demonstrate that our method achieves state-of-the-art performance across diverse manipulation tasks. The overview and demos are available on our project page: https://point-map.github.io/Point-Map/

Keywords

Cite

@article{arxiv.2510.20406,
  title  = {PointMapPolicy: Structured Point Cloud Processing for Multi-Modal Imitation Learning},
  author = {Xiaogang Jia and Qian Wang and Anrui Wang and Han A. Wang and Balázs Gyenes and Emiliyan Gospodinov and Xinkai Jiang and Ge Li and Hongyi Zhou and Weiran Liao and Xi Huang and Maximilian Beck and Moritz Reuss and Rudolf Lioutikov and Gerhard Neumann},
  journal= {arXiv preprint arXiv:2510.20406},
  year   = {2026}
}