神经辐射场（Neural Radiance Field，NeRF）的基本理论和原理 ｜ NeRF

hjy9725
2,340 阅读5分钟

系列文章目录

下面是关于 "NeRF的基本理论和原理" 的复习大纲：

@TOC

概述

1. Neural Radiance Field (NeRF) 神经辐射场：NeRF是一种基于神经网络的场景表示方法，通过学习场景中每个点的辐射和颜色信息，从而能够在任意视角下合成逼真的图像。它通过在三维空间中对点进行采样，并为每个点预测辐射和颜色，从而构建场景的隐式表示。

2. 隐式表示和视角合成：传统的三维场景表示方法通常使用显式几何模型，如网格或体素。而NeRF采用隐式表示，不需要显式地存储几何信息，而是通过神经网络预测场景属性。这使得NeRF能够实现从不同视角合成图像，即视角合成。

3. NeRF的应用领域：NeRF在多个领域具有广泛的应用，如机器人技术中的场景建模、城市建图用于创建逼真的城市模型、自主导航中的环境感知、虚拟现实和增强现实中的逼真场景渲染等。

4. 可微分体素渲染：NeRF的训练涉及到对场景中的体素进行渲染，以比较预测的辐射和颜色与真实图像之间的差异。这个渲染过程是可微分的，意味着可以使用反向传播进行优化，从而有效地训练NeRF模型。

基本理论和原理

1. Neural Radiance Fields (NeRF) 的背景和目的
NeRF 最初是由 Mildenhall 等人在 2020 年提出的，用于新颖的视图合成。由于能够高度真实地合成复杂场景的视图，NeRF 在该领域引起了广泛关注。

2. NeRF 的基本形式
基本上，NeRF 模型将三维场景表示为由神经网络近似的辐射场。这个辐射场为场景中的每个点和每个观察方向描述颜色和体积密度。

3. 数学公式的描述
这可以写为:

F(x,θ,ϕ)→(c,σ)F(\mathbf{x},\theta,\phi)\to(\mathbf{c},\sigma)

其中 x=(x,y,z)\mathbf{x}=(x,y,z) 是场景中的坐标，(θ,ϕ)(\theta,\phi) 表示方位角和极角，c=(r,g,b)\mathbf{c}=(r,g,b) 代表颜色，σ\sigma 代表体积密度。这个 5D 函数通常由一个或多个多层感知器 (MLP) 近似，有时被表示为 FΘF_{\Theta} 。

4. 视图方向的描述
两个观察角 (θ,ϕ)(\theta,\phi) 通常由 d=(dx,dy,dz)\mathbf{d}=(d_x,d_y,d_z) 表示，这是一个 3D 笛卡尔单位向量。

5. 神经网络的表示
这种神经网络表示通过限制 σ\sigma 的预测，即场景的内容，与观察方向无关来实现多视图的一致性。然而，颜色 c\mathbf{c} 被允许同时依赖于观察方向和场景坐标。

6. NeRF 模型的实现
在 NeRF 中通过设计 MLP 为两个阶段来实现基准模型的。第一阶段输入为 x\mathbf{x}，并输出 σ\sigma 和一个高维特征向量。在第二阶段，特征向量与观察方向 d\mathbf{d} 连接，并传递给另一个 MLP，该 MLP 输出 c\mathbf{c}。值得注意的是，Mildenhall 等人认为 σ\sigma MLP 和 c\mathbf{c} MLP 是同一个神经网络的两个分支，但后续的大多数研究者认为它们是两个独立的 MLP 网络。

7. 新视图合成的过程
使用经过训练的 NeRF 模型进行新的视图合成大致如下：

  • 对于合成图像中的每个像素：通过场景发送相机光线，并生成一组采样点。这在图 1 的 (a) 中展示。
  • 对于每个采样点：使用观察方向和采样位置使用 NeRF MLP 计算局部颜色和密度。这在图 1 的 (b) 中展示。
  • 使用体积渲染：使用这些颜色和密度生成图像。这在图 1 的 (c) 中展示。

图片.png

8. 体积渲染的数学描述
考虑到要渲染的场景的体积密度和颜色函数，使用体积渲染来获得任何相机光线的颜色 C(r)C(\mathbf{r})，其中相机光线 r(t)=o+td\mathbf{r}(t)=\mathbf{o}+t\mathbf{d} 有相机位置 o\mathbf{o} 和观察方向 d\mathbf{d}。公式描述为：

C(r)=∫t1t2T(t)⋅σ(r(t))⋅c(r(t),d)⋅dt,C(\mathbf{r})=\int_{t_1}^{t_2}T(t)\cdot\sigma(\mathbf{r}(t))\cdot\mathbf{c}(\mathbf{r}(t),\mathbf{d})\cdot dt,

这里，σ(r(t))\sigma(\mathbf{r}(t)) 和 c(r(t),d)\mathbf{c}(\mathbf{r}(t),\mathbf{d}) 分别表示沿着观察方向 d\mathbf{d} 的相机光线上点 r(t)\mathbf{r}(t) 的体积密度和颜色，而 dtdt 表示光线在每个积分步骤上行进的微小距离。

T(t)T(t) 是累积的透射率，表示光线从 t1t_1 到 tt 行进而不被拦截的概率，由以下公式给出：

T(t)=exp⁡(−∫t1tσ(r(u))⋅du).T(t)=\exp(-\int_{t_1}^t\sigma(\mathbf{r}(u))\cdot du).

9. 新视图的渲染过程
新的视图是通过追踪相机光线 C(r)C(r) 经过要合成的图像的每个像素来渲染的。这个积分可以通过数值方法计算。

数值采样方法：原始实现和大多数后续方法使用了一种非确定性的分层采样方法，其中光线被分成 NN 个等间隔的区间，并从每个区间均匀地取一个样本。然后，上述方程可以近似为：

C^(r)=∑i=1NαiTici\hat{C}(\mathbf{r})=\sum_{i=1}^N\alpha_iT_i\mathbf{c}_i

其中，

Ti=exp⁡(−∑j=1i−1σjδj)T_i=\exp(-\sum_{j=1}^{i-1}\sigma_j\delta_j)

δi\delta_i 是从样本 ii 到样本 i+1i + 1 的距离。(σi,ci)(\sigma_i, c_i) 是沿着给定光线的样本点 ii 计算的密度和颜色，由 NeRF MLP(s) 计算得出。αi\alpha_i 在样本点 ii 处的透明度/不透明度，由以下公式给出：

αi=1−exp⁡(σiδi)\alpha_i=1-\exp(\sigma_i\delta_i)

这种方法的关键在于正确地估计每个像素的颜色，同时考虑到场景中的物体和其他因素。

10. 计算光线的期望深度
首先介绍通过积累的透射率来完成计算光线的期望深度。公式为:

d(r)=∫t1t2T(t)⋅σ(r(t))⋅t⋅dt.d(\mathbf{r})=\int_{t_1}^{t_2}T(t)\cdot\sigma(\mathbf{r}(t))\cdot t\cdot dt.

公式被近似为：

D^(r)=∑i=1NαitiTi\hat{D}(\mathbf{r})=\sum_{i=1}^N\alpha_it_iT_i

这个公式基于前面的公式近似了深度的计算。

某些深度正规化方法使用期望深度来限制场景表面上的密度为delta-like函数，或者强制深度平滑性。

对于每个像素，使用平方误差光度损失来优化 MLP 参数。在整个图像上，该损失由以下公式给出:

L=∑r∈R∣∣C^(r)−Cgt(r)∣∣22L=\sum_{r\in R}||\hat{C}(\mathbf{r})-C_{gt}(\mathbf{r})||_2^2

这里的 Cgt(r)C_{gt}(\mathbf{r}) 是与 rr 相关的训练图像像素的真实颜色，而 RR 是与待合成图像相关的光线的批次。

11. NeRF 模型的位置编码
NeRF 模型经常使用位置编码，这在渲染视图中改进了细节重建。在原始实现中，对场景坐标 x\text{x} 和观察方向单位向量 d\mathbf{d} 的每个组件应用了以下位置编码:

γ(v)=(sin(20πv),cos(20πv),sin(21πv),cos(21πv),...,sin(2N−1πv),cos(2N−1πv)),\gamma(v)=(sin(2^0\pi v),cos(2^0\pi v),sin(2^1\pi v),cos(2^1\pi v),...,sin(2^{N-1}\pi v),cos(2^{N-1}\pi v)),

其中 NN 是用户确定的编码维度参数，在原始论文中，x\text{x} 设置为 N=10N=10 , d\mathbf{d} 设置为 N=4N=4 。但是，最近的研究已经通过位置编码的替代形式进行了实验并获得了很好的结果，包括可训练的参数化、积分和层次化变体。

avatar
文章
阅读
粉丝