We deeply appreciate the insightful reviews and discussions throughout the review period. We will revise our paper to incorporate your valuable feedback. Below, we provide summaries of the reviews, our responses, and discussions.
Our primary objective is to reconcile two seemingly contrasting goals: efficient inference and expansive capacity in an implicit neural representation (INR). Unlike other research on mixtures of INRs, our approach constructs mixtures within the INR's weight space. The computational cost of INR is dictated by the size of inference parameters, not by the total size of learnable parameters. But the sample quality is in favor of the size of the learnable parameter. Ensembling within the weight space maintains the size of the inference parameter compact while allowing us to easily increase learnable parameter size by adding more mixtures. This design allows our model to handle tremendous coordinate inputs with comparable visual quality.
We evaluate the efficiency of our proposed method against INR baselines based on constructive feedback from reviewers fjUK and JMK5. We observe that our approach demonstrates a high degree of efficiency. Especially in NeRF scenes with a significant number of input coordinates, our method achieves 199 times fewer FLOPS, 49 times faster fps, and 22 times reduced memory than Functa during single view inference. Detailed metrics can be found in Table A of the attached file.
SRN Cars | Ours | Functa
|--------------------------------|---------|-----------|
Inference speed (fps) | 97.7 | 2.0
Inference cost (TFLOPS) | 0.009 | 1.789
Inference memory (GB) | 1.26 | 28.01
Reconstruction PSNR | 25.9 | 24.2
Generation FID | 79.5 | 80.3
Reviewers fjUK and uPSW YZcJ raised questions regarding our choice of baselines. We have selected GEM, GASP, and Functa for this purpose. Our model and selected baselines focus on algorithms for generating neural fields, especially those algorithms handling various modalities such as 2D images, 3D voxels, and NeRF scenes. Our model exploits a two-step training process. Initially, we perform context adaptation on our INR model to obtain context vectors for the training set. Subsequently, we train a latent diffusion model for sampling context vectors. The reconstruction performance pertains to the first step, while the generation performance is influenced by both steps. As reviewer YZcJ noted, increasing the latent space's representation size improves reconstruction performance. But, generation performance may require a different optimal representation size.
Upon request by reviewer YZcJ, we delved deeper into our model's internal states. Our observations indicate that information for each sample is evenly distributed across all mixtures within the hidden layers except the input and output layers. The main paper's Figure 5 showcases the diversity of learned weights in each mixture, implying that each mixture's neural implicit function contributes equally to representing a training set sample. Figure A(a) in the attached file further substantiates our model's distributed nature. As the number of mixtures increases, the quality of the reconstructed image steadily improves. It also explains that the information is distributed information over mixtures. Figure A (b) offers a visual representation of the impact of adding or subtracting coefficients from each layer. We can observe that information is predominantly located in the hidden layer, excluding the input and output layers.
In Figure B of the attached file, we visualize the inter-class interpolations requested by reviewer 2vix. We plot three interpolations with four different classes correspondingly. (a) chair (top left), cabinet (top right), display (bottom left), sofa (bottom right) (b) loudspeaker, sofa, airplane, bench (c) airplane, bench, car, rifle. Figure B (c) shows smooth interpolation even with distinct shapes.
We have one more day for discussions related to this work. We invite you to review our findings and share any further queries or concerns.
Thank you for your time and feedback.