针对生物数据细粒度特性,用分布对齐提升CLIP模型性能。
DALIP: Distribution Alignment-based Language-Image Pre-Training for Domain-Specific Data
- 用图像文本特征分布替代原[cls]向量进行对齐训练。
- 在植物数据集上达到94.2%准确率,优于现有方法。
- 适合生物、遥感、医学等细粒度视觉任务研究者使用。
近期对比语言-图像预训练(CLIP)在领域特定数据(如生物学)中表现出色,引发广泛关注。现有方法多依赖收集大量领域数据并微调原始CLIP模型,但未充分考虑领域数据的特性(如生物数据的细粒度),导致模型在通用领域的表现下降。本文提出基于分布对齐的语言-图像预训练方法DALIP,通过匹配图像-文本对特征分布而非原[cls]标记,捕捉其内在有效信息,更好地适应生物数据的细粒度特征。特别地,利用一阶与二阶统计量近似特征分布,并引入多头布朗距离协方差(MBDC)模块高效获取标记特征的二阶统计量。我们还构建了包含1000万植物数据与300万通用数据的全新数据集PlantMix-13M,遵循数据混合规律。大量实验表明,DALIP在生物领域显著优于现有CLIP方法,同时在遥感和医学影像领域也具备良好泛化能力。此外,PlantMix-13M进一步提升其在植物领域的性能,且保持通用领域能力。
原文摘要 · Abstract (English)
Recently, Contrastive Language-Image Pre-training (CLIP) has shown promising performance in domain-specific data (e.g., biology), and has attracted increasing research attention. Existing works generally focus on collecting extensive domain-specific data and directly tuning the original CLIP models. Intuitively, such a paradigm takes no full consideration of the characteristics lying in domain-specific data (e.g., fine-grained nature of biological data) and so limits model capability, while mostly losing the original ability of CLIP in the general domain. In this paper, we propose a Distribution Alignment-based Language-Image Pre-Training (DALIP) method for biological data. Specifically, DALIP optimizes CLIP models by matching the similarity between feature distribution of image-text pairs instead of the original [cls] token, which can capture rich yet effective information inherent in image-text pairs as powerful representations, and so better cope with fine-grained nature of biological data. Particularly, our DALIP efficiently approximates feature distribution via its first- and second-order statistics, while presenting a Multi-head Brownian Distance Covariance (MBDC) module to acquire second-order statistics of token features efficiently. Furthermore, we collect a new dataset for plant domain (e.g., specific data in biological domain) comprising 10M plant data with 3M general-domain data (namely PlantMix-13M) according to data mixing laws. Extensive experiments show that DALIP clearly outperforms existing CLIP counterparts in biological domain, while well generalizing to remote sensing and medical imaging domains. Besides, our PlantMix-13M dataset further boosts performance of DALIP in plant domain, while preserving model ability in general domain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。