3D-DLP: Self-Supervised 3D Object-Centric Scene Representation Learning
About
We introduce 3D-DLP, a self-supervised object-centric representation learning model that decomposes scene-level RGB-D or voxel observations into a set of 3D latent particles. Building on the Deep Latent Particles (DLP) framework, each particle encodes disentangled attributes, including 3D keypoint position, bounding box dimensions, and appearance features, and represents a distinct entity in the scene. The model learns interpretable per-particle segmentation maps through an end-to-end self-supervised reconstruction objective. We demonstrate on both simulated and real-world datasets that the learned latent space is interpretable and controllable: by manipulating particle positions and decoding, we can generate novel scene configurations. Furthermore, we show that leveraging these compact 3D latent particles for downstream robotic manipulation improves performance over baselines that either lack explicit 3D information or rely on memory-intensive dense 3D inputs without object-centric structure. Code and videos are available at https://eubooks3003.github.io/3d-dlp.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Occupancy voxel reconstruction | GenericShapes | IoU32.2 | 6 | |
| Imitation Learning | MimicGen (test) | Stack Success94.6 | 4 | |
| Imitation Learning | RLBench (test) | Close Jar Success Rate16 | 4 | |
| Object-centric Scene Reconstruction | MimicGen RGB-D (test) | PSNR39.39 | 3 | |
| Occupancy voxel reconstruction | ShapeNetScenes | IoU26.2 | 3 | |
| RGB-D Reconstruction | 2DGenericShapes RGB-D | PSNR39.16 | 3 | |
| RGB-D Reconstruction | MimicGen RGB-D | PSNR39.39 | 3 | |
| Voxel Reconstruction | RGB-D-SD v2 | Masked PSNR21.57 | 3 | |
| Voxel Reconstruction | MimicGen | Masked PSNR24.41 | 3 | |
| RGB-D Reconstruction | BlenderShapes RGB-D | PSNR32.38 | 3 |