Neural Importance Sampling

 

We propose to use deep neural networks for generating samples in Monte Carlo integration.

July 12, 2019
ACM Siggraph 2019

 

Authors

Thomas Müller (Disney Research/ETH Joint PhD)

Brian McWilliams (Disney Research)

Fabrice Rousselle (Disney Research)

Markus Gross (Disney Research/ETH Zurich)

Jan Novak (Disney Research)

 

Neural Importance Sampling

Abstract

Our work is based on non-linear independent components estimation (NICE), which we extend in numerous ways to improve performance and enable its application to integration problems. First, we introduce piecewise-polynomial coupling transforms that greatly increase the modeling power of individual coupling layers. Second, we propose to preprocess the inputs of neural networks using one-blob encoding, which stimulates localization of computation and improves inference. Third, we derive a gradient-descent-based optimization for the KL and the chi-square divergence for the specific application of Monte Carlo integration with unnormalized stochastic estimates of the target distribution. Our approach enables fast and accurate inference and efficient sample generation independently of the dimensionality of the integration domain. We show its benefits on generating natural images and in two applications to light-transport simulation: first, we demonstrate learning of joint path-sampling densities in the primary sample space and importance sampling of multi-dimensional path prefixes thereof. Second, we use our technique to extract conditional directional densities driven by the product of incident illumination and the BSDF in the rendering equation and we leverage the densities for path guiding. In all applications, our approach yields on-par or higher performance than competing techniques at equal sample count.

Copyright Notice