CLIP-MoE: Towards Building Mixture of Experts for CLIP with Diversified Multiplet Upcycling

Jihai Zhang, Xiaoye Qu, Tong Zhu, Yu Cheng
9/28/2024
cs.CVcs.AI

No official code found. Best implementation: OpenSparseLLMs/CLIP-MoE (58 stars, last commit 2024-10-10).

Abstract

Contrastive Language-Image Pre-training (CLIP) has become a cornerstone in multimodal intelligence. However, recent studies discovered that CLIP can only encode one aspect of the feature space, leading to substantial information loss and indistinctive features. To mitigate this issue, this paper introduces a novel strategy that fine-tunes a series of complementary CLIP models and transforms them into a CLIP-MoE. Specifically, we propose a model-agnostic Diversified Multiplet Upcycling (DMU) framework for CLIP. Instead of training multiple CLIP models from scratch, DMU leverages a pre-trained CLIP and fine-tunes it into a diverse set with highly cost-effective multistage contrastive learning, thus capturing distinct feature subspaces efficiently. To fully exploit these fine-tuned models while minimizing computational overhead, we transform them into a CLIP-MoE, which dynamically activates a subset of CLIP experts, achieving an effective balance between model capacity and computational cost. Comprehensive experiments demonstrate the superior performance of CLIP-MoE across various zero-shot retrieval, zero-shot image classification tasks, and downstream Multimodal Large Language Model (MLLM) benchmarks when used as a vision encoder.

View on arXivView PDF

Code Implementations(1)

CLIP-MoE: Mixture of Experts for CLIP

584Python, Jupyter NotebookSep 19, 20242 years ago
cliplvlmmixture-of-expertsmoeopenai-clip

Similar Papers

Cite this paper

@article{zhang2024clipmoe,
  title  = {CLIP-MoE: Towards Building Mixture of Experts for CLIP with Diversified Multiplet Upcycling},
  author = {Jihai Zhang and Xiaoye Qu and Tong Zhu and Yu Cheng},
  year   = {2024},
  eprint = {2409.19291},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2409.19291v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2409.19291.svg)](https://codeofpaper.com/paper/2409.19291)

Discussion