SALD: S IGN A GNOSTIC L EARNING WITH D ERIVATIVES
Anonymous authors
Paper under double-blind review
A BSTRACT
Learning 3D geometry directly from raw data, such as point clouds, triangle soups, or unoriented meshes is still a challenging task that feeds many downstream computer vision and graphics applications.
In this paper, we introduce SALD: a method for learning implicit neural representations of shapes directly from raw data.
We generalize sign agnostic learning (SAL) to include derivatives: given an unsigned distance function to the input raw data, we advocate a novel sign agnostic regression loss, incorporating both pointwise values and gradients of the unsigned distance function.
Optimizing this loss leads to a signed implicit function solution, the zero level set of which is a high quality and valid manifold approximation to the input 3D data.
The motivation behind SALD is that incorporating derivatives in a regression loss leads to a lower sample complexity, and consequently better ﬁtting.
In addition, we provide empirical evidence, as well as theoretical motivation in 2D that SAL enjoys a minimal surface property, favoring minimal area solutions.
More importantly, we are able to show that this property still holds for SALD, i.e., with derivatives included.
We demonstrate the efﬁcacy of SALD for shape space learning on two challenging datasets: ShapeNet (Chang et al., 2015) that contains inconsistent orientation and non-manifold meshes, and D-Faust (Bogo et al., 2017) that contains raw 3D scans (triangle soups).
On both these datasets, we present state-of-the-art results.
I NTRODUCTION
Recently, neural networks (NN) have been used for representing and reconstructing 3D surfaces.
Current NN-based 3D learning approaches differ in two aspects: the choice of surface representation, and the supervision method.
Common representations of surfaces include using NN as parameteric charts of surfaces (Groueix et al., 2018b; Williams et al., 2019); volumetric implicit function representation deﬁned over regular grids (Wu et al., 2016; Tatarchenko et al., 2017; Jiang et al., 2020); and NN used directly as volumetric implicit functions (Park et al., 2019;
Mescheder et al., 2019; Atzmon et al., 2019; Chen & Zhang, 2019), referred henceforth as implicit neural representations .
Supervision methods include regression of known or approximated volumetric implicit representations (Park et al., 2019; Mescheder et al., 2019; Chen & Zhang, 2019), regression directly with raw 3D data (Atzmon & Lipman, 2020; Gropp et al., 2020; Atzmon & Lipman, 2020), and differentiable rendering using 2D data (i.e., images) supervision (Niemeyer et al., 2020; Liu et al., 2019;
Saito et al., 2019; Yariv et al., 2020).
The goal of this paper is to introduce SALD, a method for learning implicit neural representations of surfaces directly from raw 3D data .
The beneﬁt in learning directly from raw data, e.g., nonoriented point clouds or triangle soups (e.g., Chang et al.
and raw scans (e.g., Bogo et al.
(2017)), is avoiding the need for a ground truth signed distance representation of all train surfaces for supervision.
This allows working with complex models with inconsistent normals and/or missing parts.
In Figure 1 we show reconstructions of zero level sets of SALD learned implicit neural representations of car models from the ShapeNet dataset (Chang et al., 2015) with variational autoencoder; notice the high detail level and the interior, which would not have been possible with, e.g., previous data pre-processing techniques using renderings of visible parts (Park et al., 2019).
Our approach improves upon the recent Sign Agnostic Learning (SAL) method (Atzmon & Lipman, 2020) and shows that incorporating derivatives in a sign agnostic manner provides a signiﬁcant improvement in surface approximation and detail.
SAL is based on the observation that given an unsigned distance function h to some raw 3D data X ⊂ R 3 , a sign agnostic regression to h will introduce new local minima that are signed versions of h ; in turn, these signed distance functions can be used as implicit representations of the underlying surface.
In this paper we show how the sign agnostic regression loss can be extended to compare both function values h and derivatives ∇ h , up to a sign.
The main motivation for performing NN regression with derivatives is that it reduces the sample complexity of the problem (Czarnecki et al., 2017), leading to better accuracy and generalization.
For example, consider a one hidden layer NN of the form f ( x ) = max { ax, bx } + c .
Prescribing two function samples at {− 1 , 1 } are not sufﬁcient for uniquely determining f , while adding derivative information at these points determines f uniquely.
We provide empirical evidence as well as theoretical motivation suggesting that both SAL and
SALD possess the favorable minimal surface property (Zhao et al., 2001), that is, in areas of missing parts and holes they will prefer zero level sets with minimal area.
We justify this property by proving that, in 2D, when restricted to the zero level-set (a curve in this case), the SAL and SALD losses would encourage a straight line solution connecting neighboring data points.
We have tested SALD on the dataset of man-made models, ShapeNet (Chang et al., 2015), and human raw scan dataset, D-Faust (Bogo et al., 2017), and compared to state-of-the-art methods.
In all cases we have used the raw input data X as is and considered the unsigned distance function to
X , i.e., h X , in the SALD loss to produce an approximate signed distance function in the form of a neural network.
Comparing to state-of-the-art methods we ﬁnd that SALD achieves superior results on this dataset.
On the D-Faust dataset, when comparing to ground truth reconstructions we report state-of-the-art results, striking a balance between approximating details of the scans and avoiding overﬁtting noise and ghost geometry.
Summarizing the contributions of this paper:
• Introducing sign agnostic learning with derivatives.
Identifying and providing a theoretical justiﬁcation for the minimal surface property of sign agnostic learning in 2D. • Training directly on raw data (end-to-end) including unoriented or not consistently oriented triangle soups and raw 3D scans.
P REVIOUS WORK
Learning 3D shapes with neural networks and 3D supervision has shown great progress recently.
We review related works, where we categorize the existing methods based on their choice of 3D surface representation.
Parametric representations.
The most fundamental surface representation is an atlas , that is a collection of parametric charts f : R 2 → R 3 with certain coverage and transition properties (Do Carmo, 2016).
Groueix et al.
(2018b) adapted this idea using neural network to represent a surface as union of such charts; Williams et al.
(2019) improved this construction by introducing better transitions between charts; Sinha et al.
(2016) use geometry images (Gu et al., 2002) to represent an entire shape using a single chart; Maron et al.
(2017) use global conformal parameterization for learning surface data; Ben-Hamu et al.
(2018) use a collection of overlapping global conformal charts for human-shape generative model.
Hanocka et al.
(2020) shrink-wraps a template mesh to ﬁts a point cloud.
The beneﬁt in parametric representations is in the ease of sampling the learned surface (i.e., forward pass) and work directly with raw data (e.g., Chamfer loss); their main struggle is in producing charts that are collectively consistent, of low distortion, and covering the shape.
Implicit representations.
Another approach for representing surfaces is as zero level sets of a function, called an implicit function .
There are two popular methods to model implicit volumetric functions with neural networks: i) Convolutional neural network predicting scalar values over a predeﬁned ﬁxed volumetric structure (e.g., grid or octree) in space (Tatarchenko et al., 2017; Wu et al., 2016); and ii) Multilayer Perceptron of the form f : R 3 → R deﬁning a continuous volumetric function (Park et al., 2019; Mescheder et al., 2019; Chen & Zhang, 2019).
Currently, neural networks are trained to be implicit function representations with two types of supervision: (i) regression of samples taken from a known or pre-computed implicit function representation such as occupancy function (Mescheder et al., 2019; Chen & Zhang, 2019) or a signed distance function (Park et al., 2019); and (ii) working with raw 3D supervision, by particle methods relating points on the level sets to the model parameters (Atzmon et al., 2019), using sign agnostic losses (Atzmon & Lipman, 2020), or supervision with PDEs deﬁning signed distance functions (Gropp et al., 2020).
Primitives.
Another type of representation is to learn shapes as composition or unions of a family of primitives .
Gradient information have been used to improve and facilitate ﬁtting of invariant polynomial representations (Tasdizen et al., 1999; Birdal et al., 2019).
Li et al.
(2019) represent a shape using a parametric collection of primitives.
Genova et al.
(2019; 2020) use a collection of
Gaussians and learn consistent shape decompositions.
Chen et al.
(2020) suggest a differentiable
Binary Space Partitioning tree (BSP-tree) for representing shapes.
Deprelle et al.
(2019) combine points and charts representations to learn basic shape structures.
Deng et al.
(2020) represent a shape as a union of convex sets.
Williams et al.
(2020) learn cites of Voronoi cells for implicit shape representation.
Template ﬁtting.
Lastly, several methods learn 3D shapes of a certain class (e.g., humans) by learning the deformation from a template model.
Classical methods use matching techniques and geometric loss minimization for non-rigid template matching (Allen et al., 2002; 2003; Anguelov et al., 2005).
Groueix et al.
(2018a) use an auto-encoder architecture and Chamfer distance to match target shapes.
Litany et al.
(2018) use graph convolutional autoencoder to learn deformable template for shape completion.
M ETHOD
Given raw geometric input data X ⊂ R 3 , e.g., a triangle soup, our goal is to ﬁnd a multilayer perceptron (MLP) f
: R 3 × R m → R whose zero level-set,
S = (cid:8) x ∈ R 3 | f ( x ; θ ) = 0 (cid:9) (1) is a manifold surface that approximates X .
Sign agnostic learning.
Similarly to SAL, our approach is to consider the (readily available) unsigned distance function to the raw input geometry,
[Equation]
and perform sign agnostic regression to get a signed version f of h .
SAL uses a loss of the form
[Equation]
where D is some probability distribution, e.g., a sum of gaussians with centers uniformly sampled over the input geometry X , and τ is an unsigned similarity.
That is, τ ( a, b ) is measuring the difference between scalars a, b ∈ R up-to a sign.
For example
[Equation] [Equation]
[Figure]
is an example that is used in Atzmon & Lipman (2020).
The key property of the sign agnostic loss in equation 3 is that, with proper weights initialization θ 0 , it ﬁnds a new signed local minimum f which in absolute value is similar to h .
In turn, the zero level set S of f is a valid manifold describing the data X .
Sign agnostic learning with derivatives.
Our goal is to generalize the SAL loss (equation 3) to include derivative data of h and show that optimizing this loss provides implicit neural representations,
S , that enjoy better approximation properties with respect to the underlying geometry X .
Generalizing equation 3 requires designing an unsigned similarity measure τ for vector valued functions.
The key observation is that equation 4 can be written as τ ( a, b )
= min {| a − b | , | a + b |} , a, b ∈ R , and can be generalized to vectors a , b ∈ R d by
[Equation]
We deﬁne the SALD loss:
[Equation]
where λ > 0 is a parameter, D (cid:48) is a probability distribution, e.g., it could be identical to D , or uniform over the input geometry X , and ∇ x f ( x ; θ ) , ∇ x h ( x ) are the gradients f, h (resp.)
with respect to their input x .
In Figure 2 we show the unsigned distance h to an L-shaped curve (left), and the level sets of the MLPs optimized with the SALD loss (middle) and the SAL loss (right); note that SALD loss reconstructed the sharp features (i.e., corners) of the shape and the level sets of h , while SAL loss smoothed them out; the implementation details of this experiment can be found in Appendix A.4.
Minimal surface property.
We show that the SAL and SALD losses possess a minimal surface property (Zhao et al., 2001), that is, they strive to minimize surface area of missing parts.
For example, Figure 4 shows the unsigned distance to a curve with a missing segment (left), and the zero level sets of MLPs optimized with SALD loss (middle), and SAL loss (right).
Note that in both cases the zero level set in the missing part area is the minimal length curve (i.e., a line) connecting the points of that missing part.
SALD also preserves sharp features of the rest of the shape.
Figure A1 in the supplementary shows adddtional 2D experiments comparing to the Implicit Geometric Regularization (IGR) method (Gropp et al., 2020) that learns implicit representations by regularizing the gradient norm and do not posses the minimal surface property.
We will provide a theoretical justiﬁcation to this property in the 2D case.
We consider a geometry deﬁned by two points in the plane, X = { x 1 , x 2 } ⊂ R 2 and possible solutions where the zero level set curve S is connecting x 1 and x 2 .
We prove that among a class of
[Figure]
curves U connecting x 1 and x 2 , the straight line minimizes the losses in equation 3 and equationrestricted to U , when assuming uniform distributions D , D (cid:48) .
We assume (without losing generality) that x 1 = (0 , 0) T , x
2 = ( (cid:96), 0) T and consider curves u ∈ U deﬁned by u ( s )
= ( s, t ( s ))
T , where s ∈ [0 , (cid:96) ] , and t : R → R is some differentiable function such that t (0) =
=
t ( (cid:96) ) , see Figure 3.
For the SALD loss we prove the claim for a slightly simpliﬁed agnostic loss motivated by the following lemma proved in Appendix A.1:
Lemma 1.
For any pair of unit vectors a , b : min {(cid:107) a − b (cid:107) , (cid:107) a + b (cid:107)} ≥ | sin ∠ ( a , b ) | .
We consider τ ( a , b ) = | sin ∠ ( a , b ) | for the derivative part of the loss in equation 6, which is also sign agnostic.
Theorem 1.
Let X
= { x 1 , x 2 } ⊂ R 2 , and the family of curves U connecting x 1 and x 2 .
Furthermore, let loss SAL ( u ) and loss SALD ( u ) denote the losses in equation 3 and equation 6 (resp.)
when restricted to u with uniform distributions D , D (cid:48) .
Then in both cases the straight line, i.e., the curve u ( s ) =
( s, 0) , is the strict global minimizer of these losses.
Proof.
The unsigned distance function is
[Equation]
[Equation] From symmetry it is enough to consider only the ﬁrst half of the curve, i.e., s ∈ [0 , (cid:96)/ 2) .
Then, the
SAL loss, equation 3, restricted to the curve u (i.e., where f vanishes) takes the form loss SAL
[Equation]
[Equation] where (cid:112) 1 + ˙ t 2 ds is the length element on the curve u , and τ ( f ( s, t ; θ ) , h ( s, t ))
=
| h ( s, t ) | = √ s 2 + t 2 , since f ( s, t ; θ ) = 0 over the curve u .
Plugging t ( s ) ≡ 0 in loss SAL ( u )
we see that the curve u
= ( s, 0) T , namely the straight line curve from x 1 to 0 .
5( x 1 + x 2 ) is a strict global minimizer of loss SAL ( u ) .
Similar argument on s ∈ [ (cid:96)/ 2 , (cid:96) ] prove the claim for the SAL case.
For the SALD case, we want to calculate τ ( ∇ x f ( u ; θ ) , ∇ x h ( u )) restricted to the curve u ; let a = ∇ x f ( u ; θ ) and b = ∇ x h ( u ) .
First, b = ( s 2 + t 2 ) − 1 / 2 ( s, t ) T .
Second, a is normal to the curve u , therefore it is proportional to ˙ u ⊥ =
( − ˙ t, 1) T .
Next, note that
[Equation]
where the last equality can be checked by differentiating (cid:107) ( s, t ) (cid:107) w.r.t.
s .
Therefore, loss SALD ( u ) − loss SAL ( u ) λ
[Table]
This bound is achieved for the curve u
= ( s, 0) , which is also a minimizer of the
SAL loss.
The straight line also minimizes this version of the SALD loss since loss SALD ( u ) =
(loss SALD ( u ) − loss SAL ( u ))
+ loss SAL ( u ) .
E XPERIMENTS
We tested SALD on the task of shape space learning from raw 3D data.
We experimented with two different datasets: i)
ShapeNet dataset (Chang et al., 2015), containing synthetic 3D Meshes; and ii)
D-Faust dataset (Bogo et al., 2017) containing raw 3D scans.
Furthermore, we empirically test our sample complexity hypothesis (i.e., that incorporating derivatives improve sample complexity) by inspecting surface reconstruction accuracy for SAL and SALD when trained with ﬁxed size sample sets.
Shape space learning architecture.
Our method can be easily incorporated into existing shape space learning architectures: i) Auto-Decoder (AD) suggested in Park et al.
; and the ii)
Modiﬁed Variational Auto-Encoder (VAE) used in Atzmon & Lipman (2020).
For VAE, the encoder is taken to be PointNet (Qi et al., 2017).
For both options, the decoder is the implicit representation in equation 1, where f ( x ; θ ) is taken to be an 8-layer MLP with 512 hidden units in each layer and
Softplus activation.
In addition, to enable sign agnostic learning we initialize the decoder weights, θ , using the geometric initialization from Atzmon & Lipman (2020).
See Appendix A.2.4 for more details regarding the architecture.
The point samples x , x (cid:48) for the empirical computation of the expectations in equation 6 are drawn according to distributions D , D (cid:48) explained in Appendix A.2.1.
Baselines.
The baseline methods selected for comparison cover both existing supervision methodologies: DeepSDF (Park et al., 2019) is chosen as a representative out of the methods that require pre-computed implicit representation for training.
For methods that train directly on raw 3D data, we compare versus SAL (Atzmon & Lipman, 2020) and IGR (Gropp et al., 2020).
See Appendix
A.6 for a detailed description of the quantitative metrics used for evaluation.
[Table]
4. S HAPE N ET
In this experiment we tested the ability of
SALD to learn a shape space by training on a challenging 3D data such as nonmanifold/non-orientable meshes.
We tested
SALD with both AD and VAE architectures.
In both settings, we set λ = 0 .
1 for the
SALD loss.
We follow the evaluation protocol as in DeepSDF (Park et al., 2019): using the same train/test splits, we train and evaluate our method on 5 different categories.
Note that comparison versus IGR is omitted as IGR requires consistently oriented normals for shape space learning, which is not available for ShapeNet, where many models have non-consistent triangles’ orientation.
Results.
Table 1 and Figure 6 show quantitative and qualitative results (resp.)
for the held-out test set, comparing SAL, DeepSDF and SALD.
As can be read from the table and inspected in the ﬁgure, our method, when used with the same auto-decoder as in DeepSDF, compares favorably to
DeepSDF’s reconstruction performance on this data.
Qualitatively the surfaces produced by SALD are smoother, mostly with more accurate sharp features, than SAL and DeepSDF generated surfaces.
Figure 1 shows typical train and test results from the Cars class with VAE.
Figure 5 shows a comparison between SALD shape space learning with
VAE and AD in reconstruction of a test car model (left).
Note that the AD (middle) seems to produce more details of the test model than the VAE (right), e.g., steering wheel and headlights.
Figureshow SALD (AD) generated shapes via latent space interpolation between two test models.
4. D-F AUST
The D-Faust dataset (Bogo et al., 2017) contains raw scans (triangle soups) of 10 humans in multiple poses.
There are approximately 41k scans in the dataset.
Due to the low variety between adjacent scans, we sample each pose scans at a ratio of 1 : 5 .
The leftmost column in Figure 8 shows examples of raw scans used for training.
For evaluation we use the registrations provided with the data set.
Note that the registrations where not used for training.
We tested SALD using the
VAE architecture, with λ = 1 . 0 set for the SALD loss.
We followed the evaluation protocol as in Atzmon & Lipman (2020), using the same train/test split.
Note that Atzmon & Lipman (2020) already conducted a comprehensive comparison of SAL versus DeepSDF and AtlasNet (Groueix et al., 2018b), establishing SAL as a state-of-the-art method for this dataset.
Thus, we focus on comparison versus SAL and IGR.
Results.
Table 2 and Figure 8 show quantitative and qualitative results (resp.)
; although SALD does not produces the best test quantitative results, it is roughly comparable in every measure to the best among the two baselines.
That is, it produces details comparable to IGR while maintaining the minimal surface property as SAL and not adding undesired surface sheets as IGR; see the ﬁgure for visual illustrations of these properties: the high level of details of SALD and IGR compared to
SAL, and the extraneous parts added by IGR, avoided by SALD.
These phenomena can also be seen quantitatively, e.g., the reconstruction-to-registration loss of IGR.
Figure 9 show SALD generated shapes via latent space interpolation between two test scans.
Notice the ability of SALD to generate novel mixed faces and body parts.
4. S AMPLE C OMPLEXITY
[Figure]
C h a m f e r
d
i s t a n ce
SALDSAL 1000  0.
[Figure]
SALDSAL
[Figure]
C h a m f e r
d
i s t a n ce (chair) (sofa) (table)
In this experiment we test the sample complexity hypothesis: namely, whether regressing with derivatives improves shape reconstruction accuracy, under a ﬁxed budget of point samples.
This experiment considers 3 different shapes chosen randomly from the chair, sofa and table test sets of the ShapeNet dataset.
For each shape, we prepared a ﬁxed sample set of points { x i } mi =1 , where m ∈ { 1 K, 5 K, 10 K, 20 K } , together with the unsigned distance value and derivative { h ( x i ) , ∇ x h ( x i ) } mi =1 .
The point samples are drawn according to distribution D as explained in Appendix A.2.1.
We separately trained the SAL and SALD losses on the same sample data, using the same hyper-parameters, in two different scenarios: i)
Individual shape reconstruction: optimizing the weights θ of a randomly initialized 8-layer MLP f ( x ; θ ) ; and ii) latent shape reconstruction: given a trained auto-decoder network f ( x , z ; θ ) (as in 4.1), we optimize solely the latent code z , keeping the weights θ ﬁxed.
Lastly, we computed the Chamfer distance between the learned shape S and the input geometry X .
For the individual shape reconstruction, the inset ﬁgure shows the Chamfer distance, d C ( S , X ) , as a function of the sample size m .
Figure 10 shows for each sample size, the learned sofa and table.
Note that SALD demonstrates better approximation to the input geometry in comparison to SAL, [Table] [Figure] [Equation]
[Figure]
[Table]
[Figure]
[Equation] [Figure] [Table] [Equation]
[Table]
[Figure]
in particular as the sample size gets smaller, and thus supporting the sample complexity hypothesis. When optimizing the latent code of a fully trained auto-decoder, the sample size has a little to no effect on the approximation quality of a test shape reconstruction. This can be explained by the fact that the auto-decoder is trained on the maximal sample size, and therefore provides a strong prior for the latent reconstruction. See the supplementary A.5, for the results.
4. L IMITATIONS
Figure 11 shows typical failure cases of our method from the ShapeNet experiment described above.
We mainly suffer from two types of failures: First, since inside and outside information is not known (and often not even well deﬁned in ShapeNet models)
SALD can add surface sheets closing what should be open areas (e.g., the bottom side of the lamp, or holes in the chair).
Second, thin structures can be missed (e.g., the electric cord of the lamp on the left).
A useful strategy to sample thin structures is to make sure the sample frequency is inversely proportional to the distance to the medial axis
Amenta et al.
(1998), where an approximation can be made using curvature estimation.
Furthermore, it is important to note that implicit representations of the type presented in equation 1 cannot model surfaces with boundaries and therefore cannot represent ﬂat dimensionless surfaces with boundaries.
A potential solution could be incorporating additional implicits to handle boundaries.
C ONCLUSIONS
We introduced SALD, a method for learning implicit neural representations from raw data.
The method is based on a generalization of the sign agnostic learning idea to include derivative data.
We demonstrated that the addition of a sign agnostic derivative term to the loss improves the approximation power of the resulting signed implicit neural network.
In particular, showing improvement in the level of details and sharp features of the reconstructions.
Furthermore, we identify the favorable minimal surface property of the SAL and SALD losses and provide a theoretical justiﬁcation in 2D.
Generalizing this theoretical analysis to 3D is marked as interesting future work.
We see two more possible venues for future work: First, it is clear that there is room for further improvement in approximation properties of implicit neural representations.
Although the results in D-Faust are already close to the input quality, in ShapeNet we still see a gap between input models and their implicit neural representations; this challenge already exists in overﬁtting a large collection of diverse shapes in the training stage.
Improvement can come from adding expressive power to the neural networks, or further improving the training losses; adding derivatives as done in this paper is one step in that direction but does not solves the problem completely.
Combining sign agnostic learning with the recent positional encoding method (Tancik et al., 2020) could also be an interesting future research venue.
Another interesting project is to combine the sign-agnostic losses with gradient regularization such as the one employed in IGR (Gropp et al., 2020).
Second, it is interesting to think of applications or settings in which SALD can improve the current state-ofthe-art.
Generative 3D modeling, learning geometry with 2D supervision, or other types of partially observed scans such as depth images are all potentially fruitful options.
A A PPENDIX A. P ROOF OF L EMMA
Lemma 1.
For any pair of unit vectors a , b : min {(cid:107) a − b (cid:107) , (cid:107) a + b (cid:107)} ≥ | sin ∠ ( a , b ) | .
Proof.
Let a , b ∈ R d be arbitrary unit norm vectors.
Then,
[Equation]
Where the last inequality can be proved by considering two cases: α ∈ [0 , π/ 2] and α ∈ [ π/ 2 , π ] , where we denote α = ∠ ( a , b ) .
In the ﬁrst case α ∈ [0 , π/ 2] , cos α ≥ 0 and in this case (cid:113) 1 − cos α= (cid:12)(cid:12) sin α 2 (cid:12)(cid:12) .
The inequality is proved by considering
[Equation]
for α ∈ [0 , π/ 2] .
For the case α ∈ [ π/ 2 , π ] we have (cid:113) 1+cos α= (cid:12)(cid:12) cos α 2 (cid:12)(cid:12) .
This case is proved by considering
[Equation]
for α ∈ [ π/ 2 , π ] unsigned distance
SALD
IGR (Gropp et al., 2020)
A. I MPLEMENTATION D ETAILS A.2. D ATA P REPARATION
Given some raw 3D data X , SALD loss (See equation 6) is computed on points and corresponding unsigned distance derivatives, { h ( x ) } x ∈D and {∇ x h ( x (cid:48) ) } x (cid:48) ∈D (cid:48)
(resp.) sampled from some distributions D and D (cid:48) .
In this paper, we set D
=
D 1 ∪ D 2 , where D 1 is chosen by uniform sampling points { y } from X and placing two isotropic Gaussians, N ( y , σ 21 I ) and N ( y , σ 22 I ) for each y .
The distribution parameter σ 1 depends on each point y , set to be as the distance of the 50 th closest point to y , whereas σ 2 is set to 0 .
3 ﬁxed.
D 2 is chosen by projecting D 1 to S .
The distribution D (cid:48) is set to uniform on X ; note that on X , ∇ x h ( x (cid:48) ) is a sub-differential which is the convex hull of the two possible normal vectors ( ± n ) at x (cid:48) ; as the sign-agnostic loss does not differ between the two normal choices, we arbitrarily use one of them in the loss.
Computing the unsigned distance to X is done using the CGAL library (The CGAL Project, 2020).
To speed up training, we precomputed for each shape in the dataset, 500K samples of the form { h ( x ) } x ∈D and {∇ x h ( x (cid:48) ) } x (cid:48) ∈D (cid:48) .
A.2. G RADIENT COMPUTATION
The SALD loss requires incorporating the term ∇ x f ( x ; θ ) in a differentiable manner.
Our computation of ∇ x f ( x ; θ ) is based on A UTOMATIC D IFFERENTIATION (Baydin et al., 2017) forward mode.
Similarly to Gropp et al.
(2020), ∇ x f ( x ; θ ) is constructed as a network consists of layers of the form
[Equation]
where y (cid:96) denotes the output of the (cid:96) layer in f ( x ; θ ) and θ =
( W (cid:96) , b (cid:96) ) are the learnable parameters.
A.2. T IMINGS AND NETWORK SIZE
In ﬁgure A2, we report the timings and memory footprint of a 8-layer MLP with 512 hidden units.
As the gradients calculation, ∇ x f ( x ; θ ) , is based on automatic differentiation forward mode, in theory it should yield doubling of the forward time.
However, in practice we see that the gap increases as we increase the number of points for evaluation.
For the DFaust experiment (which is the largest dataset in the paper), training was done a batch of 64 shapes and a sample size of 92 2 .
It took around 1.5 days to complete 3000 epochs with 4 Nvidia V100 32GB gpus.
Note that for VAE, the computational cost in test time is equivalent between SAL and SALD.
[Figure]
A.2. A RCHITECTURE D ETAILS VAE A RCHITECTURE
Our VAE architecture is based on the one used in Atzmon & Lipman (2020).
The encoder g ( X ; θ 1 ) , where X ∈ R N × 3 is the input point cloud, is composed of DeepSets (Zaheer et al., 2017) and
PointNet (Qi et al., 2017) layers.
Each layer consists of
[Equation]
where [ · , · ] is the concat operation,
W ∈ R d in × d out and b ∈ R d out are the layer weights and bias and ν ( · ) is the pointwise non-linear ReLU activation function.
Our encoder architecture is:
PFC(3 , 128) → PFC(128 , 128) → PL(128 , 256) →
PFC(256 , 128) → PL(128 , 256) → PFC(256 , 128) →
PL(128 , 256)
→ PFC(256 , 128) → PL(128 , 256) →
PFC(256 , 256) → MaxPool × 2 → FC(256 , 256) , where FC( d in , d out ) :
x
(cid:55)→ ν ( W x + b ) denotes a fully connected layer.
The ﬁnal two fully connected layers outputs vectors µ ∈ R 256 and η ∈ R 256 used for parametrization of a multiviariate
Gaussian N ( µ , diag exp η ) used for sampling a latent vector z ∈ R 256 .
Our encoder architecture is similar to the one used in Mescheder et al.
Our decoder f
([ x , z ] ; θ 2 ) is a composition of 8 layers where the ﬁrst layer is FC(256 + 3 , 512) , middle layers are FC(512 , 512) and the ﬁnal layer is Linear(512 , 1) .
Notice that the input for the decoder is [ x , z ] where x ∈ R 3 and z is the latent vector.
In addition, we add a skip connection between the input to the middle fourth layer.
We chose the Softplus with β = 100 for the non linear activation in the FC layers.
For regulrization of the latent z , we add the following term to training loss
[Equation]
similarly to Atzmon & Lipman (2020).
A UTO -D ECODER A RCHITECTURE
We use an auto-decoder architecture, similar to the one suggested in Park et al.
We deﬁned the latent vector z ∈ R 256 .
The decoder architecture is the same as the one described above for the
VAE.
For regulrization of the latent z , we add the following term to the loss
[Equation]
similarly to Park et al.
A. T RAINING DETAILS
We trained our networks using the A DAM (Kingma & Ba, 2014) optimizer, setting the batch size to 64 .
On each training step the SALD loss is evaluated on a random draw of 92 2 points out of the precomputed 500K samples.
For the VAE, we set a ﬁxed learning rate of 0 . 0005 , whereas for the
AD we scheduled the learning rate to start from 0 . 0005 and decrease by a factor of 0 .
5 everyepochs.
All models were trained for 3000 epochs.
Training was done on 4 Nvidia V-100 GPUs, using PYTORCH deep learning framework (Paszke et al., 2017).
A. F IGURES 2 AND
For the two dimensional experiments in ﬁgures 2 and 4 we have used the same decoder as in the
VAE architecture with the only difference that the ﬁrst layer is FC(2 , 512) (no concatenation of a latent vector to the 2D input).
We optimized using the A DAM (Kingma & Ba, 2014) optimizer, for 5000 epochs.
The parameter λ in the SALD loss was set to 0 . SALDSAL
Sample Size
C h a m f e r
d
i s t a n c e   20000 0.
SALDSAL Sample Size C h a m f e r d
i s t a n c e    0.
SALDSAL
Sample Size
C h a m f e r
d
i s t a n c e (chair) (sofa) (table)
A. S AMPLE C OMPLEXITY
Figures A3 and A4 show quantitative and qualitative results for auto-decoder latent test shape reconstruction on samples of sizes 1 K, 5 K, 10 K, 20 K .
Note that the reconstruction is oblivious to the sample size.
This is possibly due to the fact that the auto-decoder was trained with the maximal sample size.
A. E VALUATION
Evaluation metrics.
We use the following Chamfer distance metrics to measure similarity between shapes:
[Equation]
[Equation]
where
[Equation]
and the sets X i are either point clouds or triangle soups.
In addition, to measure similarity of the normals of triangle soups T 1 , T 2 , we deﬁne:
[Equation]
where
[Equation]
where ∠ ( a , b ) is the positive angle between vectors a , b ∈ R 3 , n ( x 1 ) denotes the face normal of a point x 1 in triangle soup T 1 , and ˆ x 1 is the projection of x 1 on T 2 .
Tables 1 and 2 in the main paper report quantitative evaluation of our method, compared to other baselines.
The meshing of the learned implicit representation was done using the M ARCHING
C UBES algorithm (Lorensen & Cline, 1987) on a uniform cubical grid of size [512] 3 .
Computing the evaluation metrics d C and d N is done on a uniform sample of 30 K points from the meshed surface.