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

OT-VP:面向测试时适应的最优传输引导视觉提示

OT-VP: Optimal Transport-guided Visual Prompting for Test-Time Adaptation

Tulane University(杜兰大学)

Yunbei Zhang, Akshay Mehra, Jihun Hamm

AI总结:

本文提出 OT-VP,通过测试时优化最优传输距离学习仅含四个 token 的通用视觉提示,在不改动预训练参数的情况下对齐源域与目标域,并在多个域适应及损坏基准上取得最优性能。

摘要:

Vision Transformers(ViT)在学习表征方面展现出卓越能力,但在应用于未见过的域时,其性能会受到损害。以往方法要么在训练阶段进行提示学习,要么在测试时通过熵最小化修改模型参数。前者往往忽视未标注的目标数据,而后者未能完全解决域偏移问题。在本工作中,我们的方法——最优传输引导的测试时视觉提示(Optimal Transport-guided Test-Time Visual Prompting,OT-VP)——通过在测试时利用提示学习来对齐目标域与源域,从而处理这些问题,且无需访问训练过程或更改预训练模型参数。该方法通过优化最优传输距离,为目标域学习一个通用视觉提示。OT-VP 仅使用四个学习到的提示 token,就在三个风格化数据集 PACS、VLCS、OfficeHome 以及一个损坏数据集 ImageNet-C 上超过了最先进性能。此外,OT-VP 在内存和计算方面均运行高效,并可灵活扩展到在线设置。

英文摘要:

Vision Transformers (ViTs) have demonstrated remarkable capabilities in learning representations, but their performance is compromised when applied to unseen domains. Previous methods either engage in prompt learning during the training phase or modify model parameters at test time through entropy minimization. The former often overlooks unlabeled target data, while the latter doesn't fully address domain shifts. In this work, our approach, Optimal Transport-guided Test-Time Visual Prompting (OT-VP), handles these problems by leveraging prompt learning at test time to align the target and source domains without accessing the training process or altering pre-trained model parameters. This method involves learning a universal visual prompt for the target domain by optimizing the Optimal Transport distance.OT-VP, with only four learned prompt tokens, exceeds state-of-the-art performance across three stylistic datasets-PACS, VLCS, OfficeHome, and one corrupted dataset ImageNet-C. Additionally, OT-VP operates efficiently, both in terms of memory and computation, and is adaptable for extension to online settings.

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

↑