M ESH D IFFUSION : S CORE - BASED G ENERATIVE 3D M ESH M ODELING
Project Page: meshdiffusion.github.io
[Figure]
A BSTRACT
We consider the task of generating realistic 3D shapes, which is useful for a variety of applications such as automatic scene generation and physical simulation.
Compared to other 3D representations like voxels and point clouds, meshes are more desirable in practice, because (1) they enable easy and arbitrary manipulation of shapes for relighting and simulation, and (2) they can fully leverage the power of modern graphics pipelines which are mostly optimized for meshes.
Previous scalable methods for generating meshes typically rely on sub-optimal post-processing, and they tend to produce overly-smooth or noisy surfaces without fine-grained geometric details.
To overcome these shortcomings, we take advantage of the graph structure of meshes and use a simple yet very effective generative modeling method to generate 3D meshes.
Specifically, we represent meshes with deformable tetrahedral grids, and then train a diffusion model on this direct parameterization.
We demonstrate the effectiveness of our model on multiple generative tasks.
I NTRODUCTION
As one of the most challenging tasks in computer vision and graphics, generative modeling of high-quality 3D shapes is of great significance in many applications such as virtual reality and metaverse [11].
Traditional methods for generative 3D shape modeling are usually built upon representations of voxels [51] or point clouds [1], mostly because ground truth data of these representations are relatively easy to obtain and also convenient to process.
Both representations, however, do not produce fine-level surface geometry and therefore cannot be used for photorealistic rendering of shapes of different materials in different lighting conditions.
And despite being convenient to process for computers, both voxels and point clouds are relatively hard for artists to edit, especially when the generated 3D shapes are complex and of low quality.
Moreover, modern graphics pipelines are built and optimized for explicit geometry representations like meshes, making meshes one of the most desirable final 3D shape representations.
While it is still possible to use methods like Poisson reconstruction to obtain surfaces from voxels and points clouds, the resulted surfaces are generally noisy and contain many topological artifacts, even with carefully tuned hyperparameters.
To improve the representation flexibility, sign distance fields (SDFs) have been adopted to model shape surfaces, which enables us to use marching cubes [29] to extract the zero-surfaces and thus 3D meshes.
However, SDFs are typically harder to learn as it requires a carefully designed sampling strategy and regularization.
Because SDFs are usually parameterized with multi-layer perceptrons (MLPs) in which a smoothness prior is implicitly embedded, the generated shapes tend to be so smooth that sharp edges and important (and potentially semantic) details are lost.
Moreover, SDFs are costly to render and therefore less suitable for downstream tasks like conditional generation with
RGB images, which require an efficient differentiable renderer during inference.
We instead aim to generate 3D shapes by directly producing 3D meshes, where surfaces are represented as a graph of triangular or polygon faces.
With 3D meshes, all local surface information is completely included in the mesh vertices (along with the vertex connectivity), because the surface normal of any point on the shape surface is simply a nearest neighbor or some local linear combination of vertex normals.
Such a regular structure with rich geometric details enables us to better model the data distribution and learn generative models that are more geometry-aware.
In light of recent advances in score-based generative modeling [16, 47] where powerful generative performance and effortless training are demonstrated, we propose to train diffusion models on these vertices to generate meshes.
However, it is by no means a trivial task and poses two critical problems: (1) the numbers of vertices and faces are indefinite for general object categories, and (2) the underlying topology varies wildly and edges have to be generated at the same time.
A natural solution is to enclose meshes with another structure such that the space of mesh topology and spatial configuration is constrained.
One common approach is to discretize the 3D space and encapsulate each mesh in a tiny cell, and it is proven useful in simulation [37] and human surface modeling [35].
Observing that this sort of mesh modeling is viable in recent differentiable geometry modeling literature [32, 38, 43], we propose to train diffusion models on a discretized and uniform tetrahedral grid structure which parameterizes a small yet representative family of meshes.
With such a grid representation, topological change is subsumed into the SDF values and the inputs to the diffusion model now assume a fixed and identical size.
More importantly, since SDF values are now independent scalars instead of the outputs of an MLP, the parameterized shapes are no longer biased towards smooth surfaces.
Indeed, by such an explicit probabilistic modeling we introduce an explicit geometric prior into shape generation, because a score matching loss of diffusion models on the grid vertices has direct and simple correspondence to the vertex positions of triangular mesh.
We demonstrate that our method, dubbed MeshDiffusion , is able to produce high-quality meshes and enables conditional generation with a differentiable renderer.
MeshDiffusion is also very stable to train without bells and whistles.
We validate the superiority of the visual quality of our generated samples qualitatively with different rendered views and quantitatively by proxy metrics.
We further conduct ablation studies to show that our design choices are necessary and well suited for the task of 3D mesh generation.
Our contributions are summarized below:
• To our knowledge, we are the first to apply diffusion model for unconditionally generating 3D high-quality meshes and to show that diffusion models are well suited for 3D geometry.
• Taking advantage of the deformable tetrahedral grid parameterization of 3D mesh shapes, we propose a simple and effortless way to train a diffusion model to generate 3D meshes.
• We qualitatively and quantitatively demonstrate the superiority of MeshDiffusion on different tasks, including (1) unconditional generation, (2) conditional generation and (3) interpolation.
R ELATED W ORK
3D Shape Generation .
3D shape generation is commonly done by using generative models on voxels [51] and point clouds [53], due to their simplicity and accessibility.
These resulted (often noisy) voxels or point clouds, however, do not explicitly encode surface information and therefore has to be processed with surface reconstruction methods in order to be used in applications like relighting and simulation.
Recent advances in implicit representations [31] lead to a series of generative models on neural fields [5, 6, 23, 42].
Based on these representations, one can learn 3D shapes directly from 2D images [5, 6, 42] with differentiable rendering in the generator.
Most of these implicit representations, despite representing shapes in a more photorealistic way, require additional post-processing steps to extract explicit meshes, and are often more time-consuming to render images.
If the underlying topology is known, one may directly generate the vertices of a mesh [30, 48].
[33] extends this approach to the topology-varying case with autoregressive models, sequentially generating vertices and edges, but it is hardly scalable and yields unsatisfactory results on complex geometry.
A batch of concurrent work propose similar solutions to mesh generation, including: GET3D [13] which uses StyleGAN [19] with a differentiable renderer on tetrahedral grid representations and learns to generate 3D meshes from 2D RGB images, TetGAN [14] which trains generative adversarial networks (GANs) on tetrahedral grids and LION [54] which uses a trained Shape-As-Points [36] network to build meshes from diffusion-model-generated latent point clouds.
Mesh Reconstruction .
Reconstructing 3D meshes of generic objects is challenging and often illposed due to its highly complex structures [18, 20, 28, 50, 52].
One often resorts to some non-mesh intermediate representations that are easy to process, and then transforms them back to meshes with mesh reconstruction methods.
One of the most popular methods is marching cubes [29] which assumes that a surface is represented by the zero level set of some continuous field, and this continuous field can be well approximated by linear interpolation of discrete grid points.
It is possible to avoid implicit representations and directly construct meshes.
For instance, by assuming the points lie on a surface, one can build triangular meshes by connecting these points, a process known as Delaunay triangulation [3, 10, 15, 22].
It is relatively rare because of its strong assumption on data.
With known mesh topology, one can deform a mesh template to fit a given representation.
This approach can be easily used to fit multiview images of shapes with known topology ( e.g. , human faces) using a differentiable renderer [25].
It is recently shown that we are able to parameterize meshes of varying topology [38, 43] and optimize them using differentiable renderers [32], which our work leverages.
Score-based Generative Models .
Recent years have witnessed a surge in modeling data distributions with score-based models [16, 46, 47], which parameterizes the logarithm of the gradient of the probability, known as the score function, rather than the probability directly.
Different from energybased models, the often intractable normalization constant can be avoided and therefore training can be performed by simply matching the score function.
It has been shown that, by using multi-scale models [16], U-Net architectures [41] and denoising score matching [49], score-based models can perform high-fidelity image generation and inpainting [16, 40, 47].
P RELIMINARIES
Deep marching tetrahedra (DMTet) [43] is a method to parameterize and optimize meshes of arbitrary topology in a differentiable way.
The 3D space is discretized with a deformable tetrahedral grid, in which each vertex possesses a SDF value.
The SDF of each 3D position in the space is computed by marching tetrahedra [12], which assumes SDF values to be barycentric interpolation of the SDF values of the vertices of the enclosing tetrahedra.
More specifically, for a point x q inside a tetrahedron of vertices x 1 , x 2 , x 3 , x 4 with SDF values being s 1 , s 2 , s 3 , s 4 , we obtain its unique barycentric coordinate ( a 1 , a 2 , a 3 , a 4 ) such that x q = (cid:80) i
a
i x i ( a i ∈ [0 , 1] ).
The SDF value s q of x q is then computed as s q = (cid:80) i
a i s i .
As a result, if there exists a triangular mesh in the tetrahedron ( i.e. , s 1 , s 2 , s 3 , s 4 are not of the same sign), then we can know that the triangular mesh is exactly the zero surface in the tetrahedron.
The triangular mesh vertex v p on a tetrahedron edge ( v a , v b ) is therefore computed by v p =( v a s b − v b s a ) /
( s b − s a ) .
The advantage of using DMTet, compared to variants of deep marching cubes [26, 38] which rely on the marching cubes algorithm [29], is that the grids are deformable and thus more capable of capturing some finer geometric details like very thin handles of chairs.
While the cubic grid used for deep marching cubes can be deformable as well, the deformed cubes are much worse objects to deal with, compared to tetrahedra which will remain tetrahedra after deformation.
Notably, DMTet can be fitted with a differentiable renderer by jointly optimizing for the geometry, the texture and the lighting of a 3D shape given multi-view RGB images within a reasonable amount of time and memory budget [32].
Diffusion models are a set of score-based generative models which learn and infer with the groundtruth time-dependent score function defined by a forward diffusion process [16].
Under the stochastic differential equation (SDE) framework proposed in [47], diffusion models construct the probability of each point in a space by diffusing all data points x 0 ∈ D to a target distribution at time T (normally
[Figure]
p T ( x T ) =
N ( x T ; 0 , I ) ) in a time-variant manner: log p t ( x t |D )
=
E x 0 ∈D [log p t ( x t | x 0 )] , where p t ( x t | x 0 ) is modeled by the following Itô process, with w ( t ) being a standard Wiener process: dx = F ( x, t )
dt + G ( t ) dw , in which F ( x, t ) and G ( x, t ) are some predefined functions.
The training objective is to fit the score function ∇ x log p t ( x t | x 0 ) with a neural net s θ ( x t , t ) , with λ ( t ) being a scalar weight subject to model design:
[Equation]
With the trained score function predictor, samples can be generated by solving the reverse SDE [47] with the initial distribution p T ( x T ) using solvers like Euler-Maruyama method [21]:
[Equation]
A popular diffusion model DDPM [16], which utilizes a discrete diffusion process p ( x t | x t − 1 )
=
N ( x t ; (1 − β t ) x t − 1 , β t I ) , can be formulated within this SDE framework as [47]: dx = − β t xdt + √ β t dw .
The parameters β t are chosen such that p T ( x T ) is close to a standard Gaussian distribution.
MeshDiffusion : D IFFUSION M ODEL ON M ESHES
With DMTet, 3D meshes can be approximated and parameterized with tetrahedral grids.
Moreover, by initializing the tetrahedral grid using body-centered cubic (BCC) tiling [24] and treating the deformation of each vertex only as an attribute, we obtain a uniform and almost-regular tetrahedral grid (with fewer vertex degrees on the boundary) in which 3D translational symmetry is preserved.
Such nice properties enable us to train a diffusion model for generating 3D meshes.
From now on, we always assume that the undeformed tetrahedral grid is uniformly initialized and predict the deformation of tetrahedral grid vertices from their initial positions.
As a result, our model takes as input a uniform tetrahedral grid with 4-dimensional attributes (3 dimensions for deformation andfor SDF).
4. 3D CNN FOR D EFORMABLE T ETRAHEDRAL G RIDS
While it is natural to use graph neural nets (GNNs) for diffusion model on tetrahedral grids, we argue that it is better to use CNNs which generally have more model capacity than GNNs due to the embedded spatial information in the convolution weights.
Specifically, the set of vertex positions of a uniform tetrahedral grid, created in the way described in [24], is a subset of the vertex position set of a uniform cubic lattice.
A standard voxel-like representation can therefore be easily created by infilling the “missing” sites of a tetrahedral grid (illustrated in Fig. 2).
We follow typical practices in diffusion models [47] and use a 3D U-Net for the score function predictor.
Details are in the appendix.
4. T RAINING O BJECTIVE
Since the tetrahedral grid representation of each object in the datasets is not given, our objective involves the following constrained optimization:
[Equation]
in which D is the 3D shape dataset, y 0 ∼ D is the set of 2D views of an object sampled from D , t is uniformly sampled from { 1 , ..., T } , s θ ( x, t ) is the score function approximator network, g ϕ ( y 0 ) is the mapping from 2D views of y 0 to its tetrahedral grid representation and L Render is the rendering (and relevant regularization) loss used in [32].
[Table]
[Table] [Table]
[Table] [Table]
While in theory it is possible to train g ϕ such that s θ can be better learned, we instead take the simplest two-stage and non-amortized approach as detailed optimization and encoder architecture design to use is beyond our scope.
Specifically, we first solve for the constraint L Render and create a dataset D x of tetrahedral grids by fitting a tetrahedral grid x 0 for each y 0 ∼ D in a non-amortized way.
The second stage is simply optimize for the main objective, a L 2 denoising loss of diffusion model, on D x .
In principle, the first data fitting stage can be trained with multiview RGB images by following the procedure in [32].
However, we notice that with RGB images only, it fails to learn some complex geometries, especially when surface materials are highly specular.
To demonstrate our idea while not delving too deep into the rendering details, we instead assume that we have access to 3D mesh datasets (in our experiments, ShapeNet datasets [7]) so that RGBD images can be rendered with random lighting and simple surface materials.
More specifically, we use a single default material with diffuse components only for all ground truth meshes, and render multiview RGBD images with some known but randomly rotated environment light (represented as a cubemap [44]).
With the additional depth information, our reconstruction objective is L Render =
α image L image + α depth L depth + α chamfer L chamfer + α penalty L penalty , in which L image and L depth are the image loss (RGB and silhouette) and the depth loss between the rendered views and the ground truth views, respectively; L chamfer is the Chamfer distance [1] between a sampled point cloud from the predicted mesh and one from the ground truth mesh; L penalty is the regularization terms to move reconstructed meshes out from bad local minima.
Details of these losses are included in the appendix.
To remove absolute scales from SDFs (Sec. 4.3), we solve the optimization objective twice, where during the second pass we fix the SDFs to be the signs ( i.e. , ± 1 ) of the resulted SDF values from the first pass.
4. R EDUCING N OISE -S ENSITIVITY OF M ARCHING T ETRAHEDRA
It is tempting to train diffusion models on arbitrary D x , the dataset of fitted tetrahedral grids of objects, with low L Render .
However, the naively trained models generate highly non-smooth surfaces (Fig. 3) due to a mismatch between the L 2 denoising objective on the 4-dimensional inputs (deformation and SDF) and the ideal L 2 reconstruction objective on triangular mesh vertex positions.
Recall that the triangular mesh vertex position v p on a single tetrahedron edge e =
( a, b ) is computed by linear interpolation v p
= ( v a s b − v b s a ) /
( s b − s a ) , in which v a , v b are the positions of tetrahedron vertices a and b , and s a , s b are the corresponding SDF values.
With perfectly fitted v a , v b and s b but a small noise ϵ on s a and s b , the triangular mesh vertex is subject to a perturbation inversely proportional to s b − s a :
[Equation]
An arbitrarily fitted dataset D x does not provide any guarantee on the scale of | s b − s a | and clearly ∥ v p, noisy − v p ∥ can be unbounded in the worst case, even when the L 2 denoising loss is low on D x .
A similar issue arises when we consider the topological change due to a small noise in the SDF values.
For illustration, consider some tetrahedral vertex s a with − C <
s a < 0 where C ≪ 1 is a tiny positive scalar, and all its neighboring vertices possess SDF value of 1.
With a small noise on s a ,
[Figure]
C h a
i r
C a r
A i r p l a n e
T a b l e
R i f l e there can be a high probability of local topological changes ( e.g. , an undesirable hole) in the resulted 3D meshes even when the L 2 denoising loss is small.
Due to these undesired behaviors, we propose to normalize the SDF values and adopt the two-pass optimization scheme described in Sec.
4.2, so that s a
, s b ∈ { +1 , − 1 } and | s b − s a | = 2 for all mesh-generating tedrahedron edges.
With normalized SDFs values and continuous deformation vectors, it is natural train a hybrid diffusion model in which the SDFs are modeled like in D3PM [2] since normalized SDFs take discrete values of ± 1 .
However, we empirically find that it suffices to simply treat the normalized SDFs as float numbers and train a standard Gaussian diffusion model.
The inference thus requires a final normalization operation to round SDFs to ± 1 according to their signs.
Because the generative model is trained under the assumption that
SDF values are un-normalized, a further refinement step can be applied by setting the generated and normalized SDFs as conditional inputs and run a conditional generation step on the deformation vectors only.
4. C ONDITIONAL G ENERATION WITH S INGLE - VIEW I MAGES
In many cases, we would like to generate 3D objects given a (possibly partial) single view RGBD image.
Such a task is simple with MeshDiffusion by a two-stage process, first to fit a tetrahedral grid with the given single RGBD view and second to use diffusion models to (stochastically) correct the wrong parts.
We follow the procedure shown in Alg. 2.
For the conditional inference process, we consider the simplest conditional generation method [16] (called replacement method in [17]).
Specifically, we sample ˆ x t − 1 from x t and replace ˆ x bt − 1 by p t − 1 ( x bt − 1 | x a 0 ) which is defined by the forward diffusion.
More details are included in the appendix.
E XPERIMENTS AND R ESULTS 5. G ENERAL S ETTINGS
Data Fitting .
We fit tetrahedral grids on five ShapeNet subcategories: Chair, Airplane, Car, Rifle and
Table, the same set of categories used in [9, 55].
Fitting each object takes roughly 20-30 minutes on a single Quadro RTX 6000 GPU.
We use the same train/test split in [55].
The detailed architecture, training and hyperparameter settings are explained in the appendix.
[Table]
Generative Model .
To show the effectiveness and universality of our approach of training diffusion models on 3D meshes, for the U-Net architecture we take an extremely approach by simply replacing the 2D convolution and other 2D modules in the U-Net architecture used by [47] with their 3D counterparts.
To improve model capacity, we provide to the model as a conditional input a mask indicating which lattice sites are artificially introduced.
We slightly increase the depths of the network due to the larger input size.
We do not tune the hyperparameters of diffusion SDEs but instead use the same set of hyperparemters as described in [47].
We train discrete-time category-specific diffusion models for all datasets for 90k iterations with batch size 48 on 8 A100-80GB GPUs.
The training process takes 2 to 3 days.
For ablation on alternative architectures, we train GANs on our datasets with an architecture similar to the one used in [51].
Our SDF-based baselines include IM-GAN [9] and SDF-StyleGAN [55].
We also compare MeshDiffusion against GET3D [13] which also uses DMTet for mesh parametrization.
5. U NCONDITIONAL G ENERATION
We randomly sample 3D shapes from each of the trained generative models for each dataset.
We remove isolated meshes of tiny sizes and then apply both remeshing and the standard Laplace smoothing [34] on all the generated meshes (smoothed with λ = 0 .
25 with 5 optimization steps).
We visualize samples produced by MeshDiffusion and the existing state-of-the-art 3D mesh generative models in Fig.
We note that MeshDiffusion produces the sharpest samples and preserve the most of geometric details, while pure SDF-based methods tend to be too smooth.
Part of the reason is that these SDF-based methods assume a very smooth interpolation between points, while MeshDiffusion explicitly models the interpolation in a piecewise linear way.
5.2. Q UANTITATIVE E VALUATION
[Table]
In Table 1, we show various point cloud distances [1, 55] and light field distance (LFD) [8] between the test set and the generated samples by sampling point clouds of size 2048 from both ground truth and generated meshes. Results consistently show that
MeshDiffusion better captures the geometric details, and therefore achieves better point cloud metrics in most of the cases.
For a more detailed description on the metrics and our experimental settings, please refer to the appendix.
We also follow [55] and measure Frechet inception distances (FIDs) [4] on rendered views as a proxy of goodness of surfaces.
Same as in [55], the views are rendered with 24 camera poses uniformly distributed on a sphere; 3 fixed point light sources are used to light the meshes; a gray diffuse material is used for all meshes.
The FID score of each view is computed with an ImageNet-pretrained model
[Table]
and averaged to obtain the FID scores for 3D shapes.
With suitable hyperparameters for resolution and Laplacian smoothing, our MeshDiffusion is able to achieve competitive scores compared to
SDF-StyleGAN.
We note, however, the computed FID is not a genuine score for 3D shapes since the distribution of rendered views of these meshes is very different from the distribution of natural RGB images.
In addition, the multiview FID score used by SDF-StyleGAN [55] assumes flat shading, while it is more natural and common to use other shading methods like Phone shading for smoother images.
Results in Table 2 show that even with such a over-simplified shading, MeshDiffusion still yields very competitive FID scores compared to recent state-of-the-art methods.
Additionally, we perform ablation study on the choices of models in Table.
It is clear that the our SDF normalization strategy described in Sec. 4.3 is beneficial for the diffusion model and our diffusion model is better suited for our mesh generation setting.
5. C ONDITIONAL G ENERATION
[Figure]
Our model can generate meshes by conditioning on a single-view RGBD image, as shown in Fig.
Because the geometry estimated from the single-view fitting is not perfect even in the given single view, we allow the originally-fixed tetrahedral vertices to be slightly updated by the diffusion model near the end of the diffusion completion process (in our experiments, T = 50 ; the inference process starts at T =and ends at T = 0 ).
Results demonstrate that MeshDiffusion can generate plausible and reasonable completion results conditioned on the given view.
5. I NTERPOLATION R ESULTS
We use DDIM [45] to convert the stochastic sampling process into a deterministic one.
With DDIM, the initial random Gaussian noise x T is treated as the “latent code” of the generated image, and we run the following sampling process:
[Equation]
Following the settings in DDIM paper [45], we use spherical interpolation for the latent codes.
We visualize some of the interpolation sequences in Fig.
We set the number of inference steps tofor faster inference and use quadratic time spacing as described in [45].
5. T EXT - CONDITIONED T EXTURE G ENERATION
We show in Fig.
1 and Fig. 6 that the 3D meshes generated by MeshDiffusion can be easily painted with some texture generation methods.
In our experiment, we use a recent work – TEXTure [39] to generate text-conditioned textures on generated raw 3D meshes.
As can be observed from the results, MeshDiffusion along with TEXTure can produce fairly realistic and reasonable textured 3D mesh models.
With more advanced texture generation methods, we believe that the advantage of
MeshDiffusion’s high-quality 3D geometry can be better demonstrated.
D ISCUSSION
Optimization issues with DMTet.
While DMTet is capable of fitting geometries, it fails in cases where the underlying topology is complex.
Besides, it is not encouraged to learn the true topology of shapes and may produce invisible topological holes by contracting the neighboring triangular mesh [Figure] vertices close enough.
Furthermore, we observe that the optimization process with differentiable rasterization of 3D meshes may produce floating and isolated meshes, especially when depth supervision is introduced.
It is therefore worth designing better optimization techniques, regularization methods and possibly better parametrization of meshes for the purpose of training mesh diffusion models.
[Figure]
Diffusion model design.
Our experiments demonstrate the effectiveness of the simple design with few hyperparameter and architecture changes: a 3D-CNN-based UNet on augmented cubic grids with DDPM.
We note that it is straightforward to utilize advanced diffusion models and switch to more memory-efficient architectures.
Especially, since it is possible to regularize
SDFs with methods such as Lipschitz constraints [27], it is a promising approach to train diffusion models on the latent space produced by a regularized autoencoder, the same strategy adopted in [40, 54].
Limitations.
Diffusion model typically assumes a known dataset in the input modality (augmented tetrahedral grids in our case), but to efficiently train diffusion models on 2D images, we need a better way to amortize the costs and fully leverage the power of differentiable rendering.
Our paper avoids this important aspect but instead adopts the twostage approach of "reconstruction-then-generation".
Moreover, in our formulation, the differentiable renderer is useful only during the tetrahedral grid creation process, while in principle we believe there can be ways to incorporate the differentiable render in the training and inference process of diffusion models.
Finally, our diffusion model is built with a very naïve architecture, thus limiting the resolution of input tetrahedral grids, while we notice that some of the fine details cannot be fully captured with the current resolution of 64 during the dataset creation stage.
With better architecture designs or adaptive resolution techniques (as in [43]), we may greatly increase the resolution and generate a more diverse set of fine-level geometric details.
C ONCLUSION
We demonstrate that our MeshDiffusion , a diffusion model on 3D meshes parameterized by tetrahedral grids, is able to generate fine-details of 3D shapes with arbitrary topology.
Such a model is minimally designed but still outperforms the baselines.
We demonstrate that, despite being trained and performing inference in the 3D space, can be easily used when only 2.5D information is available.
We believe that our model can potentially shed some light on future studies of score-based models for learning and generating high-fidelity shapes.
Future work may include but not limited to text-conditioned mesh generation with diffusion models, joint synthesis of texture and geometry with more realistic materials and motion synthesis and physical simulation on generated meshes.
A CKNOWLEDGEMENT
We thank Yuliang Xiu, Jinlong Yang, Tim Xiao, Haiwen Feng, Yandong Wen for constructive suggestions.
We would like to thank Samsung Electronics Co., Ldt.
for funding this research.
Disclosure.
MJB has received research gift funds from Adobe, Intel, Nvidia, Meta/Facebook and
Amazon.
MJB has financial interests in Amazon, Datagen Technologies and Meshcapade GmbH.
While MJB is a part-time employee of Meshcapade, his research was performed solely at, and funded solely by, the Max Planck Society.
DN is supported by NSERC Discovery Grant (RGPIN-5011360) and LP is supported by NSERC Discovery Grant (RGPIN-04653).
Ethics Statement.
Our model is developed for general 3D mesh generation and is in a very preliminary stage in terms of automatically generating product-quality meshes.
Still, our model may potentially be used to generate inappropriate contents if trained on specific datasets
Contents
A Details on
A.1 Architecture and Losses . . . . . . . . . . . . . . . . . . . . . . . . . . . . .
A.2 Implmentation Details . . . . . . . . . . . . . . . . . . . . . . . . . . . . . .
B
D Generation Sequence
E Nearest
Visualization
F Qualitative
GET3D
G Inner Structure Generation
H More Unconditional Generation Samples
A D ETAILS ON D ATASET P REPARATION A. A RCHITECTURE AND L OSSES
RGB and silhouette loss.
By interpolating between a black background image and the binary mask (produced by rasterization to indicate the existence of rasterized meshes on each pixel), we obtain a smoothed binary mask M and M gt , for the fitted meshes and predicted meshes respectively.
We then compute the silhouette loss:
[Equation]
We compute the averaged pixel-wise log- L 1 loss between rendered images and ground truth images:
[Equation]
in which D pose is the set of random camera poses which always look at the center of the object, and x is the sampled tetrahedral grid from the dataset D .
Depth loss.
We use a mixed loss for depth: when the depth is greater than a threshold (set to 1 . 0 ), we use a L 2 loss; otherwise, we switch to L 1 loss as we find it producing much better surfaces.
[Equation]
The depth images are computed by barycentric interpolation following rasterization, in which the background is assumed to have a large default depth of 20 , which is much greater than the object depth (constrained in [ − 1 , 1] ).
To better remove inner artifacts and generate inner structures, we also include the depth loss on the second layer from rasterization, i.e. , the second triangular mesh (if any) intersected by each view ray.
We weight this second layer depth loss term by 0 .
1 as we observe that it can interfere with the first layer depth loss.
The complete depth loss term is thus
[Equation]
Chamfer loss.
At every iteration, we randomly sample 50 , 000 points from both ground truth meshes and predicted meshes and compute the standard Chamfer distance:
[Equation]
SDF regularization loss.
Following [32], we use the same L 2 regularization loss and penalize the difference of SDFs of two neighboring tetrahedral vertices ( i.e. , E tet the set of all edges in the tetrahedral grid) as in [32] so that the occluded regions do not produce complex inner geometry.
It is also helpful for optimizing tetrahedral grids.
[Equation]
Hyperparameter setting.
We set α image =
α Chamfer = 1 .
and α depth = 100 . 0 .
We set α SDF to 0 .and
use a linear decay of the scale towards 0 .
We use an Adam optimizer for all the parameters with a learning rate of 5 e − 4 and ( β 1 , β 2 ) =
We train both reconstruction passes with 5000 iterations.
A. I MPLMENTATION D ETAILS
The initial tetrahedral grid is initialized in a cube [ − 1 , 1] 3 by a dense body-centered cubic (BCC) tiling of tetrahedra [24] (see https://github.com/crawforddoran/quartet for code examples).
As some tetrahedral generation packages produce additional tetrahedral vertices on the boundary of the cube which breaks translational symmetry (hence detrimental to the performance of 3D CNN), we remove such symmetry-breaking boundaries if any.
We find that using the clipped deformation vectors is better than using the tanh -ed deformation vectors as they remove some nonlinearity.
And to ensure that the deformation vectors are always differentiable, we clip the deformation vectors only after each gradient update, but not during the forward computational pass.
We set the range of deformation to be three times of that used in [43] so that the meshes can better capture details in the absence of SDF scales (Sec. 4.3), especially when the grid resolution is low.
As self-intersecting meshes may appear, it is necessary to perform standard mesh processing operations including remeshing and Laplacian smoothing.
For grids of higher resolution ( e.g. , 128), we instead set the deformation to be 1.5x of hat used in [43].
Because we normalize SDFs after the first reconstruction pass, it is more desirable for the underlying shape to be captured mostly by the topology implied by SDF values, not the vertex deformation.
To encourage convergence to such solutions, during the first 2000 iterations, we periodically scale the learned SDF values by 0 .
4 for every 300 iterations.
We observe that depth supervision is not enough to remove some isolated floater meshes in the space.
Therefore, we cull these floaters by setting the SDF values of all tetrahedral vertices lying outside the visual hull to positive values (with the assumption that positive values represent ouside-ness).
Due to constraints on computational resources, we do not perform culling in a full 3D way, but use rendered shape silhouette to determine floaters visible in the rendered views only.
We notice that some of the meshes in ShapeNet are not watertight while some others have parts which are very thin.
Depth supervision on these parts can lead to more topological holes.
Therefore, during rasterization we extract the depths (stored in Z-buffers) of the two closet meshes for each pixel and decide if we compute the depth loss on these pixels according to the difference in depth of these two meshes.
Specially, depth supervision is not included if the difference is too small or there is only one rasterized mesh (meaning that the ray only intersects with a single mesh).
Motivated by the fact that only SDFs of the mesh-generating tetrahedra ( i.e. , tetrahedra with vertices of different SDF signs) matter, we set the SDFs of all the non-mesh-generating tetrahedral vertices to ± 1 (signs depending on if vertices inside/outside the shape) to reduce the complexity of the dataset.
For the single-view reconstruction stage in the conditional generation experiments, we observe that the lack of other views leads to many floating artifacts when the target shape consists of a bulk volume.
Therefore, during the first 300 iterations, we periodically (every 10 iterations) query the nearest neighbor ground truth mesh vertices for every tetrahedral vertex, and determine if the vertex is in front of the camera and not behind the current view (by computing the nearest surface normal, the displacement from the nearest face center and the view direction from camera).
If the tetrahedral vertex satisfies the condition, we set its SDF value to +1 .
[Table]
B D ETAILS ON D IFFUSION M ODEL
We adapt the network of DDPM in [16] and use a base width of 64 .
The encoder in the U-Net is shown in Fig. 4, and the decoder follows the same but reverse pattern.
For higher resolution grids, we slightly reduce the number of layers in each resolution stage but double the base width.
All the values at the artificially-introduced sites of a cubic lattice are set to zero.
We append to the cubic grid in the first and last few layers a binary mask that indicates which vertices are from the tetrahedral grid and which are fake.
The score matching loss is also masked accordingly so that the augmented vertices in the predicted cubic grids do not contribute.
To prevent overfitting, we augment the dataset by randomly translate all tetrahedron vertices by the same but tiny amount.
C Q UANTITATIVE M ETRICS FOR 3D M ESHES
We briefly explain the metrics we used for evaluating the quality of 3D Meshes.
Minimum Matching Distance (MMD).
MMD measures the average distance to the nearest neighbor of individual points in one point cloud to another point cloud.
Coverage.
Suppose every point cloud in a set A is approximated by its nearest neighbor in another set B .
Coverage measures the fraction of elements in B which are used to cover A in the nearest neighbor matching sense.
Higher the coverage, Better that A is a representative set of B .
Leave-one-out Accuracy (1-NNA).
As its name indicates, 1-NNA is the average leave-one-out accuracy of the 1-NN classifier fitted on A to classify B .
It measures if each element in the set A is important in representing another set B .
A low 1-NNA score means that A well covers B .
Jensen-Shannon Divergence (JSD).
With JSD, we compute the distance between distribution of ground truth point clouds and that of the generated point clouds: JSD ( P A , P B ) =
12 KL ( P A || M ) + 12 KL ( P B || M ) , in which M = 12 ( P A + P B ) .
Light Field Distance (LFD).
LFD leverages the so-called light field descriptor [8], which combines local region-based and contour-based descriptors to measure the silhouettes and thus object shapes.
D G ENERATION S EQUENCE
For each time step t ∈ { 1 , ..., T } during the DDPM inference process, we can compute the predicted x 0 by the current value of x t [16]:
[Equation]
, in which s θ ( x, t ) is the learned score function predictor, α t =
Π ts =1 (1 − β s ) and β t is the noise scale hyperparameter in DDPM.
We show example generation trajectories in Fig. 8 by visualizing the predicted x 0 through time.
All ˆ x 0 ’s are clipped to [0 , 1] .
We visualize the sequence from T = 250 to T = 50 , as the predicted tetrahedral grids before T = 256 produce nothing other than noise.
E N EAREST N EIGHBOR V ISUALIZATION
We show the nearest neighbors of some generated samples in the validation set in Fig.
[Figure]
F Q UALITATIVE C OMPARISON AGAINST GET3D
We visulize some generated meshes from MeshDiffusion and GET3D in Fig.
Both models are able to generate finer details of shapes due to the use of DMTet.
On the car category of ShapeNet, generated meshes by GET3D tend to have fewer or less noticeable holes, possibly because GET3D is directly trained on adversarial losses on rendered images.
In comparison, MeshDiffusion has to learn the complex and often invisble 3D structures.
However, potentially due to the limited capacity of the decoder (compared to a recurrent U-Net in diffusion models) and the hardness of GAN training ( v.s. supervised learning with denoising loss), we observe that the failure cases appear more often compared to MeshDiffusion.
And the novel examples in MeshDiffusion seem to make more sense compared to those from GET3D.
[Figure]
G I NNER S TRUCTURE G ENERATION
Since our MeshDiffusion is explicitly trained with 3D information, it is capable of generating inner structures invisble from the outside.
We give a comparison between samples generated by
MeshDiffusion (trained with a higher resolution of 128) and GET3D in Fig.
To fully evaluate the capability of surface generation (as GET3D is only trained to optimize surface appearance), we follow the same procedure described in https://github.com/nv-tlabs/
GET3D/tree/master/evaluation_scripts and compare the point cloud metrics in Table.
5, with generated sample set 5x larger than the validation set.
[Figure]
H M ORE U NCONDITIONAL G ENERATION S AMPLES
[Figure]