Neural Product Importance Sampling
via Warp Composition
In ACM SIGGRAPH Asia 2024 (Conference Proceedings), December 2024
Our method composes a neural spline flow head warp with an emitter tail warp to achieve approximate product importance sampling of environment lighting with other terms (cosine and BRDF). Applied to cosine-weighted environment sampling on the Temple scene, we demonstrate significant variance reduction over multiple importance sampling (MIS) at equal rendering time (35 ms, 4 spp). We also visualize the conditional distribution learned by our model at the shading point marked in green. Our learned PDF closely matches the true (unshadowed) product. Our head warp does not have to learn the intricate details of the environment map already captured by the tail warp, and can be represented as a compact normalizing flow that can be baked for fast inference.
Abstract
Achieving high efficiency in modern photorealistic rendering hinges on using Monte Carlo sampling distributions that closely approximate the illumination integral estimated for every pixel. Samples are typically generated from a set of simple distributions, each targeting a different factor in the integrand, which are combined via multiple importance sampling. The resulting mixture distribution can be far from the actual product of all factors, leading to sub-optimal variance even for direct-illumination estimation. We present a learning-based method that uses normalizing flows to efficiently importance sample illumination product integrals, e.g., the product of environment lighting and material terms. Our sampler composes a flow head warp with an emitter tail warp. The small conditional head warp is represented by a neural spline flow, while the large unconditional tail is discretized per environment map and its evaluation is instant. If the conditioning is low-dimensional, the head warp can be also discretized to achieve even better performance. We demonstrate variance reduction over prior methods on a range of applications comprising complex geometry, materials and illumination.
Downloads
Publication
- Paper (author's version), PDF (13.2MB)
- arXiv version – external link
- Publisher's official version – external link, may require a subscription
Code
- Coming soon
BibTeX
Fast-forward Video
Cite
Joey Litalien, Miloš Hašan, Fujun Luan, Krishna Mullia, and Iliyan Georgiev. Neural Product Importance Sampling via Warp Composition. ACM SIGGRAPH Asia 2024 (Conference Proceedings), 1(1), Article 1, December 2024.
@inproceedings{Litalien:2024:Warp, title = {Neural Product Importance Sampling via Warp Composition}, author = {Joey Litalien and Miloš Hašan and Fujun Luan and Krishna Mullia and Iliyan Georgiev}, journal = {ACM SIGGRAPH Asia 2024 Conference Proceedings}, year = {2024}, month = dec, doi = {10.1145/3680528.3687566} }Copy to clipboard