用随机图模型提升视觉语言模型微调的鲁棒性与语义捕捉能力
Beyond Graph Model: Reliable VLM Fine-Tuning via Random Graph Adapter
- 引入随机图结构建模类别内描述多样性与类间关系
- 通过概率消息传递学习每个类别的上下文感知分布表示
- 适合需要高鲁棒性微调的视觉语言任务研究者
基于文本适配器的微调方法在将预训练视觉-语言模型(VLM)知识迁移至下游任务中展现出显著潜力。现有方法通常使用确定性文本特征适配器来优化每类文本表示,但各类别文本描述存在显著差异,这种多样性蕴含丰富的区分性语义信息,传统确定性适配器难以充分捕捉。同时,也亟需利用适配器中的类间关系。为此,本文提出将随机图模型引入VLM适配器,构建顶点随机知识图(VRKG),并设计顶点随机图适配器(VRGAdapter)。VRGAdapter首先通过VRKG同时建模每类内部描述多样性与跨类关系,再通过概率消息传播学习每个类别节点的上下文感知分布表示,最后采用重参数化采样函数实现文本适配器学习。该方法为更通用的适配器方案,可涵盖传统图基适配器作为特例。此外,为提升下游任务鲁棒性,提出不确定性引导的多分支融合(UMF)策略,动态集成多个预训练模型进行集成预测。在多个基准数据集上的大量实验验证了所提方法的有效性。
原文摘要 · Abstract (English)
Textual adapter-based tuning methods have shown significant potential in transferring knowledge from pre-trained Vision-Language Models (VLMs) to downstream tasks. Existing works generally employ the deterministic textual feature adapter to refine each category textual representation. However, due to inherent factors such as different attributes and contexts, there exists significant diversity in textual descriptions for each category. Such description diversity offers rich discriminative semantic knowledge that can benefit downstream visual learning tasks. Obviously, traditional deterministic adapter model cannot adequately capture this varied semantic information. Also, it is desirable to exploit the inter-class relationships in VLM adapter. To address these issues, we propose to exploit random graph model into VLM adapter and develop a novel Vertex Random Graph Adapter (VRGAdapter). VRGAdapter first models the inherent diverse descriptions of each category and inter-class relationships of different categories simultaneously by leveraging a Vertex Random Knowledge Graph (VRKG) model. Then, it employs probabilistic message propagation on VRKG to learn context-aware distribution representation for each class node. Finally, it adopts a reparameterized sampling function to achieve textual adapter learning. Note that, VRGAdapter provides a more general adapter solution that encompasses traditional graph-based adapter as a special case. In addition, to enable more robust performance for downstream tasks, we also introduce a new Uncertainty-guided Multi-branch Fusion (UMF) scheme that dynamically integrates multiple pre-trained models for ensemble prediction. Extensive experiments on multiple benchmark datasets demonstrate the effectiveness of our approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。