References & Citations
Computer Science > Machine Learning
Title: CLIPood: Generalizing CLIP to Out-of-Distributions
(Submitted on 2 Feb 2023 (v1), last revised 13 Jul 2023 (this version, v2))
Abstract: Out-of-distribution (OOD) generalization, where the model needs to handle distribution shifts from training, is a major challenge of machine learning. Contrastive language-image pre-training (CLIP) models have shown impressive zero-shot ability, but the further adaptation of CLIP on downstream tasks undesirably degrades OOD performances. This paper aims at generalizing CLIP to out-of-distribution test data on downstream tasks. We propose CLIPood, a fine-tuning method that can adapt CLIP models to OOD situations where both domain shifts and open classes may occur on the unseen test data. To exploit the semantic relations between classes from the text modality, CLIPood introduces a new training objective, margin metric softmax (MMS), with class adaptive margins for fine-tuning. To incorporate both pre-trained zero-shot model and fine-tuned task-adaptive model, CLIPood leverages a new optimization strategy, Beta moving average (BMA), to maintain a temporal ensemble weighted by Beta distribution. Experiments on diverse datasets with different OOD scenarios show that CLIPood consistently outperforms existing generalization techniques.
Submission history
From: Yang Shu [view email][v1] Thu, 2 Feb 2023 04:27:54 GMT (1486kb,D)
[v2] Thu, 13 Jul 2023 09:16:39 GMT (1470kb,D)
Link back to: arXiv, form interface, contact.