Source-linked AI summary

Weakly supervised 3D Reconstruction with Adversarial Constraint

JunYoung Gwak, Christopher B. Choy, Animesh Garg, Manmohan Chandraker, Silvio Savarese

arXiv:1705.10904v2cs.CV

TL;DR

Aligned 3D annotations are expensive, and reconstructing volumes from silhouettes is ill-posed. McRecon uses foreground masks, raytrace pooling, and a learned manifold constraint from unlabeled shapes; it reports accuracy comparable to prior work with full 3D supervision.

  • Problem

    Deep-network 3D reconstruction commonly requires costly aligned CAD or point-cloud supervision, while silhouette-based volumetric reconstruction remains ill-posed.

    Source: §1 · p. 1
  • Method

    McRecon trains with foreground masks using raytrace pooling for reprojection and a learned log-barrier constraint that restricts reconstructions to a manifold of realistic unlabeled 3D shapes.

    Source: §1 · p. 2, §3 · p. 3
  • Results

    McRecon reports weak-supervision reconstruction accuracy comparable to prior works using full 3D supervision and strong generalization on single-view real images with noisy viewpoint estimates.

    Source: §5 · p. 8
  • Takeaways & Limitations

    The framework demonstrates high-quality reconstruction using inexpensive 2D supervision alongside unlabeled 3D shapes, including single-view real-image reconstruction.

    Source: §5 · p. 8

    Convergence properties are nontrivial to prove, though the authors report that the optimization behaves reasonably well in their experiments.

    Source: §3.4 · p. 5

Abstract

from arXiv · show

Supervised 3D reconstruction has witnessed a significant progress through the use of deep neural networks. However, this increase in performance requires large scale annotations of 2D/3D data. In this paper, we explore inexpensive 2D supervision as an alternative for expensive 3D CAD annotation. Specifically, we use foreground masks as weak supervision through a raytrace pooling layer that enables perspective projection and backpropagation. Additionally, since the 3D reconstruction from masks is an ill posed problem, we propose to constrain the 3D reconstruction to the manifold of unlabeled realistic 3D shapes that match mask observations. We demonstrate that learning a log-barrier solution to this constrained optimization problem resembles the GAN objective, enabling the use of existing tools for training GANs. We evaluate and analyze the manifold constrained reconstruction on various datasets for single and multi-view reconstruction of both synthetic and real images.

1. Introduction

The paper addresses the cost of aligned 3D supervision and the ambiguity of reconstructing shapes from silhouettes. It proposes weak mask supervision with a learned manifold constraint and raytrace pooling, reporting reconstruction quality comparable to prior work using full 3D supervision.

1. Introduction

  • Aligned CAD models or point clouds are costly to obtain, and labeling the closest model as ground truth can produce suboptimal training.Conventional CNN-based reconstruction commonly relies on large-scale supervision.Source: §1 · p. 1
  • Silhouette-based volumetric reconstruction is ill-posed: concavities may be crucial to object function but cannot be recovered from silhouettes.Dense viewpoint collection can also be difficult in practical settings such as online retail.Source: §1 · p. 1
  • The method constrains reconstructions to a manifold of realistic shapes, learned from hand-designed or scanned 3D shapes, to regularize the ill-posed problem.For example, it aims to recover a seatable chair with concavity from chair images.Source: §1 · p. 2
  • A learned log-barrier function enforces the manifold constraint and resembles a GAN discriminator, providing a rationale for adversarial training.The barrier is learned from unlabeled realistic 3D shapes.Source: §1 · p. 2
  • Raytrace pooling models reprojection error, mimics volumetric reconstruction such as voxel carving, and is described as avoiding aliasing compared with.After training, the network uses images at test time.Source: §1 · p. 2
  • Across ShapeNet, ObjectNet3D, and OnlineProduct, the proposed framework reports quantitative error reductions versus various baselines while requiring less supervision or relatively inexpensive weak supervision.Source: §1 · p. 2

2. Prior Work

Prior neural reconstruction methods commonly require associated 3D shapes for training, while silhouette-based methods use masks to form visual hulls. McRecon combines unmatched 3D shapes with inexpensive 2D supervision and introduces projection and recurrent components for its setting.

2. Prior Work

  • Neural voxel reconstruction methods including [15], [6], and [50] use associated 3D shapes for training, while uses silhouettes and viewpoints to form visual hulls.The cited approaches differ in whether they regress shape embeddings, reconstruct from multiple images, or use a learned latent space.Source: §2 · p. 2
  • McRecon combines unmatched 3D shapes with inexpensive 2D weak supervision to generate realistic shapes without explicit 3D supervision.The approach can learn with as little as one view of a 2D mask.Source: §2 · p. 2
  • Unlike, Raytrace Pooling is not limited to grid sampling; the paper compares it experimentally with that projection approach.The authors identify aliasing as a concern when sampling below the Nyquist rate.Source: §2 · p. 2
  • A recurrent neural network handles both single- and multi-view images, matching the paper's use of single- or multi-view weak supervision.Source: §2 · p. 2

3. Weakly supervised 3D Reconstruction with Adversarial Constraint

McRecon combines silhouette-based reprojection with a learned manifold constraint from unlabeled 3D shapes to address ill-posed single- or sparse-view reconstruction. Its log-barrier formulation connects the shape constraint to adversarial training, while raytrace pooling projects 3D voxels into 2D masks.

3. Weakly supervised 3D Reconstruction with Adversarial Constraint

  • McRecon constrains silhouette-based reconstruction to realistic 3D shapes, using a learned log barrier and raytrace pooling to support weak supervision.The framework uses silhouettes and approximate viewpoints alongside unlabeled 3D shapes.Source: §3 · pp. 2–3, §3.1 · p. 3

3.1. Log Barrier for Constrained Optimization

  • The constrained objective minimizes reprojection error while requiring the reconstruction to lie on the manifold of realistic shapes.The reprojection term compares predicted and observed masks using their associated camera viewpoints.Source: §3.1 · p. 3

3.2. Learning the Barrier for Manifold Constraint

  • A neural network learns the constraint-violation penalty in an adversarial setting, capturing geometric and semantic constraints that are difficult to hand-design.The authors frame the adversarial objective as learning a log barrier for the manifold constraint.Source: §3.2 · p. 3, §3.2.1 · p. 4

3.2.1 Penalty Functions and Discriminators

  • The learned log barrier is related to GAN discriminators, but the paper interprets it as a manifold constraint rather than a merged reconstruction discriminator.With overlapping reconstruction and real-shape distributions, the optimal penalty becomes less important as reconstructions become more realistic.Source: §3.2.1 · p. 4, §3.2.2 · p. 4

3.3. Raytrace Pooling for Reprojection Error

  • Raytrace pooling projects a voxel reconstruction into a mask by pooling features from voxels hit by each camera ray.The method uses max pooling in an occupancy grid; unlike sampling-based projection, its hit-test rendering avoids aliasing and depth-sampling-range issues.Source: §3.3 · pp. 4–5

3.4. McRecon Optimization

  • McRecon trains the reconstruction function and log barrier using image-mask-camera data and unlabeled 3D shapes.The paper notes that convergence properties are nontrivial to prove, although its empirical results indicate reasonable behavior in practice.Source: §3.4 · p. 5

4. Experiments

The experiments compare McRecon with baselines across synthetic and real-image reconstruction, including ablations, qualitative results, and representation analysis. McRecon benefits from unlabeled 3D shapes, particularly with weaker 2D supervision and more complex categories.

4. Experiments

  • Experiments span controlled ShapeNet renderings, single-view real images on ObjectNet3D, multi-view real images on OnlineProduct, and representation analysis.ShapeNet experiments vary supervision level, viewpoint count, and category; furniture and vehicle categories are trained separately.Source: §4 · p. 5, §4.3 · pp. 6–7, §4.4 · p. 8

4.1. Baselines

  • The baselines span 2D weak supervision, 2D plus unlabeled 3D shapes, and full 3D supervision, with the last serving as an approximate upper bound.The neural-network baselines share a base architecture to attribute performance differences to supervision mode.Source: §4.1 · p. 5

4.2. Metrics and Visualization

  • Average Precision evaluates occupancy confidence, while Intersection-over-Union uses binarized occupancy at threshold 0.4.Visualization colors indicate occupancy probability, with voxels below 0.1 omitted.Source: §4.2 · p. 6

4.3. Ablation Study on ShapeNet [5]

  • McRecon outperforms simple nearest-neighbor retrieval by a large margin, with larger gains when fewer 2D silhouettes supervise training and on complex categories.More supervision improves reconstruction; the adversarial-network benefit is smaller for simpler shapes such as cars.Source: §4.3 · p. 7
  • McRecon carves concavities that are difficult to learn from masks alone, but its occupancy probabilities tend to become more binary.The authors suggest binarization may be an artifact of the generator fooling the discriminator.Source: Figure 5 · p. 6, §4.3 · p. 7

4.4. Single-view reconst. on ObjectNet3D [52]

  • McRecon’s single-view ObjectNet3D results indicate stronger generalization than silhouette-based learning, including with noisy viewpoint estimates.Voxel carving requires camera parameters at test time, unlike McRecon.Source: §4.4 · p. 8, Table 3 · p. 8

4.5. Multi-view Reconst. on OnlineProduct [42]

  • On OnlineProduct, the network integrates information across real-image views and reconstructs reasonable 3D shapes.The network was trained on ShapeNet with random PASCAL backgrounds to improve robustness to background noise.Source: Figure 7 · p. 7, §4.5 · p. 8

4.6. Representation analysis

  • Interpolating the recurrent representation produces smooth transitions in shape properties, and semantic vectors can modify another image’s reconstruction.Examples include changing wing length or a chair-back hole.Source: §4.6 · p. 8

5. Conclusion

McRecon reconstructs 3D shapes from foreground masks by constraining outputs to unlabeled real-shape space and using raytrace pooling. The paper reports high-quality weakly supervised reconstructions comparable in accuracy to prior full-3D-supervision work, with generalization to single-view real images under noisy viewpoint estimation.

5. Conclusion

  • McRecon achieves reconstruction accuracy comparable to prior works using full 3D supervision, while using weak 2D supervision.The conclusion also reports strong generalization for single-view real-image reconstruction with noisy viewpoint estimation.Source: §5 · p. 8
  • The framework constrains reconstructions to the space of unlabeled real 3D shapes and uses foreground masks with a raytrace pooling layer.The layer bridges the representation gap between 2D masks and 3D volumes.Source: §5 · p. 8
  • The conclusion identifies noisy-viewpoint single-view real-image reconstruction as a case of strong generalization and potential practical utility.The conclusion phrases practical utility as a suggestion rather than a demonstrated deployment outcome.Source: §5 · p. 8

S.1. Implementation Details

McRecon combines an encoder, generator, and discriminator to map image inputs to voxel reconstructions, with raytrace pooling for silhouette projection.

Network

  • The encoder and generator reconstruct volumes from 2D images, while the discriminator imposes the manifold constraint under weak 2D supervision.The encoder-generator follows the Deep-Residual-GRU network proposed by Choy et al. [6].Source: §S.1 · p. 11

Encoder

  • A 3D convolutional LSTM aggregates encoded multi-view features into a spatial latent representation that resolves viewpoints and self-occlusion.Each image is encoded into a 1024-dimensional feature vector; the latent variable has dimensions 4 × 4 × 4 × 128.Source: §S.1 · p. 11

Generator and discriminator

  • The generator produces a 32×32×32 voxel occupancy map, which raytrace pooling projects into a silhouette for the projection loss.The discriminator receives either a reconstruction or an unlabeled shape and outputs a scalar.Source: §S.1 · p. 11

Implementation

  • The implementation used a symbolic-math neural-network library, and the authors stated that code and models would be publicly released upon publication.Source: §S.1 · p. 11

Optimization:

The optimization section describes training measures for the learned barrier and reports empirical behavior, while noting that convergence properties are difficult to prove.

Optimization

  • Training the barrier for weakly supervised reconstruction with a manifold constraint faces challenges observed in prior work, including possible divergence when p and q have nonoverlapping supports during discriminator log q/p computation.The authors use instance noise to smooth the probability space, making the support of p infinite.Source: Optimization · p. 11
  • Instance noise smooths the probability space to prevent support mismatch from causing divergence when computing log q/p.Here p is the distribution of unlabeled shapes and q is the distribution of reconstructions.Source: Optimization · p. 11
  • The discriminator is trained only when its prediction error exceeds 20%, and the learning rates differ for the reconstruction and discriminator functions.The reported rates are 10^-2 for θ_f and 10^-4 for θ_g; training runs for 40,000 iterations with batch size 8.Source: Optimization · p. 11

S.2. Baseline Methods

The baselines span silhouette-based carving, mask-supervised projection networks, retrieval using unlabeled shapes, and fully 3D-supervised reconstruction.

  • Voxel Carving removes voxels outside projected silhouettes and requires camera parameters and masks, unlike learning-based methods that take images at test time.Source: §S.2 · p. 11
  • Raytrace Pooling trains an encoder-generator with mask reprojection supervision but no discriminator to provide gradients toward the 3D shape manifold.Source: §S.2 · p. 11
  • Perspective Transformer samples a 3D grid over a fixed depth range, requiring range and density hyperparameters, whereas Raytrace Pooling uses accurate raytracing.Source: §S.2 · p. 11
  • RP + NN retrieves the nearest unlabeled shape to an RP prediction, improving accuracy when a similar shape exists and the prediction is accurate.Source: §S.2 · p. 11
  • Full 3D Supervision uses the same network without a discriminator and serves as an upper-bound reference because 2D projections provide partial shape information.Source: §S.2 · pp. 11–12

S.3. Single-view reconst. on IKEA dataset[26]

The IKEA evaluation compares single-view reconstructions trained on rendered furniture against real-image 3D models using an aligned evaluation protocol.

  • A single network was trained on ShapeNet furniture renderings composited with random PASCAL backgrounds.Source: §S.3 · p. 12
  • Evaluation follows [15] [50], aligning ground-truth models over permutations, flips, and translations of up to 10%.Source: §S.3 · p. 12
  • Table S1 reports per-class average precision percentages on IKEA; the other baselines use full 3D supervision as an upper-bound reference.Source: Table S1 · p. 12

S.4. Multi-view synthetic images reconstruction

Supplemental ShapeNet examples show both successful and less-successful reconstructions, with exotic shapes identified as a particular challenge.

  • McRecon generally reconstructs reasonable 3D shapes from a small number of silhouettes and viewpoints.Source: §S.4 · p. 12
  • Figure S2 presents both successful and less-successful qualitative reconstructions on ShapeNet.Source: §S.4 · p. 12
  • Exotic shapes may be difficult to reconstruct when they are absent from the unlabeled shape repository used as the discriminator's target manifold.Source: §S.4 · p. 12

S.5. Single real image reconstruction

Supplemental results show reasonable reconstructions from single masks and multi-view real images, alongside semantically expressive latent representations. At test time, McRecon has the same computation time as RP because the discriminator is not evaluated.

S.5. Single real image reconstruction

  • On ObjectNet3D, McRecon reconstructs reasonable 3D shapes from single-mask supervision.Source: §S.5 · p. 12

S.6. Multi-view real image reconstruction

  • On Stanford Online Product images, multi-view reconstructions are reasonable despite the train–test domain difference, which may reduce quality relative to other experiments.Source: §S.6 · p. 12

S.7. Representation analysis

S.8. Computation Time

  • At test time, McRecon and RP have the same computation time because the manifold-projecting discriminator is not evaluated; timing used a Titan X, batch size 8, and 5 views.Source: §S.8 · p. 12
Loading 1705.10904v2…