atlas

Research map / 3D vision / 3D generation

Learning 3D Particle-based Simulators from RGB-D Videos

ICLR 2024 · 3D vision › 3D generation · approach: Learning physics from video · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Learns a 3D latent particle-based simulator end to end from posed multi-view RGB-D videos, using a hierarchical GNN dynamics model and an annular-kernel NeRF-style renderer, with no state or mask supervision.

Problem

Learning physical simulators directly from visual observations without privileged state information (object poses, geometry, masks), while retaining 3D structure, editability, and long-horizon prediction.

Approach

Jointly learn an encoder that unprojects RGB-D pixels into a latent 3D particle set, a hierarchical GNN with abstract nodes that predicts particle deltas without correspondences, and a volumetric renderer that decodes particles into images, all trained end to end on a multi-step pixel loss.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use a 3D particle representation and GNN dynamics instead of a global latent vector and MLP (ablation)VPD · MuJoCo Block, Deformable Block, Deformable Multi, Deformable Collision, Kubric MOVi-A/B/CPSNR: 29.601 -> 33.076; 26.969 -> 31.520; 26.636 -> 31.221; 22.394 -> 28.725; 27.604 -> 29.014; 26.201 -> 27.194; 20.068 -> 23Helped
Train the encoder/decoder end to end with dynamics instead of sequentially pretraining and freezing them (ablation)VPD · MuJoCo Block, Deformable Block, Deformable Multi, Deformable Collision, Kubric MOVi-A/B/CPSNR: 25.858 -> 33.076; 30.854 -> 31.520; 31.256 -> 31.221; 27.911 -> 28.725; 28.613 -> 29.014; 25.921 -> 27.194; 21.854 -> 23Mixed
Train with 16 trajectories instead of 256 (ablation)VPD · MuJoCo BlockPSNR: 31.22; 1.85 points lessHurt
Use one RGB-D camera instead of 9 (ablation)VPD · MuJoCo BlockPSNRNo clear effect
Add an additional feature encoder to process nearby particle features (ablation)VPD · MuJoCo BlockPSNRNo clear effect
Normalize feature contributions by relative distances to nearby particles (ablation)VPD · MuJoCo BlockPSNRNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: MuJoCo block Kubric MOVi-A Kubric MOVi-B Kubric MOVi-C Deformables (Block, Multi, Collision)

Benchmarks: MuJoCo Block Kubric MOVi-A Kubric MOVi-B Kubric MOVi-C Deformable Block Deformable Multi Deformable Collision

Metrics: PSNR SSIM

Baselines: SlotFormer NeRF-dy (Global ablation) Sequential ablation

Builds on: NeRF NeRF-dy PointNeRF SAVi SlotFormer Encode-Process-Decode

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy