让视觉模型在推理时自适应优化提示,避免因数据变化导致性能下降。
Historical Test-time Prompt Tuning for Vision Foundation Models
- 引入三种知识库记忆有效提示信息,提升在线学习鲁棒性。
- 在多任务和连续域数据上保持稳定性能,优于现有方法。
- 适合需要持续适应新数据的视觉模型部署场景。
测试时提示调优通过在推理阶段利用无标注测试样本在线学习提示,展现出无需任务标注即可生成有效提示的潜力。然而,随着提示不断更新,其性能常随调优过程明显下降,尤其当测试数据分布持续变化时更为严重。本文提出历史测试时提示调优(HisTPT),通过记忆已学测试样本中的有用知识,实现鲁棒的测试时提示调优。HisTPT引入三种知识库:局部知识库、难例知识库和全局知识库,分别采用不同机制实现知识记忆与提示优化。此外,该方法设计自适应知识检索机制,通过动态召回记忆知识来正则化每个测试样本的预测。大量实验表明,HisTPT在图像分类、语义分割和目标检测等视觉识别任务中,面对持续变化的数据域,均能持续保持优异的提示调优性能。
原文摘要 · Abstract (English)
Test-time prompt tuning, which learns prompts online with unlabelled test samples during the inference stage, has demonstrated great potential by learning effective prompts on-the-fly without requiring any task-specific annotations. However, its performance often degrades clearly along the tuning process when the prompts are continuously updated with the test data flow, and the degradation becomes more severe when the domain of test samples changes continuously. We propose HisTPT, a Historical Test-time Prompt Tuning technique that memorizes the useful knowledge of the learnt test samples and enables robust test-time prompt tuning with the memorized knowledge. HisTPT introduces three types of knowledge banks, namely, local knowledge bank, hard-sample knowledge bank, and global knowledge bank, each of which works with different mechanisms for effective knowledge memorization and test-time prompt optimization. In addition, HisTPT features an adaptive knowledge retrieval mechanism that regularizes the prediction of each test sample by adaptively retrieving the memorized knowledge. Extensive experiments show that HisTPT achieves superior prompt tuning performance consistently while handling different visual recognition tasks (e.g., image classification, semantic segmentation, and object detection) and test samples from continuously changing domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。