Mint: A Simple Test-Time Adaptation of Vision-Language Models against Common Corruptions
2510.22127v1
cs.CV, cs.LG
2025-10-29
Авторы:
Wenxuan Bao, Ruxi Deng, Jingrui He
Abstract
Pretrained vision-language models such as CLIP achieve strong zero-shot
generalization but remain vulnerable to distribution shifts caused by input
corruptions. In this work, we investigate how corruptions affect CLIP's image
embeddings and uncover a consistent phenomenon we term as embedding variance
collapse, where both intra-class and inter-class variances shrink as corruption
severity increases. We find that this collapse is closely tied to performance
degradation, with inter-class variance strongly correlated with classification
accuracy. To explain this phenomenon, we analyze how corruptions alter the
structure of the embedding space. Our theoretical results suggest that the
visual encoder tends to encode corruption-related signals, which dilute
class-discriminative features and compress the representation geometry. We
further show that maximizing inter-class variance, even when estimated from
pseudo-labels, can provably enhance embedding quality. Based on this insight,
we propose Mint, a simple test-time adaptation method that maximizes
pseudo-label-based inter-class variance on the fly using a mean accumulator and
a gradient accumulator. Mint operates effectively with small batch sizes and
consistently improves performance across multiple corruption benchmarks and
CLIP architectures. Our code is available at https://github.com/baowenxuan/Mint .
Ссылки и действия
Дополнительные ресурсы: