Journal Article
Knowledge Graphs
Featured

Instant neural graphics primitives with a multiresolution hash encoding

Thomas Müller, Alex Evans(Nvidia (United Kingdom)), Christoph Schied, Alexander Keller
July 1, 2022ACM Transactions on Graphics6,062 citations

6.1k

Citations

787

Influential Citations

ACM Transactions on Graphics

Venue

2022

Year

Abstract

Neural graphics primitives, parameterized by fully connected neural networks, can be costly to train and evaluate. We reduce this cost with a versatile new input encoding that permits the use of a smaller network without sacrificing quality, thus significantly reducing the number of floating point and memory access operations: a small neural network is augmented by a multiresolution hash table of trainable feature vectors whose values are optimized through stochastic gradient descent. The multiresolution structure allows the network to disambiguate hash collisions, making for a simple architecture that is trivial to parallelize on modern GPUs. We leverage this parallelism by implementing the whole system using fully-fused CUDA kernels with a focus on minimizing wasted bandwidth and compute operations. We achieve a combined speedup of several orders of magnitude, enabling training of high-quality neural graphics primitives in a matter of seconds, and rendering in tens of milliseconds at a resolution of 1920×1080.

Analysis

Why This Paper Matters

This paper addresses a critical bottleneck in neural graphics: the high computational cost of training and evaluating fully connected neural networks for graphics primitives. By introducing a multiresolution hash encoding, the authors enable a dramatic reduction in network size and computational requirements without sacrificing quality. This breakthrough makes neural graphics primitives practical for real-time applications, such as interactive rendering and virtual reality, where speed is paramount.

The significance extends beyond graphics to any domain that uses coordinate-based neural representations, including scientific visualization, medical imaging, and robotics. The hash encoding approach is simple, parallelizable, and GPU-friendly, making it accessible to practitioners and likely to be widely adopted.

Technical Contributions

  • Multiresolution Hash Encoding: A novel input encoding that uses a hash table of trainable feature vectors at multiple resolutions. This allows the network to disambiguate hash collisions and capture fine details efficiently.
  • Smaller Network: The encoding enables the use of a much smaller fully connected network (e.g., 2 hidden layers of 64 neurons) compared to prior work, reducing floating point and memory access operations.
  • Fully-Fused CUDA Kernels: The entire system is implemented with custom CUDA kernels that fuse operations, minimizing bandwidth waste and maximizing GPU utilization.
  • Parallelism: The multiresolution structure is trivial to parallelize on modern GPUs, enabling efficient training and inference.

Results

The paper reports a combined speedup of several orders of magnitude over previous neural graphics primitives. Training high-quality primitives (e.g., neural radiance fields, signed distance functions) takes seconds instead of hours. Rendering at 1920×1080 resolution is achieved in tens of milliseconds, enabling real-time performance. The method achieves state-of-the-art quality on standard benchmarks while being significantly faster.

Significance

This work has broad implications for the AI field, particularly in neural rendering and implicit neural representations. It demonstrates that careful input encoding and GPU optimization can overcome the computational barriers that previously limited neural graphics to offline use. The hash encoding technique is likely to inspire similar approaches in other domains where coordinate-based networks are used, such as physics simulation and generative modeling. By making neural graphics primitives practical, this paper accelerates the convergence of neural networks and computer graphics, opening new possibilities for interactive and immersive experiences.