ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
6.1k
Citations
787
Influential Citations
ACM Transactions on Graphics
Venue
2022
Year
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.
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.
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.
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.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba