NeRF-Aug: Data Augmentation for Robotics with Neural Radiance Fields
This addresses the problem of policy generalization to unknown objects in robotics, representing a novel method for a known bottleneck.
The paper tackles the challenge of training robotic policies to generalize to unseen objects by introducing NeRF-Aug, a data augmentation method using neural radiance fields that runs 63% faster and achieves a 55.6% average performance boost over existing methods.
Training a policy that can generalize to unknown objects is a long standing challenge within the field of robotics. The performance of a policy often drops significantly in situations where an object in the scene was not seen during training. To solve this problem, we present NeRF-Aug, a novel method that is capable of teaching a policy to interact with objects that are not present in the dataset. This approach differs from existing approaches by leveraging the speed, photorealism, and 3D consistency of a neural radiance field for augmentation. NeRF-Aug both creates more photorealistic data and runs 63% faster than existing methods. We demonstrate the effectiveness of our method on 5 tasks with 9 novel objects that are not present in the expert demonstrations. We achieve an average performance boost of 55.6% when comparing our method to the next best method. You can see video results at https://nerf-aug.github.io.