用负向增强提升视觉模型在图像退化下的鲁棒性
Panda: Test-Time Adaptation with Negative Data Augmentation
- 通过打乱图像块生成负向样本,抑制退化特征
- 共享批量增强计算,仅增加1.2%推理开销
- 可插件式集成,显著降低预测偏差
预训练视觉语言模型(VLM)具备强大的零样本分类能力,但在常见图像退化下性能显著下降。现有测试时自适应(TTA)方法多采用正向数据增强(PDA),对每张测试图像生成多个视图以降低预测方差,但存在两大缺陷:一是需大量增强导致计算开销大;二是无法缓解模型在退化条件下的预测偏差问题,因PDA作用于已退化的输入且不消除退化本身。为此,我们提出Panda,一种基于负向数据增强(NDA)的新颖TTA方法。不同于保持语义的正向增强,Panda通过将图像分割为块并从共享块池中随机重组生成负向样本,保留退化特征但丢弃与对象相关的信号。随后将这些负向样本的特征均值从原图特征中减去,有效抑制退化相关成分,同时保留类别相关信息,从而缓解分布偏移下的预测偏差。Panda支持批量内增强共享,计算开销极低(仅增加1.2%推理时间)。该方法可无缝集成至现有TTA框架,实验表明其性能优于主流PDA方法,且多种现有TTA方法在引入Panda后均获得显著性能提升。代码已开源。
原文摘要 · Abstract (English)
Pretrained VLMs exhibit strong zero-shot classification capabilities, but their predictions degrade significantly under common image corruptions. To improve robustness, many test-time adaptation (TTA) methods adopt positive data augmentation (PDA), which generates multiple views of each test sample to reduce prediction variance. However, these methods suffer from two key limitations. First, it introduces considerable computational overhead due to the large number of augmentations required per image. Second, it fails to mitigate prediction bias, where the model tends to predict certain classes disproportionately under corruption, as PDA operates on corrupted inputs and typically does not remove the corruption itself. To address these challenges, we propose Panda, a novel TTA method based on negative data augmentation (NDA). Unlike positive augmentations that preserve object semantics, Panda generates negative augmentations by disrupting semantic content. It divides images into patches and randomly assembles them from a shared patch pool. These negatively augmented images retain corruption-specific features while discarding object-relevant signals. We then subtract the mean feature of these negative samples from the original image feature, effectively suppressing corruption-related components while preserving class-relevant information. This mitigates prediction bias under distribution shifts. Panda allows augmentation to be shared across samples within a batch, resulting in minimal computational overhead. Panda can be seamlessly integrated into existing test-time adaptation frameworks and substantially improve their robustness. Our experiments indicate that Panda delivers superior performance compared to PDA methods, and a wide range of TTA methods exhibit significantly enhanced performance when integrated with Panda. Our code is available at https://github.com/ruxideng/Panda .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。