Accelerating Neural Radiance Field by network pruning

Deng Yu Dong
DOI: https://doi.org/10.54254/2755-2721/45/20241047
2024-03-15
Abstract:Novel view synthesis, a crucial problem in computer graphics with extensive applications, has gained significant attention due to its increasing demand in fields such as autonomous driving, virtual reality, film, and the visual arts. Recent advancements in machine learning have led to progress in generating photo-realistic imagery, particularly with the Neural Radiance Fields (NeRF) method. Despite its impressive results, NeRF requires high computational resources for training and synthesis, impeding its real-world applications. This work aims to investigate the potential of network pruning to reduce NeRFs computational complexity without compromising the quality of the generated images, addressing the limitations of the original model and paving the way for more efficient view synthesis techniques. We propose a novel pruning method tailored for NeRF, Coarse-Fine-Pruning (CFP). We note that the NeRF framework is divided into two parts: a coarse network that learns the global structure of the scene and a fine network that learns the local details. Crucially, our experiments show that the coarse and fine networks have different tolerance to pruning. As such, our method adapts the pruning rate to each network, achieving a higher compression rate without compromising the quality of the generated images. We evaluate CFP and show that it can reduce the number of parameters by 90% while maintaining the same level of image quality.
What problem does this paper attempt to address?