SpikingNeRF: Making Bio-inspired Neural Networks See through the Real World

TL;DR

SpikingNeRF introduces a spike-based, energy-efficient neural radiance field reconstruction method that aligns with neuromorphic hardware, achieving high-quality 3D rendering with significantly reduced energy consumption.

Contribution

The paper presents a novel spike-based NeRF model that incorporates temporal alignment and hardware-friendly strategies, enabling efficient 3D rendering on neuromorphic hardware.

Findings

Reduces energy consumption by approximately 70.79%

Achieves comparable quality to ANN-based NeRF methods

Demonstrates improved energy efficiency on neuromorphic hardware

Abstract

In this paper, we propose SpikingNeRF, which aligns the temporal dimension of spiking neural networks (SNNs) with the radiance rays, to seamlessly accommodate SNNs to the reconstruction of neural radiance fields (NeRF). Thus, the computation turns into a spike-based, multiplication-free manner, reducing energy consumption and making high-quality 3D rendering, for the first time, accessible to neuromorphic hardware. In SpikingNeRF, each sampled point on the ray is matched to a particular time step and represented in a hybrid manner where the voxel grids are maintained as well. Based on the voxel grids, sampled points are determined whether to be masked out for faster training and inference. However, this masking operation also incurs irregular temporal length, making it intractable for hardware processors, e.g., GPUs, to conduct parallel training. To address this problem, we develop the temporal padding strategy to tackle the masked samples to maintain regular temporal length, i.e., regular tensors, and further propose the temporal condensing strategy to form a denser data structure for hardware-friendly computation. Experiments on various datasets demonstrate that our method can reduce energy consumption by an average of 70.79\% and obtain comparable synthesis quality with the ANN baseline. Verification on the neuromorphic hardware accelerator also shows that SpikingNeRF can further benefit from neuromorphic computing over the ANN baselines on energy efficiency. Codes and the appendix are in \url{https://github.com/Ikarosy/SpikingNeRF-of-CASIA}.