Distilling Vision-Language Foundation Models: A Data-Free Approach via Prompt Diversification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xuan, Yunyi, Chen, Weijie, Yang, Shicai, Xie, Di, Lin, Luojun, Zhuang, Yueting
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913439524323328
author Xuan, Yunyi
Chen, Weijie
Yang, Shicai
Xie, Di
Lin, Luojun
Zhuang, Yueting
author_facet Xuan, Yunyi
Chen, Weijie
Yang, Shicai
Xie, Di
Lin, Luojun
Zhuang, Yueting
contents Data-Free Knowledge Distillation (DFKD) has shown great potential in creating a compact student model while alleviating the dependency on real training data by synthesizing surrogate data. However, prior arts are seldom discussed under distribution shifts, which may be vulnerable in real-world applications. Recent Vision-Language Foundation Models, e.g., CLIP, have demonstrated remarkable performance in zero-shot out-of-distribution generalization, yet consuming heavy computation resources. In this paper, we discuss the extension of DFKD to Vision-Language Foundation Models without access to the billion-level image-text datasets. The objective is to customize a student model for distribution-agnostic downstream tasks with given category concepts, inheriting the out-of-distribution generalization capability from the pre-trained foundation models. In order to avoid generalization degradation, the primary challenge of this task lies in synthesizing diverse surrogate images driven by text prompts. Since not only category concepts but also style information are encoded in text prompts, we propose three novel Prompt Diversification methods to encourage image synthesis with diverse styles, namely Mix-Prompt, Random-Prompt, and Contrastive-Prompt. Experiments on out-of-distribution generalization datasets demonstrate the effectiveness of the proposed methods, with Contrastive-Prompt performing the best.
format Preprint
id arxiv_https___arxiv_org_abs_2407_15155
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Distilling Vision-Language Foundation Models: A Data-Free Approach via Prompt Diversification
Xuan, Yunyi
Chen, Weijie
Yang, Shicai
Xie, Di
Lin, Luojun
Zhuang, Yueting
Computer Vision and Pattern Recognition
Artificial Intelligence
Multimedia
Data-Free Knowledge Distillation (DFKD) has shown great potential in creating a compact student model while alleviating the dependency on real training data by synthesizing surrogate data. However, prior arts are seldom discussed under distribution shifts, which may be vulnerable in real-world applications. Recent Vision-Language Foundation Models, e.g., CLIP, have demonstrated remarkable performance in zero-shot out-of-distribution generalization, yet consuming heavy computation resources. In this paper, we discuss the extension of DFKD to Vision-Language Foundation Models without access to the billion-level image-text datasets. The objective is to customize a student model for distribution-agnostic downstream tasks with given category concepts, inheriting the out-of-distribution generalization capability from the pre-trained foundation models. In order to avoid generalization degradation, the primary challenge of this task lies in synthesizing diverse surrogate images driven by text prompts. Since not only category concepts but also style information are encoded in text prompts, we propose three novel Prompt Diversification methods to encourage image synthesis with diverse styles, namely Mix-Prompt, Random-Prompt, and Contrastive-Prompt. Experiments on out-of-distribution generalization datasets demonstrate the effectiveness of the proposed methods, with Contrastive-Prompt performing the best.
title Distilling Vision-Language Foundation Models: A Data-Free Approach via Prompt Diversification
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Multimedia
url https://arxiv.org/abs/2407.15155