用可学习矩阵映射压缩CLIP模型,高效保留特征能力。
CLIP-Map: Structured Matrix Mapping for Parameter-Efficient CLIP Compression
- 通过克罗内克分解的全映射矩阵,组合预训练权重
- 在高压缩率下仍优于选择式方法,最高提升12.3%准确率
- 适合资源受限场景下的CLIP轻量化部署
对比语言-图像预训练(CLIP)在文本生成图像、图文检索和图像描述等任务中广泛应用,但其高内存与计算开销限制了在资源受限场景的应用。现有压缩方法通常通过掩码优化或重要性度量选择部分权重进行继承再训练,但此类选择式方法在极端压缩下常损害特征表达能力。本文提出一种基于映射的CLIP压缩框架CLIP-Map,利用可学习矩阵结合预训练权重,采用全映射与克罗内克分解,以尽可能保留原始权重信息。为缓解可学习映射带来的优化挑战,提出对角继承初始化,降低分布偏移,实现高效映射学习。大量实验表明,所提方法在多种压缩比下均优于选择式框架,尤其在高压缩设置下表现显著更优。
原文摘要 · Abstract (English)
Contrastive Language-Image Pre-training (CLIP) has achieved widely applications in various computer vision tasks, e.g., text-to-image generation, Image-Text retrieval and Image captioning. However, CLIP suffers from high memory and computation cost, which prohibits its usage to the resource-limited application scenarios. Existing CLIP compression methods typically reduce the size of pre-trained CLIP weights by selecting their subset as weight inheritance for further retraining via mask optimization or important weight measurement. However, these select-based weight inheritance often compromises the feature presentation ability, especially on the extreme compression. In this paper, we propose a novel mapping-based CLIP compression framework, CLIP-Map. It leverages learnable matrices to map and combine pretrained weights by Full-Mapping with Kronecker Factorization, aiming to preserve as much information from the original weights as possible. To mitigate the optimization challenges introduced by the learnable mapping, we propose Diagonal Inheritance Initialization to reduce the distribution shifting problem for efficient and effective mapping learning. Extensive experimental results demonstrate that the proposed CLIP-Map outperforms select-based frameworks across various compression ratios, with particularly significant gains observed under high compression settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。