用扩散模型双向恢复缺失模态,提升视觉语言模型鲁棒性
Enhancing Foundation VLM Robustness to Missing Modality: Scalable Diffusion for Bi-directional Feature Restoration
- 引入动态门控与跨模态互学习,实现语义一致的特征重建
- 零样本测试下在多个数据集上超越现有方法,适应不同缺失率
- 无需微调主干模型,可直接插入现成VLM中使用
视觉语言模型(VLM)通常假设推理时输入模态完整,但当某些模态缺失或不完整时,性能急剧下降。当前研究面临两大困境:基于提示的方法难以恢复关键缺失特征,损害VLM泛化能力;基于插补的方法缺乏有效引导,易生成语义无关噪声。为此,本文提出一种通用的缺失模态恢复策略,引入可插拔的中段训练模块——增强型扩散模型,以有效恢复缺失特征。核心创新包括:(I) 动态模态门控,自适应利用条件特征指导语义一致的特征生成;(II) 跨模态互学习机制,对齐双模型语义空间,实现双向对齐。该策略保持预训练VLM原有结构完整性,无需微调主干模型,显著提升对信息丢失的鲁棒性。零样本评估在多个基准数据集上持续优于现有基线,验证其在不同缺失率和条件下均具强鲁棒性与可扩展性。代码与模型将公开。
原文摘要 · Abstract (English)
Vision Language Model (VLM) typically assume complete modality input during inference. However, their effectiveness drops sharply when certain modalities are unavailable or incomplete. Current research on missing modality primarily faces two dilemmas: Prompt-based methods struggle to restore missing yet indispensable features and degrade the generalizability of VLM. Imputation-based approaches, lacking effective guidance, are prone to generating semantically irrelevant noise. Restoring precise semantics while sustaining VLM's generalization remains challenging. Therefore, we propose a general missing modality restoration strategy in this paper. We introduce an enhanced diffusion model as a pluggable mid-stage training module to effectively restore missing features. Our strategy introduces two key innovations: (I) Dynamic Modality Gating, which adaptively leverages conditional features to guide the generation of semantically consistent features; (II) Cross-Modal Mutual Learning mechanism, which bridges the semantic spaces of the dual models to achieve bi-directional alignment. Notably, our strategy maintains the original integrity of the pre-trained VLM, requiring no fine-tuning of the backbone models while significantly boosting resilience to information loss. Zero-shot evaluations across benchmark datasets demonstrate that our approach consistently outperforms existing baselines, establishing it as a robust and scalable extension that ensures VLM reliability across diverse missing rates and conditions. Our code and models will be publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。