DL0166 3D Gaussian and NeRF Neural Simulator

How do 3D Gaussians and Neural Radiance Fields serve as dense physical representations inside differentiable neural simulators?

Answer

A differentiable neural simulator needs one scene description that works twice: as geometry the solver can push around, and as an image the photometric loss can compare against real video. NeRF supplies the second half well, storing the scene as an implicit field that maps a position and view direction to density and radiance, rendered by differentiable volume quadrature along rays. What it lacks is handles, because an MLP field has no particle to give mass, no covariance to advect, and no obvious place to attach a constitutive model. 3D Gaussian Splatting removes that obstacle by storing the scene as an explicit set of anisotropic Gaussians with mean, covariance, opacity, and spherical-harmonic color, so every primitive doubles as a material point in a Lagrangian solver. The simulator advances those points with differentiable physics, usually MLS-MPM, the rasterizer splats them back to pixels, and gradients from the rendering loss flow through both the rollout and the renderer into material parameters such as Young’s modulus, Poisson ratio, yield stress, and friction. The loop is therefore video → physical parameters → rollout of unseen dynamics, with pixels as the only supervision.

(1) Dense Means Physics Everywhere: unlike a mesh, both representations describe the interior volume, so mass, velocity, and stress are defined at every sampled point rather than only on a surface shell.
(2) Gaussians Are Already Particles: a splat’s mean is a particle position and its covariance is a local volume element, which is exactly the state an MPM particle carries, so the reconstruction and the simulation share one data structure.
(3) NeRF Needs A Bridge: a continuous field has to be voxelized or resampled onto particles each step, the Eulerian-Lagrangian conversion that PAC-NeRF introduced, which costs an extra transfer and an extra source of gradient noise.
(4) Covariance Follows The Deformation Gradient: kinematics are preserved by advecting the mean through the flow map and conjugating the covariance with F, otherwise stretched material still renders as undeformed blobs.
(5) The Loss Is Purely Photometric: no ground-truth stiffness is available, so system identification is driven by rendered-versus-observed pixel error over a multi-frame rollout.
(6) The Cost Lives In The Backward Pass: memory grows as O(TN) in substeps and particles, and gradients through contact and friction are non-smooth, so checkpointing and softened contact models are mandatory.

Left-to-right pipeline: multi-view video with camera poses feeds a dense reconstruction stage producing 3D Gaussians or a NeRF field, which becomes a physical particle state with mass, velocity and deformation gradient, advanced by a differentiable MPM simulator, rendered by a differentiable renderer and compared to the observed frames, with a dashed backward gradient path into a material parameter block that feeds the simulator

Figure 1: The identification loop. The dense representation appears twice in the graph, once as the initial physical state and once as the thing being rendered, which is why gradients from a single photometric loss can reach material parameters that were never observed directly.

The reason 3D Gaussians displaced radiance fields in this role is mechanical rather than aesthetic. In a Lagrangian solver the unknowns are particle positions and their local deformation, and a Gaussian already stores a mean plus a covariance factored as scale and rotation, so PhysGaussian can drive splats directly with a continuum solver and keep rendering in real time. A radiance field, by contrast, is a function of space with no identity attached to any location, so tracking material requires either a learned deformation field mapping observation time back to a canonical frame, or a particle proxy that carries the physics while the field carries appearance. Two consequences matter in practice. First, the spherical-harmonic coefficients are expressed in world coordinates, so when a splat rotates by R its appearance basis must be rotated too, otherwise highlights stay welded to the world frame while the geometry turns. Second, lighting is baked, so shadows and specular reflections do not respond to motion, which biases the photometric gradient exactly on the frames where an object moves most.

Two panels of ellipse grids: the left panel shows a three by three grid of identical circular Gaussians in the rest state, the right panel shows the same grid after a shear and vertical compression, where each ellipse is stretched and tilted consistently with the displaced particle centers, connected by an arrow labelled F

Figure 2: Kinematics of a splat. The mean rides the flow map while the covariance is conjugated by the deformation gradient, so shear and compression change both where a Gaussian sits and how it is shaped; skipping the covariance update leaves visibly isotropic blobs inside sheared material and corrupts the rendering gradient.

Mathematical Formulation:
\alpha_i = 1 - \exp(-\sigma_i \delta_i)
T_i = \prod_{j=1}^{i-1}(1 - \alpha_j)
C(r) = \sum_{i=1}^{N} T_i \alpha_i c_i
\Sigma = R S S^\top R^\top
\mu_p^{t} = \phi_t(\mu_p^{0})
\Sigma_p^{t} = F_p^{t}\,\Sigma_p^{0}\,(F_p^{t})^\top
s_{t+1} = \mathcal{S}(s_t;\theta)
\mathcal{L} = \sum_t \lVert \mathcal{R}(s_t) - I_t \rVert^2
\nabla_{\theta}\mathcal{L} = \sum_t \frac{\partial \mathcal{L}}{\partial s_t}\frac{\partial s_t}{\partial \theta}

Where:

  • C(r) is the rendered color of ray r, accumulated over N ordered samples or splats with per-sample color c_i; this same alpha-compositing form covers NeRF quadrature and Gaussian splatting.
  • \sigma_i is density, \delta_i the sample spacing along the ray, \alpha_i the resulting opacity, and T_i the transmittance reaching sample i.
  • \Sigma is a splat’s covariance, parameterized by rotation R and diagonal scale S so that it stays positive semi-definite under gradient updates.
  • p indexes particles, \phi_t is the flow map from rest to time t, and F_p^{t} = \partial \phi_t / \partial \mu is the deformation gradient whose determinant gives local volume change.
  • s_t is the full simulator state (positions, velocities, F, affine momentum) and \mathcal{S} one differentiable substep.
  • \theta collects the unknown physical parameters such as Young’s modulus E, Poisson ratio \nu, density \rho, yield stress, and friction coefficient.
  • \mathcal{R} is the differentiable renderer and I_t the observed frame, so \mathcal{L} is a multi-frame photometric residual and its gradient is a product of per-substep Jacobians.

That last product is where the engineering happens. Explicit MPM needs a CFL-limited timestep, so one video frame at 30 fps typically hides a few hundred substeps, and reverse-mode differentiation over a one-second clip means backpropagating through several thousand Jacobians. Storing every intermediate state is what kills the run: 200k particles at roughly 96 bytes of state per particle is about 19 MB per substep, so a 1,000-substep rollout needs 19 GB of tape. The standard remedies are gradient checkpointing at an interval near \sqrt{T}, truncating the loss window to a handful of frames, and recomputing forward segments during the backward pass. Numerically, the same product of Jacobians makes gradients explode for stiff materials and vanish once the trajectory has been dissipated by friction, which is why practitioners fit a log-parameterized stiffness and soften contact before trusting any gradient at all.

Log-log chart of backward-pass memory in bytes versus number of simulator substeps for 200 thousand particles, comparing storing the full trajectory which grows linearly and reaches 40 gigabytes near 2000 substeps, against square-root gradient checkpointing which stays under 3 gigabytes, with a dash-dotted horizontal line marking a 40 gigabyte HBM budget

Figure 3: Why differentiable rollouts are short. At 19 MB of particle state per substep, the naive tape exhausts a 40 GB HBM budget after roughly 2,000 substeps, about one second of simulated time, while square-root checkpointing keeps memory in the low gigabytes at the price of one extra forward pass.

PropertyNeRF (implicit field)3D Gaussians (explicit primitives)
PrimitiveMLP or hash grid queried at continuous positions; no persistent identitySet of anisotropic Gaussians, each with mean, covariance, opacity, SH color
Coupling to solverNeeds a deformation field or an Eulerian-Lagrangian resampling step onto particlesSplat is used directly as an MPM material point, no conversion layer
Render cost per frameHundreds of network queries per ray; typically tens of ms to secondsTile-based rasterization, real time at 1080p on one GPU
Handles topology changeNaturally, since density is just a field value that can appear or vanishPoorly; fracture needs splitting or spawning new Gaussians outside the solver
Dominant failure modeSlow rollouts, floaters in unobserved regions, noisy gradients through resamplingBaked lighting and unrotated SH, plus spiky splats that behave like bad particles

Login to view more content


Log in to track your progress

Comments

Leave a Reply

Your email address will not be published. Required fields are marked *