用图文生成增强测试时适应,提升模型在未知场景下的泛化能力
Diffusion-Enhanced Test-time Adaptation with Text and Image Augmentation
- 结合图文生成进行多模态数据增强,突破单模态局限
- 零样本下准确率提升5.50%,显著优于现有方法
- 适合应对分布偏移和领域差异的下游任务应用
现有测试时提示调优(TPT)方法主要依赖单模态数据增强,通常仅增强图像并基于置信度筛选不准确样本。然而,图像生成模型虽能产生视觉多样性,但单模态增强仍无法充分捕捉多模态提供的完整知识。此外,当增强图像数量受限时,TPT方法性能显著下降——这在生成式增强计算成本较高的背景下并不罕见。为此,我们提出IT3A,一种新的测试时适应方法,利用预训练生成模型对未知新域的每个测试样本进行多模态增强。通过融合预训练视觉与语言模型生成的图文数据,提升模型对未知测试数据的适应能力。为确保生成增强中关键语义的准确性,我们采用增强图像与文本的logits间余弦相似度过滤机制,剔除虚假或不匹配的组合。为进一步灵活利用跨模态生成的多样化增强,我们以适配器替代提示调优。在存在分布偏移与领域差距的测试集上实验表明,在零样本设置下,IT3A相比顶尖测试时提示调优方法准确率提升5.50%。
原文摘要 · Abstract (English)
Existing test-time prompt tuning (TPT) methods focus on single-modality data, primarily enhancing images and using confidence ratings to filter out inaccurate images. However, while image generation models can produce visually diverse images, single-modality data enhancement techniques still fail to capture the comprehensive knowledge provided by different modalities. Additionally, we note that the performance of TPT-based methods drops significantly when the number of augmented images is limited, which is not unusual given the computational expense of generative augmentation. To address these issues, we introduce IT3A, a novel test-time adaptation method that utilizes a pre-trained generative model for multi-modal augmentation of each test sample from unknown new domains. By combining augmented data from pre-trained vision and language models, we enhance the ability of the model to adapt to unknown new test data. Additionally, to ensure that key semantics are accurately retained when generating various visual and text enhancements, we employ cosine similarity filtering between the logits of the enhanced images and text with the original test data. This process allows us to filter out some spurious augmentation and inadequate combinations. To leverage the diverse enhancements provided by the generation model across different modals, we have replaced prompt tuning with an adapter for greater flexibility in utilizing text templates. Our experiments on the test datasets with distribution shifts and domain gaps show that in a zero-shot setting, IT3A outperforms state-of-the-art test-time prompt tuning methods with a 5.50% increase in accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。