GS-LRM: Large Reconstruction Model for 3D Gaussian Splatting

Kai Zhang,Sai Bi,Hao Tan,Yuanbo Xiangli,Nanxuan Zhao,Kalyan Sunkavalli,Zexiang Xu
2024-05-01
Abstract:We propose GS-LRM, a scalable large reconstruction model that can predict high-quality 3D Gaussian primitives from 2-4 posed sparse images in 0.23 seconds on single A100 GPU. Our model features a very simple transformer-based architecture; we patchify input posed images, pass the concatenated multi-view image tokens through a sequence of transformer blocks, and decode final per-pixel Gaussian parameters directly from these tokens for differentiable rendering. In contrast to previous LRMs that can only reconstruct objects, by predicting per-pixel Gaussians, GS-LRM naturally handles scenes with large variations in scale and complexity. We show that our model can work on both object and scene captures by training it on Objaverse and RealEstate10K respectively. In both scenarios, the models outperform state-of-the-art baselines by a wide margin. We also demonstrate applications of our model in downstream 3D generation tasks. Our project webpage is available at:
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper proposes a large-scale reconstruction model named GS-LRM for fast prediction of high-quality 3D Gaussian raw data from sparse images. Existing methods have limitations in dealing with complex and large-scale scenes, while GS-LRM solves these problems by adopting a transformer-based simple architecture and directly regressing the 3D Gaussian parameters for each pixel through multi-view image tokens. This method can adapt to scenes of different scales and complexities and outperforms the existing state-of-the-art baseline methods in object and scene reconstruction tasks.