arXivDaily arXiv每日学术速递 周一至周五更新
稍安勿躁,官方尚未更新
arXiv 2408.09441cs.CV
收藏

CLIP-CID:通过聚类-实例判别实现高效的CLIP蒸馏

CLIP-CID: Efficient CLIP Distillation via Cluster-Instance Discrimination

DeepGlint(格灵深瞳);University of Sydney(悉尼大学);Imperial College(帝国理工学院)

Kaicheng Yang, Tiancheng Gu, Xiang An, Haiqiang Jiang, Xiangzi Dai, Ziyong Feng, Weidong Cai, Jiankang Deng

AI总结:

本文提出CLIP-CID蒸馏机制,通过图像语义平衡过滤43.7%数据及聚类-实例判别,将大型视觉-语言模型知识高效迁移至小模型,在多项下游任务上达到最先进性能。

摘要:

对比语言-图像预训练(CLIP)在广泛的任务中取得了优异的性能。然而,CLIP的有效性严重依赖于大量的预训练数据,导致计算资源的显著消耗。尽管知识蒸馏已广泛应用于单模态模型,但如何高效地将知识蒸馏扩展到具有大量数据的视觉-语言基础模型仍相对未被探索。在本文中,我们引入了CLIP-CID,一种新颖的蒸馏机制,能够有效地将知识从大型视觉-语言基础模型迁移到较小的模型。我们首先提出了一种简单但高效的图像语义平衡方法,以减少迁移学习偏差并提高蒸馏效率。该方法从LAION400M中过滤掉43.7%的图像-文本对,同时保持优越的性能。之后,我们利用聚类-实例判别来促进从教师模型到学生模型的知识迁移,从而使学生模型能够获得对预训练数据的整体语义理解。实验结果表明,CLIP-CID在包括线性探针和零样本分类在内的各种下游任务上取得了最先进的性能。

英文摘要:

Contrastive Language-Image Pre-training (CLIP) has achieved excellent performance over a wide range of tasks. However, the effectiveness of CLIP heavily relies on a substantial corpus of pre-training data, resulting in notable consumption of computational resources. Although knowledge distillation has been widely applied in single modality models, how to efficiently expand knowledge distillation to vision-language foundation models with extensive data remains relatively unexplored. In this paper, we introduce CLIP-CID, a novel distillation mechanism that effectively transfers knowledge from a large vision-language foundation model to a smaller model. We initially propose a simple but efficient image semantic balance method to reduce transfer learning bias and improve distillation efficiency. This method filters out 43.7% of image-text pairs from the LAION400M while maintaining superior performance. After that, we leverage cluster-instance discrimination to facilitate knowledge transfer from the teacher model to the student model, thereby empowering the student model to acquire a holistic semantic comprehension of the pre-training data. Experimental results demonstrate that CLIP-CID achieves state-of-the-art performance on various downstream tasks including linear probe and zero-shot classification.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑