A detail preserving neural network model for Monte Carlo denoising

Monte Carlo based methods such as path tracing are widely used in movie production. To achieve low noise, they require many samples per pixel, resulting in long rendering time. To reduce the cost, one solution is Monte Carlo denoising, which renders the image with fewer samples per pixel (as little as 128) and then denoises the resulting image. Many Monte Carlo denoising methods rely on deep learning: they use convolutional neural networks to learn the relationship between noisy images and reference images, using auxiliary features such as position and normal together with image color as inputs. The network predicts kernels which are then applied to the noisy input. These methods show powerful denoising ability, but tend to lose geometric or lighting details and to blur sharp features during denoising. In this paper, we solve this issue by proposing a novel network structure, a new input feature—light transport covariance from path space—and an improved loss function. Our network separates feature buffers from the color buffer to enhance detail effects. The features are extracted separately and then integrated into a shallow kernel predictor. Our loss function considers perceptual loss, which also improves detail preservation. In addition, we use a light transport covariance feature in path space as one of the features, which helps to preserve illumination details. Our method denoises Monte Carlo path traced images while preserving details much better than previous methods.

[1]  Soheil Darabi,et al.  On filtering the noise from the random parameters in Monte Carlo rendering , 2012, TOGS.

[2]  Pradeep Sen,et al.  A machine learning approach for filtering Monte Carlo noise , 2015, ACM Trans. Graph..

[3]  Olga Sorkine-Hornung,et al.  Path‐space Motion Estimation and Decomposition for Robust Animation Filtering , 2015, Comput. Graph. Forum.

[4]  Bochang Moon,et al.  Adaptive Rendering Based on Weighted Local Regression , 2014, ACM Trans. Graph..

[5]  Frédo Durand,et al.  A frequency analysis of light transport , 2005, SIGGRAPH '05.

[6]  Martín Abadi,et al.  TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems , 2016, ArXiv.

[7]  Jaakko Lehtinen,et al.  Sample-based Monte Carlo denoising using a kernel-splatting network , 2019, ACM Trans. Graph..

[8]  Andrew Zisserman,et al.  Two-Stream Convolutional Networks for Action Recognition in Videos , 2014, NIPS.

[9]  Xuanqin Mou,et al.  Low-Dose CT Image Denoising Using a Generative Adversarial Network With Wasserstein Distance and Perceptual Loss , 2017, IEEE Transactions on Medical Imaging.

[10]  Matthias Zwicker,et al.  Robust Denoising using Feature and Color Information , 2013, Comput. Graph. Forum.

[11]  Matthias Zwicker,et al.  Denoising your Monte Carlo renders: recent advances in image-space adaptive sampling and reconstruction , 2015, SIGGRAPH Courses.

[12]  Toshiya Hachisuka,et al.  Robust Image Denoising Using a Virtual Flash Image for Monte Carlo Ray Tracing , 2013, Comput. Graph. Forum.

[13]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[14]  Tamy Boubekeur,et al.  Bayesian Collaborative Denoising for Monte Carlo Rendering , 2017, Comput. Graph. Forum.

[15]  Yoshua Bengio,et al.  Understanding the difficulty of training deep feedforward neural networks , 2010, AISTATS.

[16]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[17]  Frédo Durand,et al.  5D Covariance tracing for efficient defocus and motion blur , 2013, TOGS.

[18]  Mark Meyer,et al.  Denoising with kernel prediction and asymmetric loss functions , 2018, ACM Trans. Graph..

[19]  Kenny Mitchell,et al.  Nonlinearly Weighted First‐order Regression for Denoising Monte Carlo Renderings , 2016, Comput. Graph. Forum.

[20]  Timo Aila,et al.  Interactive reconstruction of Monte Carlo image sequences using a recurrent denoising autoencoder , 2017, ACM Trans. Graph..

[21]  Mark Meyer,et al.  Kernel-predicting convolutional networks for denoising Monte Carlo renderings , 2017, ACM Trans. Graph..

[22]  Steven McDonagh,et al.  Adaptive polynomial rendering , 2016, ACM Trans. Graph..

[23]  Qiang Zhang,et al.  DEMC: A Deep Dual-Encoder Network for Denoising Monte Carlo Rendering , 2019, Journal of Computer Science and Technology.

[24]  Luca Fascione,et al.  The path tracing revolution in the movie industry , 2015, SIGGRAPH Courses.

[25]  Beibei Wang,et al.  Fast Computation of Single Scattering in Participating Media with Refractive Boundaries Using Frequency Analysis , 2020, IEEE Transactions on Visualization and Computer Graphics.

[26]  Cyril Soler,et al.  A Local Frequency Analysis of Light Scattering and Absorption , 2014, ACM Trans. Graph..