arrow
返回

Context-Aware Robust Fine-Tuning

delete2023-12-03
delete3
PRE
AI
X
Xiaofeng Mao *
陈
陈钰枫 (Yufeng Chen)
X
Xiaojun Jia
R
Rong Zhang
H
Hui Xue
L
Li Zhao
DOI:10.1007/s11263-023-01951-2delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Contrastive language-image pre-trained (CLIP) models have zero-shot ability of classifying an image belonging to [CLASS] by using similarity between the image and the prompt sentence a [CONTEXT] of [CLASS]. Based on exhaustive text cues in [CONTEXT], CLIP model is aware of different contexts, e.g. background, style, viewpoint, and exhibits unprecedented robustness against a wide range of distribution shifts. However, recent works find further fine-tuning of CLIPmodels improves accuracy but sacrifices the robustness on downstream tasks. We conduct an empirical investigation to show fine-tuning will corrupt the context-aware ability of pre-trained CLIP features. To solve this problem, we propose Context-Aware Robust Fine-tuning (CAR-FT). CAR-FT regularizes the model during fine-tuning to capture the context information. Specifically, we use zero-shot prompt weights to get the context distribution contained in the image. By minimizing the KullbackLeibler divergence (KLD) between context distributions induced by original/fine-tuned CLIP models, CAR-FT makes the context-aware ability of CLIP inherited into downstream tasks, and achieves both higher in-distribution (ID) and out-ofdistribution (OOD) accuracy. The experimental results show CAR-FT achieves superior robustness on five OOD test datasets of ImageNet, and meanwhile brings accuracy gains on nine downstream tasks. Additionally, CAR-FT surpasses previous domain generalization (DG) methods and gets 78.5% averaged accuracy on DomainBed benchmark, building the new stateof-the-art.
Keyword:
Pre-trained models
CLIP
Fine-tuning
Robustness

期刊

International Journal of Computer Vision 封面图
International Journal of Computer Vision
IF:
9.3
论文数:
3.9K
被引数:
2.8W

机构

A
alibaba group
学者数:
1.1K
论文数: 789
被引数: 0
I
institute of information engineering, cas
学者数:
474
论文数: 466
被引数: 0
C
chinese academy of sciences
学者数:
56.7W
论文数: 45.0W
被引数: 704
学者 查看更多机构
引用论文

引用论文

err
IF0
err
err0
PREAI
err
err分享
err收藏
A unifying view on dataset shift in classification关于分类中数据集移位的统一观点
err2012-01-01
err611
PREAI
errMoreno-Torres, Jose G.; Raeder, Troy; Alaiz-Rodriguez, Rocio; Chawla, Nitesh V.; Herrera, Francisco
err分享
err收藏
Role of interferon-γ in Vα14+ natural killer T cell-mediated host defense against Streptococcus pneumoniae infection in murine lungs
err2007-03-01
err0
PREAI
errMasashi Nakamatsu; Natsuo Yamamoto; Masumitsu Hatta; Chikara Nakasone; Takeshi Kinjo; Kazuya Miyagi; Kaori Uezu; Kiwamu Nakamura; Toshinori Nakayama; Masaru Taniguchi; Yoichiro Iwakura; Mitsuo Kaku; Jiro Fujita; Kazuyoshi Kawakami
err分享
err收藏
High prevalence of malaria in a non-endemic setting among febrile episodes in travellers and migrants coming from endemic areas: a retrospective analysis of a 2013–2018 cohort
err2021-11-27
err0
errOAAI
errAlejandro Garcia-Ruiz de Morales; Covadonga Morcate; Elena Isaba-Ares; Ramon Perez-Tanoira; Jose A. Perez-Molina
err分享
err收藏
Elements of Quaternions四元数的元素
err
IF0
err2015-07-05
err0
PREAI
errWilliam Rowan Hamilton
err分享
err收藏
SUN Database: Exploring a Large Collection of Scene CategoriesSUN数据库: 探索大量场景类别
err2014-08-13
err181
PREAI
errXiao, Jianxiong; Ehinger, Krista A.; Hays, James; Torralba, Antonio; Oliva, Aude
err分享
err收藏
err分享
err收藏
学者 查看更多内容