通过增强跨模态对齐,提升多模态模型的分布外检测能力。
Enhanced OoD Detection through Cross-Modal Alignment of Multi-Modal Representations
- 提出在分布内数据上强化图像与文本嵌入的跨模态对齐
- 在ImageNet-1k OoD基准上达到最优性能,同时保持高识别准确率
- 适合关注多模态表示学习与分布外检测的研究者
以往的分布外检测(OoDD)研究主要集中在单模态模型。随着CLIP等大规模预训练视觉语言模型的出现,利用零样本和提示学习策略的多模态表示方法应运而生。然而,这些方法通常冻结或仅部分微调预训练权重,难以适配下游数据集。本文指出,多模态微调(MMFT)可显著提升OoDD性能。尽管已有研究验证微调对OoDD的影响,仍有较大改进空间。我们分析发现,简单微调失败的原因在于分布内嵌入中的模态差距。为此,提出一种新训练目标,通过正则化分布内数据的图像与文本嵌入距离,增强跨模态对齐。该方法使不同模态中语义相似的信息在超球面表示空间中更紧密对齐。理论上证明该正则化等价于超球面上能量模型的最大似然估计。在ImageNet-1k OoD基准上,结合后处理OoDD方法(如NegLabel),本方法显著优于现有方法,实现当前最优的OoDD性能与更高的分布内准确率。
原文摘要 · Abstract (English)
Prior research on out-of-distribution detection (OoDD) has primarily focused on single-modality models. Recently, with the advent of large-scale pretrained vision-language models such as CLIP, OoDD methods utilizing such multi-modal representations through zero-shot and prompt learning strategies have emerged. However, these methods typically involve either freezing the pretrained weights or only partially tuning them, which can be suboptimal for downstream datasets. In this paper, we highlight that multi-modal fine-tuning (MMFT) can achieve notable OoDD performance. Despite some recent works demonstrating the impact of fine-tuning methods for OoDD, there remains significant potential for performance improvement. We investigate the limitation of naïve fine-tuning methods, examining why they fail to fully leverage the pretrained knowledge. Our empirical analysis suggests that this issue could stem from the modality gap within in-distribution (ID) embeddings. To address this, we propose a training objective that enhances cross-modal alignment by regularizing the distances between image and text embeddings of ID data. This adjustment helps in better utilizing pretrained textual information by aligning similar semantics from different modalities (i.e., text and image) more closely in the hyperspherical representation space. We theoretically demonstrate that the proposed regularization corresponds to the maximum likelihood estimation of an energy-based model on a hypersphere. Utilizing ImageNet-1k OoD benchmark datasets, we show that our method, combined with post-hoc OoDD approaches leveraging pretrained knowledge (e.g., NegLabel), significantly outperforms existing methods, achieving state-of-the-art OoDD performance and leading ID accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。