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

学习适配 CLIP 以实现少样本单目深度估计

Learning to Adapt CLIP for Few-Shot Monocular Depth Estimation

Southern University of Science and Technology(南方科技大学);Pengcheng Laboratory(鹏城实验室)

Xueting Hu, Ce Zhang, Yi Zhang, Bowen Hai, Ke Yu, Zhihai He

AI总结:

本文提出少样本适配 CLIP 的单目深度估计方法,通过场景可选深度区间和可学习文本提示提升泛化性能,在 NYU V2 与 KITTI 上每场景仅用一张训练图,MARE 最高提升 10.6%。

摘要:

预训练的视觉-语言模型(VLM),例如 CLIP,已在一系列涉及视觉与语言模态融合的任务中展现出增强的性能。当 CLIP 用于深度估计任务时,从输入图像划分出的图像块可以与一系列深度信息的语义描述相结合,以获得相似度结果。随后,通过对与预定义语义描述相对应的深度值(称为深度区间,depth bins)进行加权和求和,得到粗略的深度估计。这种零样本方法规避了传统全监督深度估计方法在计算和时间上的高成本。然而,这种使用固定深度区间的方法可能无法有效泛化,因为来自不同场景的图像可能呈现出明显不同的深度分布。为应对这一挑战,我们提出了一种基于少样本的方法,该方法学习适配 VLM 以进行单目深度估计,从而在训练成本与泛化能力之间取得平衡。具体而言,该方法为不同场景分配不同的深度区间,模型可在推理期间对其进行选择。此外,我们引入可学习提示来预处理输入文本,将人类易于理解的文本转换为模型易于理解的向量,并进一步提升性能。尽管每个场景仅使用一张图像进行训练,我们在 NYU V2 和 KITTI 数据集上的大量实验结果表明,就 MARE 而言,本方法较此前最先进方法的性能提升最高达 10.6%。

英文摘要:

Pre-trained Vision-Language Models (VLMs), such as CLIP, have shown enhanced performance across a range of tasks that involve the integration of visual and linguistic modalities. When CLIP is used for depth estimation tasks, the patches, divided from the input images, can be combined with a series of semantic descriptions of the depth information to obtain similarity results. The coarse estimation of depth is then achieved by weighting and summing the depth values, called depth bins, corresponding to the predefined semantic descriptions. The zero-shot approach circumvents the computational and time-intensive nature of traditional fully-supervised depth estimation methods. However, this method, utilizing fixed depth bins, may not effectively generalize as images from different scenes may exhibit distinct depth distributions. To address this challenge, we propose a few-shot-based method which learns to adapt the VLMs for monocular depth estimation to balance training costs and generalization capabilities. Specifically, it assigns different depth bins for different scenes, which can be selected by the model during inference. Additionally, we incorporate learnable prompts to preprocess the input text to convert the easily human-understood text into easily model-understood vectors and further enhance the performance. With only one image per scene for training, our extensive experiment results on the NYU V2 and KITTI dataset demonstrate that our method outperforms the previous state-of-the-art method by up to 10.6\% in terms of MARE.

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

↑