CVPR 2022 · Neural Rendering

把一座城市拆成可以独立训练、按需渲染、平滑拼接的 NeRF

Block-NeRF 的主要贡献来自系统分解:每个局部模型只承担一个空间块;外观嵌入、曝光条件与位姿修正吸收跨月采集差异;可见性网络负责筛选;外观匹配和二维反距离加权完成跨块合成。

Matthew Tancik · Vincent Casser · Xinchen Yan · Sabeek Pradhan · Ben Mildenhall · Pratul P. Srinivasan · Jonathan T. Barron · Henrik Kretzschmar

UC Berkeley · Waymo · Google ResearchCVPR 2022 · 8248–8258arXiv:2202.05263DOI: 10.1109/CVPR52688.2022.0080715 页合订 PDF(主文 11 页 + 附录 4 页)
2,818,745Alamo Square 训练图像
35覆盖 960 m × 570 m 的 Block-NeRF
9–24 h每个块的训练时间
1–3每个目标视角通常实际渲染的块数
01 · RESEARCH GAP

扩展瓶颈同时出现在容量、时间、更新方式与数据一致性

把单个 NeRF 直接加宽只能暂时增加容量,渲染代价仍会随模型增大,模型最终无法放入单个设备,局部道路变化也会迫使整个场景重新训练。城市数据还跨越数月,天气、时间、曝光、车辆轨迹误差和动态物体会共同破坏静态场景假设。

单模型扩展失败

单个全局网络承担整座城市时,有限权重需要同时表示大量互不相关的局部细节;网络增大又使每条射线的查询成本上升。

跨时段采集不一致

同一地点在晴天、阴天、昼夜和不同曝光下呈现显著变化,若缺乏条件变量,网络容易把外观差异解释成错误密度。

局部更新困难

道路施工等变化只发生在局部。独立块允许替换受影响区域,同时保留其他块的训练结果。

Paper Verified论文以 Alamo Square 的 35 个块和 280 万张图像证明该系统能够覆盖整个街区;Table 2 进一步显示,在总权重固定时,增加块数仍提高 Mission Bay 的 PSNR、SSIM 与 LPIPS。
Author Claim作者称该分解可以扩展到任意大的环境。实验支持“场景规模与单帧渲染块数解耦”的系统性质,但没有在超过一个街区的连续城市尺度上测量调度、存储或长期维护成本。
原论文 Figure 1:Alamo Square 分块重建与局部更新
Figure 1 · 原论文城市场景重建与局部更新。左侧显示 Alamo Square 的空间分块,右侧展示跨月采集和施工区域的局部更新。图像直接取自论文 PDF。
02 · FORMULATION

每个块仍是 mip-NeRF;系统价值来自块间管理

输入是带标定相机射线、车辆位姿、曝光值和可移动物体掩码的 RGB 图像。每个 Block-NeRF 输出体密度、视角相关颜色以及一个辅助可见性值。最终输出是目标相机的 RGB 图像。

符号含义所在环节
\(\mathbf{r}(t)=\mathbf{o}+t\mathbf{d}\)相机原点 \(\mathbf{o}\) 沿单位方向 \(\mathbf{d}\) 发出的射线训练与渲染
\(\sigma_i,\mathbf{c}_i\)射线上第 \(i\) 个区间的密度与 RGB 颜色局部 NeRF
\(T_i,w_i\)到采样点前的累积透射率,以及该样本对像素的体渲染权重体渲染
\(\mathbf{a}_j\)第 \(j\) 张训练图像的 32 维外观嵌入颜色分支与块间外观匹配
\(e_j\)快门时间与模拟增益组合得到的标量曝光条件颜色分支
\(f, f_c, f_v\)密度与特征网络、颜色网络、辅助可见性网络每个 Block-NeRF
\(\mathbf{x}_k\)第 \(k\) 个块的中心或 origin候选选择与 IDW 合成

标准体渲染

\[\widehat{\mathbf{C}}(\mathbf{r})=\sum_{i=1}^{N}w_i\mathbf{c}_i,\qquad w_i=T_i\left(1-\exp(-\sigma_i\delta_i)\right)\]\[T_i=\exp\left(-\sum_{j\lt i}\sigma_j\delta_j\right),\qquad \delta_i=t_i-t_{i-1}\]
计算什么把沿射线的局部颜色按遮挡概率加权,得到一个像素。
为什么需要它将隐式密度场转换成可微的二维图像监督。
如何影响训练RGB 误差通过这些权重反向传播到密度和颜色 MLP。

mip-NeRF 的 Integrated Positional Encoding

\[\operatorname{IPE}(\boldsymbol{\mu},\boldsymbol{\Sigma})=\mathbb{E}_{\mathbf{X}\sim\mathcal{N}(\boldsymbol{\mu},\boldsymbol{\Sigma})}\!\left[\operatorname{PE}(\mathbf{X})\right]\]

一个像素对应的是圆锥台体积,论文沿用 mip-NeRF,用高斯 \(\mathcal{N}(\boldsymbol{\mu},\boldsymbol{\Sigma})\) 近似该体积,再对位置编码取期望。它减少车辆相机以不同距离观察街景时产生的 aliasing。

圆锥台位置IPE(x, Σ)f · 8 层密度 σ + 特征特征 + 方向 d外观 a + 曝光 ef_c · 3 层RGB 颜色f_v · 4 层可见性
Figure 3(结构重绘)。原论文 Figure 3 表明 \(f\) 仅由空间位置预测密度与特征;视线方向、曝光和图像外观嵌入只进入颜色分支 \(f_c\),从而避免让这些条件直接改写几何。\(f_v\) 独立预测训练可见性。

外观嵌入

每张训练图像拥有随机初始化并共同优化的 32 维向量。它吸收天气、时间与色彩平衡差异,并允许训练后在潜在空间插值。

证据边界:它没有提供物理可解释的光照分解,也没有保证不同块的同一向量对应相同外观。

曝光条件

相机曝光以 \(\operatorname{PE}(\text{shutter speed}\times\text{analog gain}/1000)\) 编码,使用 4 级正弦位置编码,只输入颜色分支。它既减少曝光差异造成的误差,也提供推理时可读的曝光控制。

分段位姿修正

每个 driving segment 学习平移偏移和 \(3\times3\) 残差旋转矩阵;残差加到单位矩阵后正交化。强正则在早期阻止网络用位姿偏移逃避几何学习。

瞬态物体掩码

预训练 Panoptic-DeepLab 识别车辆和行人,训练时忽略这些像素。阴影、漏检物体、树木和施工变化仍会留下伪影。

可见性监督

\[\mathcal{L}_{\mathrm{vis}}=10^{-6}\sum_i\left(f_v(\mathbf{x}_i,\mathbf{d})-\operatorname{stopgrad}(T_i)\right)^2\]

论文只明确说明 \(f_v\) 用 MSE 拟合 \(f\) 产生的透射率 \(T_i\),并将该损失缩放 \(10^{-6}\)。这里用 \(\operatorname{stopgrad}\) 明示辅助网络只读取监督目标,这是对实现关系的记号化解释;原文没有给出这一行完整公式。

03 · TRAINING

每个块完全独立训练,数据过滤发生在空间边界与动态掩码两处

典型放置策略在每个路口建立一个块,覆盖路口以及相连街道 75% 的长度,相邻块形成 50% 重叠。基本地图数据(例如 OpenStreetMap)足以自动安排块;训练图像按地理范围严格筛选。

01 · CAPTURE12 相机、10 Hz、360°,记录曝光并校准相机
02 · RAYS融合车辆位姿,计算公共坐标系中的逐像素射线,处理 rolling shutter
03 · FILTER按块的地理边界选图,语义分割掩蔽可移动物体
04 · OPTIMIZE联合优化 mip-NeRF、图像外观码、分段位姿偏移和可见性 MLP
05 · PARALLELIZE不同块无共享权重,可分布到独立设备同时训练

可复现训练设置

主网络 \(f\)8 层;Mission Bay 宽 512,其余实验宽 1024
颜色网络 \(f_c\)3 层,宽 128
可见性网络 \(f_v\)4 层,宽 128
外观向量32 维
优化器Adam
训练步数 / batch300k / 16,384 rays
学习率对数退火 \(2\times10^{-3}\rightarrow2\times10^{-5}\)
warmup前 1,024 步
coarse / fine 采样训练各 256;视频渲染各 512

计算与位姿正则

位姿初始化平移和残差旋转均为 0
位姿元素级 \(\ell_2\)权重从 \(10^5\) 线性降至 \(10^{-1}\),前 5,000 步完成
硬件每块 32 个 TPU v3 cores
论文报告总资源1,680 TFLOPS、512 GB 内存
每块训练时间9–24 小时
1200×900 单块渲染约 5.9 秒

论文未公开 Adam 的 \(\beta\) 值、weight decay、颜色损失的精确归一化、精度格式或完整训练代码。不得把 mip-NeRF 的默认值当作 Block-NeRF 已披露设置。

04 · INFERENCE

先剔除看不见目标区域的块,再渲染并在二维图像空间混合

推理时的目标相机位置先通过半径筛选得到附近块;低分辨率运行独立可见性网络,均值低于阈值的块被移除。通常剩下一到三个块。

半径候选Block A可见Block B低可见性丢弃 C渲染 A / B外观已匹配2D IDW合成
Figure 2(流程重绘)。原图展示基于距离和预测可见性的块选择,以及剩余块的颜色渲染与合成。可见性网络可以在低分辨率下单独运行,因此筛选开销低于完整颜色渲染。

二维 inverse distance weighting

\[\widetilde{w}_k=\operatorname{distance}(\mathbf{c},\mathbf{x}_k)^{-p},\qquad w_k=\frac{\widetilde{w}_k}{\sum_j\widetilde{w}_j},\qquad \mathbf{I}=\sum_k w_k\mathbf{I}_k\]

\(\mathbf{c}\) 是目标相机中心,\(\mathbf{x}_k\) 是块中心,\(\mathbf{I}_k\) 是第 \(k\) 个块完整渲染的二维图像。Alamo Square 使用 \(p=4\),Mission Bay 使用 \(p=1\)。权重在整幅图上相同,因此它牺牲部分锐度换取沿轨迹的时间稳定性。

外观匹配

  1. 在相邻块重叠区选择双方预测可见性都高的 3D 匹配位置。
  2. 固定两个块的全部网络权重以及作为目标的根块外观。
  3. 只优化待匹配块的外观码,使对应区域渲染之间的 \(\ell_2\) 误差下降。
  4. 约 100 步收敛,再从根块沿邻接关系向外传播;多个已匹配邻居会共同进入损失。

控制与停止条件

外观码确定天气、昼夜和色彩基调,曝光标量控制亮度。论文没有生成式 decoding、temperature 或采样重排;这是逐场景优化后的确定性体渲染。候选筛选依赖固定半径和均值可见性阈值,但原文没有披露阈值数值。

Our Interpretation外观匹配相当于在块邻接图上传播一个低频“视觉规范”。它解决色调和天气的一致性,却没有建立全局物理光照变量;传播路径和局部最优可能累积误差,论文没有报告闭环一致性实验。
05 · EXPERIMENTS

论文的强证据来自受控消融和固定总容量比较

Alamo Square 用于跨月外观与规模展示;单次采集的 Mission Bay 用于比较块数、容量和插值方式,以减少环境变化带来的混杂因素。

Alamo Square

  • 约 960 m × 570 m,2021 年 6–8 月采集。
  • 35 个块;每块覆盖 38–48 次采集,18–28 分钟驾驶。
  • 每块 64,575–108,216 张过滤后图像。
  • 全部 1,330 次采集、13.4 小时、2,818,745 张训练图像。

Mission Bay

  • 2020 年 11 月一次连续采集,环境条件较一致。
  • 100 秒、1.08 km、12 相机、12,000 张图像。
  • 用于 baseline、块大小、块布局、重叠比例与合成方法实验。
  • 官方 Waymo 页面公开这一数据集,TFRecord 含 RGB、曝光、逐像素射线和相机内参;验证集额外提供粗略动态物体掩码。

Table 1 · 组件消融(Alamo Square 单路口)

方法PSNR ↑SSIM ↑LPIPS ↓实际支持的结论
mip-NeRF17.860.5630.509没有跨图像条件时难以处理跨月变化
Block-NeRF − Appearance20.130.6110.458外观嵌入贡献最大,减少云雾状伪几何
Block-NeRF − Exposure23.550.6490.418定量影响很小,主要价值是推理时可控性
Block-NeRF − Pose Opt.23.050.6250.442位姿修正提高锐度并减少重复物体
Block-NeRF Full23.600.6490.417三项组合达到最佳总体表现

测试图像纵向分成两半:一半用于优化该测试图像的外观嵌入,另一半用于报告指标。这避免直接在被评测像素上拟合外观码,但评估仍是 test-time appearance fitting,不等同于完全无需适配的新视角预测。

Table 2 · 块数、容量与计算量(Mission Bay)

比较条件块数每块 / 总权重块大小相对计算PSNR ↑SSIM ↑LPIPS ↓
每块容量固定10.25M / 0.25M544 m1×23.830.8250.381
40.25M / 1.00M271 m2×25.550.8680.318
80.25M / 2.00M116 m2×26.590.8900.278
160.25M / 4.00M54 m2×27.400.9070.242
总容量固定约 1M11.00M / 1.00M544 m1×24.900.8520.340
40.25M / 1.00M271 m0.5×25.550.8680.318
80.13M / 1.00M116 m0.25×25.920.8750.306
160.07M / 1.00M54 m0.125×25.980.8770.305
Paper Verified固定总权重约 1M 时,从 1 块增加到 16 块,PSNR 由 24.90 提升至 25.98,SSIM 由 0.852 提升至 0.877,LPIPS 由 0.340 降至 0.305;论文报告的相对计算量同时由 1× 降至 0.125×。这直接支持空间局部化带来的容量利用效率。

Table 3 · 合成策略

插值时间一致PSNR ↑SSIM ↑LPIPS ↓判断
Nearest否26.400.8870.280开销最低,跨块突跳明显
IDW 2D是26.590.8900.278视频采用;指标与稳定性折中最佳
IDW 3D否26.570.8900.278深度噪声引入时间伪影
Pixelwise Visibility否27.390.9060.242静帧更锐利,视频闪烁
Imagewise Visibility否27.410.9070.242最佳静态指标,缺乏时间一致性

Table 4 · 8 块场景的重叠比例

重叠块大小PSNR ↑SSIM ↑LPIPS ↓
0%77 m26.770.8950.262
25%97 m26.750.8940.269
50%(主文设置)116 m26.590.8900.278
75%136 m26.510.8870.283
Evidence Boundary零重叠的静态指标最高,同时块大小也最小,因此该表没有隔离“重叠比例”与“空间范围”的影响。作者据此只谨慎推测指标变化可能来自块变小,并强调实际视频仍需要重叠来避免跨块时间伪影。

Table 5 · COLMAP 密集重建

方法PSNR* train ↑PSNR* test ↑协议限制
COLMAP point rendering13.01911.933忽略不可见像素
COLMAP surfel rendering13.29112.343Lambertian + 单光源;忽略不可见像素

COLMAP 将 Mission Bay 切成 8 个半径 97 m、约 25% 重叠的块;给定相机内参与位姿先验,动态物体特征被掩蔽,再运行 SfM、MVS 与点云融合。反射表面和天空造成稀疏或错误几何。该结果说明传统管线在此数据上的困难,但评估掩蔽规则与表示能力不同,不能看作严格同预算的端到端 baseline。

06 · REVIEWER

贡献坚实,但核心实验主要验证系统分解,没有完整刻画长期维护与动态场景

Strengths

  • 把城市级 NeRF 的容量、训练并行、渲染复杂度和局部更新统一到清晰的空间分解。
  • 固定总容量实验排除了“只是参数更多”的简单解释。
  • 针对真实车辆数据加入外观、曝光、位姿和动态掩码,工程问题覆盖完整。
  • 公开 Mission Bay 数据及逐像素射线,提升后续比较的可重复性。

Weaknesses

  • 完整训练代码与若干阈值、优化细节没有公开,复现仍需推断。
  • 块放置依赖道路结构和手工启发式,未比较学习式或负载均衡划分。
  • 外观传播没有全局闭环约束,也未量化跨块色差。
  • 动态物体通过丢弃监督处理,阴影、树木和施工仍会污染静态模型。
  • 每块使用 32 个 TPU v3 cores,计算与碳成本显著。

Potential Improvements

  • 用全局外观图优化或共享条件编码器,约束环路中的外观一致性。
  • 以可见性、纹理复杂度和计算预算联合学习块边界与重叠。
  • 建立静态背景块与可控对象场的组合表示,支持闭环驾驶模拟。
  • 用 hash grid、稀疏体素或缓存降低训练与渲染开销。
  • 把位姿、曝光和外观的不确定性显式建模,减少伪几何。
Research Extension

面向程序化材质生成的启发

可以把大型材质图或复杂 shader graph 按功能子图分块,让每个块独立训练或搜索,再用共享的接口变量与边界渲染约束组合。Block-NeRF 的外观匹配对应“局部程序的边界条件匹配”,可见性筛选对应“只执行影响当前输出的子程序”。这些属于跨领域研究设想,论文没有在材质图或程序生成上验证。

Reproducibility

论文与公开资源

官方 Waymo 页面提供 Mission Bay 数据集说明、云存储入口、视频和论文链接;没有提供官方训练代码。社区实现不能作为论文实现事实。论文明确给出网络宽度、训练步数、采样数、学习率、硬件、时间、位姿正则与 IDW 指数,足以重建主要框架,仍不足以保证逐数值复现。

社会影响

计算影响

作者明确承认将高计算量 NeRF 推到街区规模可能增加能耗与碳排放,并把缓存与快速 NeRF 视为缓解方向。

隐私影响

数据采集限于公共道路,并模糊人脸与车牌;更频繁、更大范围的城市扫描仍会放大隐私风险。相机影像可能包含比 LiDAR 更敏感的细节。

07 · SOURCES

一手来源与核查范围

  1. Tancik et al. CVPR 2022 Open Access 页面:正式会议信息、页码、PDF 与 supplement。
  2. Tancik et al. arXiv:2202.05263:作者版本与 15 页主文附录合订 PDF。
  3. Waymo Research. Block-NeRF 官方项目页:官方视频、CVPR 2022 标注与 Mission Bay 数据格式。
  4. Waymo. Waymo Block-NeRF Dataset:公开 Mission Bay TFRecord 数据入口。
  5. Mildenhall et al. NeRF;Barron et al. mip-NeRF:Block-NeRF 的体渲染和 integrated positional encoding 基础。

本页逐页核查本地 2202.05263 PDF 的正文与附录,重建 Table 1–5 全部行列。Figure 2 与 Figure 3 使用结构重绘并明确标注;Figure 6、Figure 7、Figure 8–11 的结论以原始图注和正文解释呈现。未公开参数均保持未确认状态。