Deep convolutional reconstruction for gradient-domain rendering

It has been shown that rendering in the gradient domain, i.e., estimating finite difference gradients of image intensity using correlated samples, and combining them with direct estimates of pixel intensities by solving a screened Poisson problem, often offers fundamental benefits over merely sampling pixel intensities. The reasons can be traced to the frequency content of the light transport integrand and its interplay with the gradient operator. However, while they often yield state of the art performance among algorithms that are based on Monte Carlo sampling alone, gradient-domain rendering algorithms have, until now, not generally been competitive with techniques that combine Monte Carlo sampling with post-hoc noise removal using sophisticated non-linear filtering. Drawing on the power of modern convolutional neural networks, we propose a novel reconstruction method for gradient-domain rendering. Our technique replaces the screened Poisson solver of previous gradient-domain techniques with a novel dense variant of the U-Net autoencoder, additionally taking auxiliary feature buffers as inputs. We optimize our network to minimize a perceptual image distance metric calibrated to the human visual system. Our results significantly improve the quality obtained from gradient-domain path tracing, allowing it to overtake state-of-the-art comparison techniques that denoise traditional Monte Carlo samplings. In particular, we observe that the correlated gradient samples --- that offer information about the smoothness of the integrand unavailable in standard Monte Carlo sampling --- notably improve image quality compared to an equally powerful neural model that does not make use of gradient samples.

[1]  Forrest N. Iandola,et al.  SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <1MB model size , 2016, ArXiv.

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

[3]  Tim Salimans,et al.  Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networks , 2016, NIPS.

[4]  Jean-Michel Morel,et al.  A Review of Image Denoising Algorithms, with a New One , 2005, Multiscale Model. Simul..

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

[6]  Pablo Bauszat,et al.  Gradient-domain path reusing , 2017, ACM Trans. Graph..

[7]  Li Fei-Fei,et al.  Perceptual Losses for Real-Time Style Transfer and Super-Resolution , 2016, ECCV.

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

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

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

[11]  Chi-Wing Fu,et al.  H-DenseUNet: Hybrid Densely Connected UNet for Liver and Tumor Segmentation From CT Volumes , 2018, IEEE Transactions on Medical Imaging.

[12]  Toshiya Hachisuka,et al.  Stochastic progressive photon mapping , 2009, ACM Trans. Graph..

[13]  Derek Nowrouzezahrai,et al.  Gradient‐Domain Photon Density Estimation , 2017, Comput. Graph. Forum.

[14]  Michael D. McCool,et al.  Anisotropic diffusion for Monte Carlo noise reduction , 1999, TOGS.

[15]  Jaakko Lehtinen,et al.  Gradient-Domain Bidirectional Path Tracing , 2015, EGSR.

[16]  Alexei A. Efros,et al.  The Unreasonable Effectiveness of Deep Features as a Perceptual Metric , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[17]  Jaakko Lehtinen,et al.  Gradient-domain path tracing , 2015, ACM Trans. Graph..

[18]  Matthias Zwicker,et al.  Regularizing Image Reconstruction for Gradient‐Domain Rendering with Feature Patches , 2016, Comput. Graph. Forum.

[19]  Bochang Moon,et al.  Feature Generation for Adaptive Gradient‐Domain Path Tracing , 2018, Comput. Graph. Forum.

[20]  Matthias Zwicker,et al.  Progressive photon beams , 2011, ACM Trans. Graph..

[21]  Mateu Sbert,et al.  Accelerating Path Tracing by Re-Using Paths , 2002, Rendering Techniques.

[22]  Matthias Zwicker,et al.  Adaptive rendering with non-local means filtering , 2012, ACM Trans. Graph..

[23]  Jaakko Lehtinen,et al.  Temporal gradient-domain path tracing , 2016, ACM Trans. Graph..

[24]  Jian Sun,et al.  Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification , 2015, 2015 IEEE International Conference on Computer Vision (ICCV).

[25]  Philipp Slusallek,et al.  Light transport simulation with vertex connection and merging , 2012, ACM Trans. Graph..

[26]  Yu-Bin Yang,et al.  Image Restoration Using Very Deep Convolutional Encoder-Decoder Networks with Symmetric Skip Connections , 2016, NIPS.

[27]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

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

[29]  Wojciech Jarosz,et al.  Beyond points and beams , 2017, ACM Trans. Graph..

[30]  Andrew L. Maas Rectifier Nonlinearities Improve Neural Network Acoustic Models , 2013 .

[31]  Matthias Zwicker,et al.  The Beam Radiance Estimate for Volumetric Photon Mapping , 2008, SIGGRAPH '08.

[32]  Derek Nowrouzezahrai,et al.  Gradient-domain volumetric photon density estimation , 2018, ACM Trans. Graph..

[33]  Jaakko Lehtinen,et al.  Noise2Noise: Learning Image Restoration without Clean Data , 2018, ICML.

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

[35]  Derek Nowrouzezahrai,et al.  Gradient-Domain Vertex Connection and Merging , 2017, EGSR.

[36]  Jaakko Lehtinen,et al.  Recent Advances in Adaptive Sampling and Reconstruction for Monte Carlo Rendering , 2015, Comput. Graph. Forum.

[37]  Yu-Bin Yang,et al.  Image Restoration Using Convolutional Auto-encoders with Symmetric Skip Connections , 2016, ArXiv.

[38]  Erik Reinhard,et al.  Photographic tone reproduction for digital images , 2002, ACM Trans. Graph..

[39]  Jan Novák,et al.  Image-space control variates for rendering , 2016, ACM Trans. Graph..

[40]  Leonidas J. Guibas,et al.  Optimally combining sampling techniques for Monte Carlo rendering , 1995, SIGGRAPH.

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

[42]  Damini Dey,et al.  Calcium removal from cardiac ct images using deep convolutional neural network , 2018, 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018).

[43]  Keiron O'Shea,et al.  An Introduction to Convolutional Neural Networks , 2015, ArXiv.

[44]  Kilian Q. Weinberger,et al.  Densely Connected Convolutional Networks , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[45]  Jaakko Lehtinen,et al.  Gradient-domain metropolis light transport , 2013, ACM Trans. Graph..

[46]  Matthias Zwicker,et al.  Adaptive sampling and reconstruction using greedy error minimization , 2011, ACM Trans. Graph..

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

[48]  Steven Guan,et al.  Fully Dense UNet for 2-D Sparse Photoacoustic Tomography Artifact Removal , 2018, IEEE Journal of Biomedical and Health Informatics.

[49]  James T. Kajiya,et al.  The rendering equation , 1986, SIGGRAPH.