Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Ellina Zhang, Madhaven Iyengar, Amir Zadeh, Chuan Li, Deepak Pathak, David Held, Tal Daniel• 2026

Related benchmarks

TaskDatasetResultRank
Occupancy voxel reconstructionGenericShapes
IoU32.2
6
Imitation LearningMimicGen (test)
Stack Success94.6
4
Imitation LearningRLBench (test)
Close Jar Success Rate16
4
Object-centric Scene ReconstructionMimicGen RGB-D (test)
PSNR39.39
3
Occupancy voxel reconstructionShapeNetScenes
IoU26.2
3
RGB-D Reconstruction2DGenericShapes RGB-D
PSNR39.16
3
RGB-D ReconstructionMimicGen RGB-D
PSNR39.39
3
Voxel ReconstructionRGB-D-SD v2
Masked PSNR21.57
3
Voxel ReconstructionMimicGen
Masked PSNR24.41
3
RGB-D ReconstructionBlenderShapes RGB-D
PSNR32.38
3
Showing 10 of 11 rows

Other info

Follow for update