Search papers, labs, and topics across Lattice.
This paper investigates the limitations of RGB-only policies in high-precision robotic manipulation and introduces a novel approach that maps point clouds into high-dimensional Fourier space to enhance spatial reasoning. By addressing the spectral bias of neural networks, the authors demonstrate that incorporating Fourier features allows for more effective utilization of geometric details, leading to improved performance on manipulation tasks. Experimental results across various benchmarks, including RoboCasa and ManiSkill3, reveal that this method significantly enhances the capabilities of point cloud-based imitation learning, regardless of the encoder architecture used.
Mapping point clouds into Fourier space unlocks high-frequency features, drastically improving robotic manipulation performance across diverse tasks.
High-precision robotic manipulation requires fine-grained spatial reasoning that is often difficult to achieve with RGB-only policies due to depth ambiguity and perspective scale issues. Policies that leverage 3D information directly, such as those based on point clouds, offer a stronger geometric prior over purely image-based ones, yet their performance remains highly task-dependent. We hypothesize that this discrepancy may be due to the spectral bias of neural networks towards learning low frequency functions, which especially affects architectures conditioned on slow-moving Cartesian features. We thus propose to map point clouds from Cartesian space into high-dimensional Fourier space, effectively equipping the point cloud encoder with direct access to high-frequency features. We experimentally validate the use of Fourier features on challenging manipulation tasks from the RoboCasa and ManiSkill3 benchmarks and on a real robot setup. Despite their simplicity, we find that Fourier features provide significant benefits across diverse encoder architectures and benchmarks and are robust across hyperparameters. Our results indicate that Fourier features let policies leverage geometric details more effectively than Cartesian features, showing their potential as a general-purpose tool for point cloud-based imitation learning. We provide source code and videos on our project page: https://fourier-il.github.io/fourier-il