【三维分割】LangSplat: 3D Language Gaussian Splatting（CVPR 2024 highlight）

论文：https://arxiv.org/pdf/2312.16084
代码：https://github.com/minghanqin/LangSplat

文章目录

一、3D language field
二、回顾 Language Fields的挑战
三、使用SAM学习层次结构语义
四、Language Fields 的 3DGS
五、开放词汇查询（Open-vocabulary Querying）
六、实验
代码实现

一、3D language field

早期构建三维特征场的尝试包括蒸馏特征场[20]和神经特征融合场[43]。他们通过跨多个视图将LSeg [21]或DINO [4]特征提炼为一个NeRF，学习了3D一致特征。Shen等人[39]通过将clip征提取成NeRF，进一步提取特征场进行few-shot 语言引导的自动操作。[Panoptic lifting for 3d scene understanding with neural fields. CVPR 2023][In-place scene labelling and understanding with implicit scene representation. ICCV 2021]将语义信息嵌入到NeRF中，例如，Semantic NeRF [54]在NeRF中联合编码外观和几何语义，用于新的语义视图合成。LERF [18] 是第一个将CLIP特性嵌入到NeRF中的公司，它利用强大的CLIP表示方式实现了开放词汇表的3D查询。DINO特性也被用于监督LERF，以提高其性能。Liu等人[Weakly supervised 3d open-vocabulary segmentation.NeurIPS 2023]还利用CLIP和DINO特征训练NeRF模型用于3D开放词汇分割。

二、回顾 Language Fields的挑战

输入图像 $I∈R^{3×H×W}$ ；取一组校准图像{ $I_t|t=1,2,...,T$ }作为输入，用其训练一个3D Language Fields $Φ$ 。大多数现有的方法使用CLIP编码器 $V$ 提取图像特征，监督3D语言字段 $Φ$ ，利用CLIP对齐的文本图像潜在空间，促进开放词汇表查询。然而， CLIP嵌入是由图像对齐的，而不是由像素对齐的 。即 $V(I_t)∈R^D$ 是一个图像级的特征，而我们需要一个像素对齐的语言嵌入 $L_t∈R^{D×H×W}$ 。同时，像素对齐的语言特征具有点模糊性 ，对象上的单个点有助于影响区域的多个语义层次（例如，猫耳朵上的一个点，同时指向猫的耳朵、猫头和整个猫，应该被三种类型的文本查询激活）

为了解决问题，现有方法[18,24]从裁剪后的图像patch中提取出CLIP特征的层次结构，即对坐标为 $v = (w, h)$ 的像素，在不同物理尺度下，以v为中心patch中获得相应的CLIP特征，patch可以完全包含对象。多尺度patch方法有两个局限性。首先，patch特征是不精确的，因为它们经常包含额外的上下文对象信息，导致过度平滑的语言域，对象边界模糊。为了缓解不稳定的问题，大多数方法[18,24]利用额外的像素对齐的DINO特性来监督网络。但是，学习到的三维语言特征仍然不精确，如图1所示。其次，推理需要在多个尺度上同时渲染，才能找到最优尺度。随着尺度的数量可能高达30 [18]，大大降低了推理速度。

除了渲染目标，三维建模方法的选择也很重要。大多数现有的方法[2,43]都使用NeRF来进行3D表示，它们在每个3D点上学习一个语言特征，然后将该语言特征渲染到图像上，类似于彩色渲染。然而，基于NeRF的方法受到其耗时的渲染过程的限制；同时，在实际应用中，特别是在智能机器人等领域，对高效开放词汇查询的需求也很高。

三、使用SAM学习层次结构语义

在这里插入图片描述

SAM 可以准确地将一个像素与周围属于同一对象的像素进行分组，从而将图像分割成多个边界清晰的对象掩模。利用SAM，我们可以捕获三维场景中对象的语义层次，为每个输入图像提供准确和多尺度的分割map。

具体来说，使用32×32的规则网格进行point 提示，获得三个不同语义级别 mask： $M_0^s、M_0^p、M_0^w$ ，分别代表子部分、部分和整个物体。然后根据IoU评分、stability评分和mask之间的重叠率，去除集合的冗余mask。过滤后的mask 集合为： $M^s、M^p、M^w$ ，将场景划分为语义上有意义的区域，用于提取区域CLIP特征。在数学上，所获得的像素对齐的语言嵌入为（ $M^l (v)$ 表示在语义级别 $l$ 上，像素 $v$ 属于的mask）：

在这里插入图片描述

从3D语言场景中渲染的每个像素，现在都拥有一个与精确的语义上下文相一致的CLIP特性。这种对齐减少了模糊性，并提高了基于语言的查询的准确性。我们可以学习一个准确的三维语言场，即使没有常用的DINO正则化。我们基于SAMs的方法的另一个优点是预定义的 语义尺度。由于我们对“整体”、“部分”和“子部分”级别有不同的分割映射，因此我们可以在这些预定义的尺度上直接查询三维语言字段。这就消除了跨多个绝对尺度进行密集搜索的需要，从而使查询过程更加高效。

四、Language Fields 的 3DGS

在获得了一组二维图像{ $L^l_t|t=1,...,T$ }上的语言嵌入后，我们可以通过建模三维点和二维像素之间的关系来学习三维语言场景。3DGS明确地将三维场景表示为各向异性三维高斯分布的集合，每个高斯 $G (x)$ 由均值 $µ∈R^3$ 和协方差矩阵Σ参数化：

在这里插入图片描述

其中 $c_i$ 是第 $i$ 个高斯值的颜色，N表示tile中的高斯值， $C (v)$ 是像素v处的渲染颜色， $α_i = o_iG_i^{2D}(·)$ 。 $o_i$ 是第 $i$ 个高斯的不透明度， $G_i^{2D}(·)$ 表示投影到二维上的第i个高斯的函数。

本文提出了3D language GS，它用三个language embedding { $f^s,f^p,f^w$ }来增强每个3DGS。这些嵌入来自于CLIP特征。增强的高斯函数被称为三维语言高斯，采用基于tile的光栅化器来保持渲染效率：

在这里插入图片描述

$F^l(v)$ 表示语义级别 $l$ 在像素 $v$ 处渲染的语言嵌入。通过将语言信息直接合并到3DGS中，3DLanguage Fields 能够响应基于语言的查询。

由于CLIP嵌入是高维特征，直接在CLIP潜在空间上学习 $f^l$ 显著增加了内存和时间成本。与学习没有球谐系数的RGB颜色相比，学习512维CLIP特征使存储3D高斯数据的内存需求增加了35倍以上，很容易导致L1 cache内存耗尽。为了降低内存成本和提高效率，我们 引入了一种场景级别的 language autoencoder，将场景中的CLIP嵌入映射到一个较低维的潜在空间，从而减少了内存需求。CLIP模型使用4亿对（图像、文本）对进行训练，它的d维潜在空间可能非常紧凑，因为它需要在这个空间中对齐任意的文本和图像。然而，Language Fields $Φ$ 是特定于场景的，可以利用场景先验来压缩CLIP特征。事实上，对于每个输入图像，我们将得到数百个被SAM mask，明显小于CLIP训练中使用的图像数量。因此，一个场景中所有的mask都稀疏地分布在CLIP潜在空间中，允许我们使用一个场景特定的 Autoencoder进一步压缩这些CLIP特征

编码器 $E$ 将 $D$ 维CLIP特征 $L^l_t(v)∈R^D$ 映射到 $H_t^l(v) = E(L^l_t (v))∈R^D$ ，其中d≪D。解码器 $Ψ$ 从压缩表示重构原始CLIP嵌入。Autoencoder 在CLIP嵌入上重建目标为 ( $d_{ae}$ 表示距离函数，具体采用 $l_1$ 和余弦距离损失)：

在这里插入图片描述

让Language Gaussian 在特定场景的latent space 中学习 language embedding，而不是CLIP潜在空间，有 $f^l∈R^d$ 。实验取d = 3（模型效率和准确性的均衡）。与直接建模D维CLIP嵌入相比，我们的方法通过合并场景先验显著降低了内存成本。语言嵌入的优化目标：

在这里插入图片描述
推理遵循等式(4)，将language embedding 从3D渲染到2D，并使用场景特定解码器 $Ψ$ 恢复CLIP图像嵌入 $Ψ(F_t^l)∈R^{D×H×W}$ ，这允许使用CLIP文本编码器进行开放词汇表查询。

五、开放词汇查询（Open-vocabulary Querying）

由于CLIP模型提供的图像和文本之间的对齐空间，学到的3D language field 支持开放词汇表3D查询，包括开放词汇表3D对象定位和开放词汇表3D语义分割。许多现有的开放词汇表三维语义分割方法[24]通常从一个类别列表中选择类别，其中包括图像中出现的类别。然而，获得一个全面的野外场景类别列表是具有挑战性的。与它们不同的是，我们的方法在给定任意文本查询时生成精确的对象mask。

按照LERF计算language embeddinig $ϕ_{img}$ 与每个文本查询 $ϕ_{qry}$ 的相关性得分(其中， $ϕ^i_{canon}$ 是从“object”, “things”, “stuff”,
和“texture”中选择的预定义规范短语的CLIP嵌入)：

在这里插入图片描述
每个文本查询可以得到三个相关性映射，表示在特定的语义级别上的结果。遵循在LERF 的策略，选择产生最高相关性得分的语义level。对于三维对象定位任务，我们直接选择相关性得分最高的点。对于三维语义分割任务，我们过滤出相关性分数低于所选阈值的点，并预测剩余区域的对象mask。

六、实验

数据集。LERF数据集[18]是使用iPhone的app多摄像头捕获的，由复杂的野外场景组成，为三维对象定位任务设计的，这里我们通过注释文本查询的地面真相掩码来扩展LERF数据集，允许在LERF数据集上评估开放词汇表的三维语义分割。由于用于三维对象定位的原始LERF注释相对简单，因此在某些场景下的性能已经接近饱和。因此，我们进一步手动注释了其他具有挑战性的定位样本，以更好地评估方法的性能。我们报告了在LERF [18]后的三维对象定位任务的定位精度，并报告了三维语义分割任务的IoU结果。我们还使用了3DOVS数据集[24]，包含了不同pose和背景下捕获的长尾对象的集合。这个数据集是为开放词汇表三维语义分割开发的，其中提供了完整的类别列表。其他方法使用完整的列表来生成预测的掩码，而我们只使用查询类别来生成相应的掩码。mIoU度量用于这个数据集

实验细节。图像语言特征提取，使用OpenCLIP ViT-B/16模型；SAM使用ViT-H模型来分割二维mask；每个场景先使用3DGS来训练一个RGB，训练3万次迭代，每个场景都包含大约250万个点；然后固定三维高斯的所有其他参数，如均值和不透明度来训练我们的三维language gaussian。在这个阶段，只有语言特征是可学习的。语言特性训练3万次迭代，Autoencoder由一个MLP实现512维的CLIP特征压缩；1440×1080分辨率的场景，模型在NVIDIA RTX-3090 GPU上进行了25分钟训练，大约需要4GB的内存。

LERF数据集上算法对比。表1：算法达到84.3%的总体准确率，优于LERF；表2显示了三维语义分割的IoU结果，比LERF好14.0%

在这里插入图片描述

可视化分析。图1按照[20]PCA成分分析，可视化了学习到的三维language field。可以看到，LERF学习到的特征不能生成物体之间的清晰的边界，而我们的方法仅使用CLIP特征给出精确的物体形状。图3和图4中展示了对象定位和语义分割的可视化结果。我们观察到，LERF产生的激活区域更分散，而我们的激活区域更集中，与LERF产生的激活区域相比，我们的激活区域可以更好地符合真实形状

在这里插入图片描述

消融实验：

在这里插入图片描述

代码实现

安装环境

# 1.克隆项目到本地
git clone https://github.com/minghanqin/LangSplat.git --recursive# 2.安装依赖包
conda env create --file environment.yml
conda activate langsplat# 3.安装 sam
pip install git+https://github.com/facebookresearch/segment-anything.git
# 4.克隆并安装 segment-anything-langsplat
git clone https://github.com/minghanqin/segment-anything-langsplat.git
cd segment-anything-langsplat
pip install .

准备数据（数据应为以下格式）

<dataset_name>
|---images
|   |---<image 0>
|   |---<image 1>
|   |---...
|---input
|   |---<image 0>
|   |---<image 1>
|   |---...
|---output
|   |---<dataset_name>
|   |   |---point_cloud/iteration_30000/point_cloud.ply
|   |   |---cameras.json
|   |   |---cfg_args
|   |   |---chkpnt30000.pth
|   |   |---input.ply
|---sparse|---0|---cameras.bin|---images.bin|---points3D.bin

开始训练（分为5步执行，可直接执行process.sh）

# 1: 生成场景的 Language Feature。图片需要放在 <dataset_name>中的"input" 文件夹
python preprocess.py --dataset_path  data/lego（这是我自己的路径） # 2.训练 Autoencoder，得到低维的 3-dims Language Feature
cd autoencoder
python train.py --dataset_path /home/xzz/LangSplat/data/lego（这是我自己的路径） --dataset_name lego --encoder_dims 256 128 64 32 3 --decoder_dims 16 32 64 128 256 256 512 --lr 0.0007python test.py --dataset_name $dataset_path --output
# 3.最终训练语言场for level in 1 2 3
dopython train.py -s $dataset_path -m output/${casename} --start_checkpoint $dataset_path/$casename/chkpnt30000.pth --feature_level ${level}# e.g. python train.py -s data/sofa -m output/sofa --start_checkpoint data/sofa/sofa/chkpnt30000.pth --feature_level 3
done# 4.对不同level分割结果的渲染
for level in 1 2 3
do# render rgbpython render.py -m output/${casename}_${level}# render language featurespython render.py -m output/${casename}_${level} --include_feature# e.g. python render.py -m output/sofa_3 --include_feature
done