arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2312.11829cs.CV

RadOcc:通过渲染辅助蒸馏学习跨模态占用知识

RadOcc: Learning Cross-Modality Occupancy Knowledge through Rendering Assisted Distillation

Huawei Noah’s Ark Lab(华为诺亚方舟实验室)

Haiming Zhang, Xu Yan, Dongfeng Bai, Jiantao Gao, Pan Wang, Bingbing Liu, Shuguang Cui, Zhen Li

AI总结:

针对基于图像的3D占用预测缺乏几何先验的问题,提出RadOcc渲染辅助跨模态蒸馏范式,通过可微体积渲染构建深度与语义一致性损失,在nuScenes数据集上有效提升预测精度。

摘要:

3D占用预测是一项新兴任务,旨在利用多视图图像估计3D场景的占用状态和语义。然而,由于缺乏几何先验,基于图像的场景感知在实现准确预测方面面临重大挑战。在本文中,我们通过探索该任务中的跨模态知识蒸馏来解决这一问题,即我们在训练过程中利用更强的多模态模型来指导视觉模型。在实践中,我们观察到,直接应用在鸟瞰图(BEV)感知中提出并广泛使用的特征或logits对齐并不能产生令人满意的结果。为了克服这个问题,我们提出了RadOcc,一种用于3D占用预测的渲染辅助蒸馏范式。通过采用可微体积渲染,我们生成透视图中的深度和语义图,并提出了教师模型和学生模型渲染输出之间的两个新颖的一致性准则。具体而言,深度一致性损失对齐渲染光线的终止分布,而语义一致性损失则模仿由视觉基础模型(VLMs)引导的段内相似性。在nuScenes数据集上的实验结果证明了我们提出的方法在改进各种3D占用预测方法方面的有效性,例如,我们提出的方法将我们的基线在mIoU指标上提升了2.2%,并在Occ3D基准测试中达到了50%的成绩。

英文摘要:

3D occupancy prediction is an emerging task that aims to estimate the occupancy states and semantics of 3D scenes using multi-view images. However, image-based scene perception encounters significant challenges in achieving accurate prediction due to the absence of geometric priors. In this paper, we address this issue by exploring cross-modal knowledge distillation in this task, i.e., we leverage a stronger multi-modal model to guide the visual model during training. In practice, we observe that directly applying features or logits alignment, proposed and widely used in bird's-eyeview (BEV) perception, does not yield satisfactory results. To overcome this problem, we introduce RadOcc, a Rendering assisted distillation paradigm for 3D Occupancy prediction. By employing differentiable volume rendering, we generate depth and semantic maps in perspective views and propose two novel consistency criteria between the rendered outputs of teacher and student models. Specifically, the depth consistency loss aligns the termination distributions of the rendered rays, while the semantic consistency loss mimics the intra-segment similarity guided by vision foundation models (VLMs). Experimental results on the nuScenes dataset demonstrate the effectiveness of our proposed method in improving various 3D occupancy prediction approaches, e.g., our proposed methodology enhances our baseline by 2.2% in the metric of mIoU and achieves 50% in Occ3D benchmark.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑