RegNeRF [Kor]

Niemeyer et al. / RegNeRF - Regularizing Neural Radiance Fields for View Synthesis from Sparse Inputs / CVPR 2022

1. Problem definition

Neural Radiance Fields(NeRF) 태스크는 하나의 장면(scene)에 대한 다양한 시점(viewing direction)의 이미지들을 입력으로 neural network 모델을 학습하고, 이를 활용하여 입력 이미지들에는 없는 새로운 시점(novel views)에서 봤을 때의 모습을 담은 2D 이미지를 생성하는 것을 목표로 합니다.

여기서 radiance fields란, 3차원 상에서 위치 좌표인 $\mathbf{x} = (x,y,z)$ 와 시점 $\mathbf{d} = (\theta, \phi)$ 를 색상 $\mathbb{c} = (R, G, B)$ 과 density인 $\sigma \in [0,\infty)$ 로 매핑하는 함수를 의미합니다. 풀어서 설명하자면, 실제 세상에서의 장면 중 어떠한 위치(Figure 1.a 불도저의 삽 부분)를 특정 시점에서 봤을 때, 2D 이미지의 픽셀 상에서 색상은 어떤 색상이며 density(간단히 이해하면 배경이 아니고 물체일수록 높은 값을 가짐)는 어느 정도인지 예측하는 함수라고 볼 수 있습니다. 따라서 이와 같은 함수를 neural network로 모델링한 NeRF 모델을 잘 학습한다면 원하는 입력을 넣음으로써 학습 데이터에 없는 시점에서 봤을 때의 2D 이미지를 만들어낼 수 있을 것입니다.

이때, 모델의 출력인 $(\mathbb{c}, \mathbb{\sigma})$ 를 2D 이미지 픽셀의 색상으로 변환하는 과정을 volume rendering이라고 합니다. 직관적으로 이해하기위해 사과를 보는 것을 예시로 든다면, 사과의 속이 꽉 차 있으니 사과에 해당하는 모든 부분의 density값은 높겠지만 내가 보고있는 색상은 사과의 표면인 빨간색이 결정하는 것이지 속 부분인 하얀색은 관여하지 않을 것입니다. NeRF 태스크에서 volume rendering은 이와 같은 상황을 모델링한 함수이고, 해당 과정을 통해 픽셀의 색상을 rendering하게 됩니다. NeRF의 학습은 이와 같은 과정으로 rendering된 픽셀이 같은 위치의 ground truth 픽셀과 같아지도록 함으로써 이뤄지게 됩니다.

2. Motivation

당연하게도 하나의 scene에 대한 이미지가 많을 수록 novel view에서의 이미지를 예측하는 것은 쉬워질 것입니다. 하지만 RegNeRF의 저자는 실제 세상의 application(AR, VR, 자율주행, 로보틱스 등)을 고려했을 때 다양한 뷰의 이미지를 많이 가지는 것은 쉽지 않은데 반해, 기존 NeRF 모델들은 일정 수준 이상의 성능을 내려면 지나치게 많은 이미지를 필요로 한다고 주장합니다. (RegNeRF 저자의 영상에서는 30~100장이라고 언급됨) 따라서 sparse한 입력(RegNeRF는 최소 3장을 가정함)에서도 높은 수준의 novel view synthesis를 할 수 있어야한다는 motivation이 생기게 됩니다.

앞서 언급했듯이 일부 기존 모델(NeRF, mip-NeRF)들은 입력이 sparse할 경우 성능이 급격히 하락하게 됩니다. (Figure 2. 참고) 이와 같은 상황을 극복하기 위해 수행된 연구들은 크게 두가지 갈래가 있습니다.

다양한 시점을 가지고 있는 많은 scene으로 구성된 큰 데이터셋으로 pre-training
depth 정보 같은 supervision을 활용하여 모델에 regularization 적용

1번과 같은 방향의 연구로는 PixelNeRF, Steredo Radiance Fields, MVSNeRF 등이 있습니다. 이와 같은 방향은 pre-training에 활용할 데이터를 수집하는 것과 이를 학습하는 데에 많은 비용을 필요로 한다는 점과, 새로 학습해야할 scene이 pre-training 단계에서 학습한 domain과 다를 경우 일반화가 되지 않는다는 단점이 존재합니다.

DS-NeRF와 DietNeRF와 같은 모델은 2번에 해당하는데, 이들은 pre-training 단계 없이 처음부터 학습해도 sparse한 입력 상황에 대처가 가능합니다. 하지만 이들이 학습되기 위해 필요한 supervision들은 늘 주어지는 것이 아니고, 만일 주어지지 않는다면 이를 위해 별도의 레이블링이 필요하기 때문에 마찬가지로 비용을 초래하게 됩니다.

Idea

저자는 별도의 supervision 없이, 주어지는 sparse한 입력만을 가지고 있는 상황에서도 좋은 성능을 보이는 NeRF 모델을 학습하기위한 regularization 기법을 제안합니다. RegNeRF 모델은 rendering된 이미지의 patch에 대해 geometry와 appearance에 대한 regularization과, 학습 초기의 divergent behavior를 대처하기 위한 space annealing strategy가 적용된 NeRF 모델입니다. 새롭게 제안된 기법들은 주어진 입력 외의 외부 정보나 별도의 pre-training 과정이 필요 없기 때문에 RegNeRF는 앞서 언급된 단점을 가지지 않고 sparse한 입력 상황에 대처가 가능합니다.

3. Method

mip-NeRF

RegNeRF는 기본적으로 Mip-NeRF 모델을 기반으로 하고 있습니다. Mip-NeRF는 하나의 픽셀이 rendering됨에 있어서 직선 형태의 ray만을 활용하는 NeRF와 달리, 원뿔 형태의 모델링을 활용해 하나의 점이 아닌 일정한 범위 내 영역의 영향을 받게 설계된 모델입니다. 이를 통해 여러 ray를 활용하는 효과를 효율적으로 얻을 수 있음과 더불어 coarse와 fine으로 나누어져 있는 NeRF의 network 구조를 하나의 network로 통합하여 더 간단한 구조를 가지게 되었습니다. 또한 카메라와 물체가 거리가 멀어질 때 계단 현상을 보이는 NeRF 고유의 artifacts를 없앨 수 있었습니다. 저자는 이렇듯 NeRF보다 장점이 많은 mip-NeRF 모델 구조를 기반으로 제안하는 regularization 기법을 적용하게 됩니다.

Patch-based Regularization

위 그림에서 보여지듯 RegNeRF는 입력에 없는 보여지지 않은 시점(unobserved view)에 대해 rendering된 작은 patch(8x8 size)에 대해 두가지 regularization을 적용해 학습하는 과정을 가지고 있습니다.

첫번째로 대부분의 경우 물체의 굴곡에 큰 변화가 적을 가능성이 높은 데에 착안하여, geometry regularization으로서 depth smoothness loss가 적용되게 됩니다.

$R_r$ 은 보여지지 않은 시점에서 샘플링된 ray의 집합, $S_{patch}$ 는 rendering된 이미지 patch의 사이즈, $r_{ij}$ 는 r이 중심이 되는 patch의 $(i,j)$ 를 통과하는 ray를 의미하며 $\hat d_\theta$ 는 해당 ray에 대한 depth 값입니다. 위 loss의 의미를 해석해보면 보여지지 않은 시점에 대해 만들어진 이미지 patch의 depth가 서로 인접한 픽셀끼리 비슷하도록 하는 강제하는 효과를 가지게 됩니다. 이는 흔히 total variation loss라고 불리는 형태와 유사하며, 위 loss를 통해 모델로 하여금 생성해내는 이미지의 depth가 smooth하게 되도록 만드는 효과를 얻을 수 있습니다.

두번째로는 생성될 이미지의 색상이 실제 세상 이미지의 색상과 비슷하도록 만들어 자연스러운 이미지를 만들 수 있도록 만드는 color regularization loss가 적용됩니다.

여기서 $\phi$ 는 JFT-300M 데이터셋으로부터 8x8 사이즈의 patch를 만들어 학습시킨 normalizing flow model(RealNVP) 모델입니다. Normalizing flow model은 loss function으로 negative log likelihood(NLL)를 사용하는 이미지 생성 모델입니다. 즉, 실제 세상의 이미지로 구성된 JFT-300M 데이터셋으로 학습된 RealNVP는 실제 이미지와 가까운 입력에 대해선 높은 값, 실제와 거리가 먼 입력에 대해선 낮은 값을 출력하게 될 것입니다. 이를 활용하여 보여지지 않은 시점에서 만들어진 이미지 patch의 color $\hat c_\theta$ 가 normalizing flow model $\phi$ 의 출력이 최대화 되도록 모델을 학습시키게 된다면, 생성될 이미지가 자연스러운 color 값을 가지게 만드는 효과를 얻을 수 있을 것입니다. 이 지점에서 한가지 의문이 들었던 것은 비록 multi-view까지 요구하지는 않지만, 다량의 데이터와 normalizing flow model에 대한 pre-training 과정을 거쳐야 color regularization을 사용할 수 있다는 점에서 RegNeRF가 마냥 외부의 것에 의존하지 않는 모델은 아니라는 생각이 들었습니다.

최종적으로 이와 같은 두가지의 regularization과 기존 NeRF 학습에 사용되는 reconstruction loss $L_{MSE}$ 를 함께 사용한 아래의 loss function을 활용하여 학습하는 방법을 제안하게 됩니다.

Sample Space Annealing

저자는 sparse 입력 세팅에서 학습이 쉽지 않은 원인이 학습 초기에 발산하는 것이라고 지적합니다.

따라서 초반에 발산하는 문제를 해결하기 위해 위와 같은 방식으로 sample space를 조절하는 방법을 제시합니다. $t_n$ 과 $t_f$ 는 ray위 point의 sampling이 이루어지는 범위와 관련하여 각각 가까운 평면과 먼 평면을 의미하며, $t_m$ 은 이 두 사이의 중앙을 뜻합니다. $i$ 는 학습 시의 interation, $p_s$ 와 $N_t$ 는 각각 시작 범위와 full range에 도달할 때까지의 interation 횟수를 의미하는 하이퍼파라미터입니다. 위와 같은 학습 진행 중에 ray 위 point를 sampling하는 범위를 업데이트하는 sample space annealing을 적용하게 되면 학습이 진행됨에 따라 sampling 범위가 점점 늘어가게 되어, 학습 초기 의미없는 부분에 대한 학습이 줄어 발산하지 않을 수 있다고 저자는 말합니다.

RegNeRF는 위와 같은 두가지 방식의 regularization기법과 학습 중에 sample space를 조절하는 전략을 통해 NeRF모델이 적은 input에 대해서 원활히 학습될 수 있도록 만들었고, 그 방법이 직관적이면서 간단하고 다른 모델에도 적용될 수 있는 확장성을 가지고 있다는 장점을 지니고 있습니다.

4. Experiment & Result

Experimental setup

Sparse한 입력 상황에 대한 성능 측정을 위해 모든 실험들은 3/6/9 장의 입력을 기준으로 수행되었고, 구체적으로는 아래와 같은 세팅으로 실험이 이루어졌습니다.

- Dataset

DTU: 테이블 위에 놓여진 물체를 촬영한 scene으로 구성.
LLFF: DTU 데이터셋보다 조금 더 복잡한 환경의 scene으로 구성.

- Evaluation Metrics

PSNR
SSIM
LPIPS
위와 같이 2D 이미지 reconstruction 세팅에서 사용되는 지표들이 활용됨.
모델 간의 쉬운 비교를 위해 본 논문에선 $10^{-\text{PSNR}/10}$ , $\sqrt{1-\text{SSIM}}$ , LPIPS의 평균 또한 표기.

- Baselines

PixelNeRF
Stereo Radiance Fields (SRF)
MVSNeRF
Mip-NeRF
DietNeRF
"ft"는 fine-tuning을 의미

Result

저자는 우선 제안한 regularizatio기법의 데이터 효율성 효과를 확인하기 위해 Mip-NeRF와의 비교 실험을 진행합니다.

위 결과에서 알 수 있듯이 같은 PSNR을 달성하는 데에 필요한 입력 이미지의 개수가 Mip-NeRF보다 55% 더 적은 것을 확인할 수 있습니다.

다음은 baseline들과의 정량평가 결과입니다.

색상의 진함 정도로로 순위가 표시되어 있으며, 두 데이터셋에 대해 모두 종합적으로 RegNeRF의 결과가 상위권임을 확인할 수 있습니다. 또한 LLFF 데이터셋에 대한 결과에서 DTU로 pre-trained된 모델들의 성능이 DTU 데이터셋에서 보다 비교적 낮아진 것을 통해 pre-training을 활용하여 sparse input 세팅을 해결하려고 한 기존 연구들이 domain shift problem을 가지고 있다는 것을 보여줍니다.

다음 결과는 논문에서 제안된 Scene Space Annealing, Geometry Regularization, Appearance Regularization에 대한 ableation study 결과입니다. 제안된 모든 기법들이 효과가 있는 것을 확인했으며, 가장 영향이 큰 것은 Scene Space Annealing 인 것으로 확인되었습니다. 한편, appearance regularization은 전체 성능에 비교적 낮은 영향을 끼치는 것을 볼 수 있는데 실제로 저자가 공개한 코드에서 pre-trained된 RealNVP가 제공되지 않지만 여전히 좋은 성능을 보인다는 내용이 있는 것까지 고려해보면 해당 모듈의 효과는 향후 조금 더 검증될 필요가 있지 않는가라는 생각이 들었습니다.

마지막으로 LLFF 데이터를 이용한 생성 결과입니다. 확대되어 있는 부분에서 볼 수 있듯이, 적은 입력에도 불구하고 RegNeRF가 다른 모델에 비해 디테일하게 생성되었음을 확인할 수 있습니다.

5. Conclusion

저자는 sparse한 입력 상황에서의 NeRF 모델이 잘 학습하기 위한 regularization 방법과 sampling space annealing 제안했습니다. 제안한 방법을 이용해 학습된 모델은 기존 baseline들에 비해 정량, 정성적으로 우수한 성능을 보였습니다. 대량의 multi-view 데이터와 pre-training 혹은 별도의 supervision 없이 명확한 원인 분석과 더불어 적절하고 간단한 해결책인 patch-base의 regularization과 학습 전략으로 이와 같은 결과를 낸 것이 인상깊은 논문이었습니다.

Take home message (오늘의 교훈)

문제 해결은 명확한 원인 분석부터 시작해야한다.
문제의 해결책이 반드시 외부의 도움을 받거나 복잡할 필요는 없다.

Author / Reviewer information

Author

구교정 (Gyojung Gu)

KAIST AI (M.S)
E-mail: gyojung.gu@kaist.ac.kr

Reviewer

Korean name (English name): Affiliation / Contact information
Korean name (English name): Affiliation / Contact information
...

Reference & Additional materials

NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, Mildenhall et al, ECCV 2020(oral)
Mip-NeRF: A Multiscale Representation for Anti-Aliasing Neural Radiance Fields, Barron et al, ICCV 2021
Density estimation using real NVP, Dinh et al, ICLR 2017