DiffCLIP: Few-shot Language-driven Multimodal Classifier

Jiaqing Zhang,Mingxiang Cao,Xue Yang,Kai Jiang,Yunsong Li
2024-12-10
Abstract:Visual language models like Contrastive Language-Image Pretraining (CLIP) have shown impressive performance in analyzing natural images with language information. However, these models often encounter challenges when applied to specialized domains such as remote sensing due to the limited availability of image-text pairs for training. To tackle this issue, we introduce DiffCLIP, a novel framework that extends CLIP to effectively convey comprehensive language-driven semantic information for accurate classification of high-dimensional multimodal remote sensing images. DiffCLIP is a few-shot learning method that leverages unlabeled images for pretraining. It employs unsupervised mask diffusion learning to capture the distribution of diverse modalities without requiring labels. The modality-shared image encoder maps multimodal data into a unified subspace, extracting shared features with consistent parameters across modalities. A well-trained image encoder further enhances learning by aligning visual representations with class-label text information from CLIP. By integrating these approaches, DiffCLIP significantly boosts CLIP performance using a minimal number of image-text pairs. We evaluate DiffCLIP on widely used high-dimensional multimodal datasets, demonstrating its effectiveness in addressing few-shot annotated classification tasks. DiffCLIP achieves an overall accuracy improvement of 10.65% across three remote sensing datasets compared with CLIP, while utilizing only 2-shot image-text pairs. The code has been released at <a class="link-external link-https" href="https://github.com/icey-zhang/DiffCLIP" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the difficulty in model training due to the scarcity of labeled data in high - dimensional multi - modal remote sensing image classification. Specifically, existing vision - language models (such as CLIP) perform poorly when processing high - dimensional remote sensing images in specific domains because these models are usually trained on natural images and cannot fully capture the diversity and complexity of specific domains. In addition, the labeling of remote sensing images requires professional knowledge and a large amount of time, resulting in very limited available image - text pairs. Therefore, the paper proposes a new framework named DiffCLIP, aiming to achieve effective classification of high - dimensional multi - modal remote sensing images with a small number of labeled samples. ### Main problems and solutions 1. **Scarcity of labeled data**: The labeling cost of remote sensing images is high and time - consuming, resulting in a limited amount of data available for training. DiffCLIP reduces the dependence on labeled data by introducing unsupervised mask diffusion learning and using unlabeled images for pre - training. 2. **Insufficient generalization ability of the model**: Existing vision - language models (such as CLIP) have limited generalization ability when processing high - dimensional remote sensing images in specific domains due to the limitations of training data. DiffCLIP enhances the model's semantic understanding ability of different modal data and improves the classification performance by introducing a language - driven few - shot classification method. ### Specific methods - **Unsupervised mask diffusion**: Randomly mask some areas in the input image and gradually add noise to generate a series of degraded images. Then, use the image encoder with shared parameters and the decoder of specific modalities to restore the original image, thereby extracting cross - modal shared features. - **Language - driven few - shot classification**: Use the text encoder of CLIP to generate text descriptions of class labels, and align the image features with the text features through the contrastive learning method to achieve few - shot classification. This method can make full use of a small amount of labeled data and improve the classification accuracy. ### Experimental results - **Performance improvement**: The experimental results on multiple high - dimensional multi - modal remote sensing image data sets show that DiffCLIP has achieved significant performance improvement in few - shot classification tasks compared with CLIP and other baseline methods. For example, on the Houston data set, when using the 2 - shot setting, the overall accuracy of DiffCLIP is 16.32% higher than that of CLIP. - **Robustness verification**: Through ablation experiments, the influence of each module (such as mask diffusion, language - driven, etc.) on the model performance is verified, further proving the effectiveness and robustness of DiffCLIP. ### Conclusion DiffCLIP successfully solves the problems of data scarcity and insufficient model generalization ability in high - dimensional multi - modal remote sensing image classification by combining unsupervised mask diffusion and language - driven few - shot classification methods, providing new ideas and methods for research in this field.