arXiv:2412.07119cs.CV2024-12AAAI被引 4

用少量图文对实现遥感图像精准分类,性能提升超10%。

DiffCLIP: Few-shot Language-driven Multimodal Classifier

  • 通过无监督掩码扩散预训练,利用未标注图像学习多模态分布。
  • 仅用2张图文对,遥感分类准确率比CLIP提升10.65%。
  • 适合标签稀缺的遥感、医学等专业领域少样本图像分类。

视觉语言模型如对比语言-图像预训练(CLIP)在自然图像的语义分析中表现优异,但在遥感等专业领域因缺乏足够图像-文本对而面临挑战。为此,本文提出DiffCLIP,一种新型少样本学习框架,通过无监督掩码扩散学习,利用未标注图像预训练,捕捉多模态数据分布。共享模态的图像编码器将多源数据映射至统一特征空间,保持参数一致;经训练的编码器进一步与CLIP的类别文本信息对齐,增强表征。实验在三个主流高维多模态遥感数据集上验证,使用仅2张图像-文本对,整体准确率相比CLIP提升10.65%。代码已开源:https://github.com/icey-zhang/DiffCLIP。

原文摘要 · Abstract (English)

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 https://github.com/icey-zhang/DiffCLIP.

少样本学习遥感图像多模态融合扩散模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。