arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.03312cs.CVcs.LG

深度补全的测试时自适应

Test-Time Adaptation for Depth Completion

Yale Vision Lab(耶鲁视觉实验室)

Hyoungseob Park, Anjali Gupta, Alex Wong

AI总结:

针对深度补全任务,提出一种在线测试时自适应方法,利用稀疏深度模态的较小协变量偏移,通过嵌入模块和自适应层在单次遍历中缩小域差距,室内外场景平均提升21.1%。

摘要:

在将模型从某些源数据集迁移到目标测试数据时,由于它们之间存在域差距,性能下降是常见现象。现有的弥合这一差距的方法,如域适应(DA),可能需要用于训练模型的源数据(这些数据通常不可用),而其他方法,即无源域适应(source-free DA),则需要多次遍历测试数据。我们提出了一种用于深度补全的在线测试时自适应方法,深度补全是从单张图像和相关的稀疏深度图推断出密集深度图的任务,该方法在单次遍历中即可缩小性能差距。我们首先研究了每种数据模态中的域偏移如何影响模型性能。基于我们的观察,即稀疏深度模态比图像表现出小得多的协变量偏移,我们设计了一个在源域中训练的嵌入模块,该模块保留了从仅编码稀疏深度的特征到编码图像和稀疏深度的特征的映射。在测试时,稀疏深度特征使用该映射作为源域特征的代理进行投影,并作为指导来训练一组辅助参数(即自适应层),以将来自目标测试域的图像和稀疏深度特征与源域的特征对齐。我们在室内和室外场景中评估了我们的方法,并表明它平均比基线提高了21.1%。

英文摘要:

It is common to observe performance degradation when transferring models trained on some (source) datasets to target testing data due to a domain gap between them. Existing methods for bridging this gap, such as domain adaptation (DA), may require the source data on which the model was trained (often not available), while others, i.e., source-free DA, require many passes through the testing data. We propose an online test-time adaptation method for depth completion, the task of inferring a dense depth map from a single image and associated sparse depth map, that closes the performance gap in a single pass. We first present a study on how the domain shift in each data modality affects model performance. Based on our observations that the sparse depth modality exhibits a much smaller covariate shift than the image, we design an embedding module trained in the source domain that preserves a mapping from features encoding only sparse depth to those encoding image and sparse depth. During test time, sparse depth features are projected using this map as a proxy for source domain features and are used as guidance to train a set of auxiliary parameters (i.e., adaptation layer) to align image and sparse depth features from the target test domain to that of the source domain. We evaluate our method on indoor and outdoor scenarios and show that it improves over baselines by an average of 21.1%.

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

↑