Neural network system with temporal feedback for denoising of rendered sequences
A neural network-based rendering technique increases temporal stability and image fidelity of low sample count path tracing by optimizing a distribution of samples for rendering each image in a sequence. A sample predictor neural network learns spatio-temporal sampling strategies such as placing more samples in dis-occluded regions and tracking specular highlights. Temporal feedback enables a denoiser neural network to boost the effective input sample count and increases temporal stability. The initial uniform sampling step typically present in adaptive sampling algorithms is not needed. The sample predictor and denoiser operate at interactive rates to achieve significantly improved image quality and temporal stability compared with conventional adaptive sampling techniques.
1. A computer-implemented method, comprising:
receiving guide data for a rendered image frame in a sequence of rendered image frames, the sequence including a previous rendered image frame and the rendered image frame;
receiving external state including a reconstructed previous rendered image frame with fewer artifacts compared with the previous rendered image frame, wherein the external state is warped, using difference data corresponding to changes between the previous rendered image frame and the rendered image frame, to produce warped external state; and
processing the guide data for the rendered image frame using layers of a neural network model to produce a sample map that indicates a number of samples to be computed for each pixel in the rendered image frame, wherein the warped external state replaces hidden state generated by one or more of the layers of the neural network.
2. The computer-implemented method of claim 1 , further comprising processing the guide data for the previous rendered image frame using layers of the neural network model to produce a first sample map that indicates a number of samples to be computed for each pixel in the previous rendered image frame, wherein an initial warped external state replaces the hidden state generated by the one or more layers of the neural network.
3. The computer-implemented method of claim 1 , wherein the guide data comprises normal vectors, depth, or albedo.
4. The computer-implemented method of claim 1 , further comprising rendering the rendered image frame according to the sample map.
5. The computer-implemented method of claim 4 , wherein the rendered image frame includes artifacts including at least a loss of high-frequency detail or residual noise.
6. The computer-implemented method of claim 1 , further comprising processing the previous rendered image frame using layers of a reconstruction neural network model to produce the external state.
7. The computer-implemented method of claim 6 , further comprising adjusting parameters of the neural network model and the reconstruction neural network model based on differences between the reconstructed rendered image frame and the rendered image frame without artifacts.
8. The computer-implemented method of claim 1 , further comprising processing the rendered image frame using layers of a reconstruction neural network model to produce second external state including a reconstructed rendered image frame with fewer artifacts compared with the rendered image frame.
9. The computer-implemented method of claim 1 , further comprising:
scaling the rendered image frame to produce scaled image frames;
applying predicted hierarchical spatially-varying filter kernels to the scaled image frames to produce filtered image frames; and
combining the filtered image frames to produce a multi-scale filtered image frame.
10. The computer-implemented method of claim 9 , further comprising:
applying a predicted temporal kernel to the warped external state to produce filtered warped external state; and
summing the multi-scale filtered image frame and the filtered warped external state to produce the reconstructed rendered image frame.
11. A system, comprising:
a processing unit configured as a neural network model to:
receive guide data for a rendered image frame in a sequence of rendered image frames, the sequence including a previous rendered image frame and the rendered image frame;
receive external state including a reconstructed previous rendered image frame with fewer artifacts compared with the previous rendered image frame, wherein the external state is warped, using difference data corresponding to changes between the previous rendered image frame and the rendered image frame, to produce warped external state; and
process the guide data for the rendered image frame using layers of a neural network model to produce a sample map that indicates a number of samples to be computed for each pixel in the rendered image frame, wherein the warped external state replaces hidden state generated by one or more of the layers of the neural network.
12. The system of claim 11 , further comprising processing the guide data for the previous rendered image frame using layers of the neural network model to produce a first sample map that indicates a number of samples to be computed for each pixel in the previous rendered image frame, wherein an initial warped external state replaces the hidden state generated by the one or more layers of the neural network.
13. The system of claim 11 , wherein the guide data comprises normal vectors, depth, or albedo.
14. The system of claim 11 , further comprising a renderer that is coupled to the neural network model and configured to render the rendered image frame according to the sample map.
15. The system of claim 14 , wherein the rendered image frame includes artifacts including at least a loss of high-frequency detail or residual noise.
16. The system of claim 11 , further comprising a reconstruction neural network model configured to process the previous rendered image frame using layers of the reconstruction neural network model to produce the external state.
17. The system of claim 11 , further comprising a reconstruction neural network model configured to process the rendered image frame using layers of the reconstruction neural network model to produce second external state including a reconstructed rendered image frame with fewer artifacts compared with the rendered image frame.
18. The system of claim 11 , wherein a reconstruction neural network model is configured to:
scale the rendered image frame to produce scaled image frames;
apply predicted hierarchical spatially-varying filter kernels to the scaled image frames to produce filtered image frames; and
combine the filtered image frames to produce a multi-scale filtered image frame.
19. A non-transitory, computer-readable storage medium storing instructions that, when executed by a processing unit, cause the processing unit to:
receive guide data for a rendered image frame in a sequence of rendered image frames, the sequence including a previous rendered image frame and the rendered image frame;
receive external state including a reconstructed previous rendered image frame with fewer artifacts compared with the previous rendered image frame, wherein the external state is warped, using difference data corresponding to changes between the previous rendered image frame and the rendered image frame, to produce warped external state; and
process the guide data for the rendered image frame using layers of a neural network model to produce a sample map that indicates a number of samples to be computed for each pixel in the rendered image frame, wherein the warped external state replaces hidden state generated by one or more of the layers of the neural network.
20. The non-transitory, computer-readable storage medium of claim 19 , further comprising instructions that cause the processing unit to process the guide data for the previous rendered image frame using layers of the neural network model to produce a first sample map that indicates a number of samples to be computed for each pixel in the previous rendered image frame, wherein an initial warped external state replaces the hidden state generated by the one or more layers of the neural network.