← 返回 2026-03-09

Mario:基于大语言模型的多模态图推理框架 Mario: Multimodal Graph Reasoning with Large Language Models

Yuanfu Sun, Kang Li, Pengkang Guo, Jiajin Liu, Qiaoyu Tan 📅 2026-03-05 👍 10 2026-07-13 08:35
图神经网络 多模态图学习 大语言模型 对比学习 指令调优

提出两阶段框架解决多模态图中的跨模态对齐和模态偏好问题

前置知识

多模态图(Multimodal Graph, MMG)

多模态图是一种结构化数据格式,其中每个节点同时具有文本描述和图像两种模态属性,节点之间的边表示实体间的关联关系。与传统的文本属性图不同,MMG中的节点级多模态信号需要被联合建模。例如在电商场景中,每个商品节点既有产品标题和描述(文本),也有商品图片(视觉),而商品间的共同购买关系构成图的边结构。

理解MMG的定义和特点是理解本文研究动机的基础,论文正是针对MMG中特有的跨模态不一致和模态偏好异构两个挑战提出解决方案。

视觉语言模型(Vision-Language Model, VLM)

视觉语言模型是能够同时处理和理解图像与文本的深度学习模型,典型代表如CLIP。CLIP通过对比学习在大规模图像-文本对上预训练,学习到共享的嵌入空间,使得语义相似的图像和文本在该空间中距离较近。VLM通常包含独立的图像编码器和文本编码器,输出固定维度的特征向量。

本文的Stage 1基于VLM进行图条件化的跨模态对齐,理解VLM的工作原理对于把握论文的方法设计至关重要。

对比学习(Contrastive Learning)

对比学习是一种自监督学习范式,通过拉近正样本对的表示距离、推远负样本对的表示距离来学习有意义的特征表示。InfoNCE是最常用的对比损失函数,形式为 $\mathcal{L} = -\sum_{v \in \mathcal{B}} \log \frac{e^{s(v,v)/\tau}}{\sum_{u \in \mathcal{B}} e^{s(v,u)/\tau}}$,其中 $s(\cdot)$ 是余弦相似度,$\tau$ 是温度参数。对比学习的核心在于如何定义正负样本对。

本文在Stage 1中使用双向InfoNCE损失进行跨模态对齐,理解对比学习的机制有助于理解论文如何实现结构感知的图像-文本对齐。

指令调优(Instruction Tuning)

指令调优是将大语言模型适配到下游任务的技术,通过结构化的提示模板将任务指令与输入结合,使LLM能够理解和执行特定任务。与传统微调不同,指令调优使用自然语言描述任务要求,可选地加入可学习的软 token 来增强适应性。给定输入序列 $X$,指令函数 $T$ 产生结构化提示 $T(X)$,可与可学习软 token $S$ 拼接形成增强提示。

论文的Stage 2采用模态自适应图指令调优机制,理解指令调优的基本原理对于理解MAPR路由器的设计至关重要。

LoRA(Low-Rank Adaptation)

LoRA是一种参数高效微调方法,通过在预训练模型的权重矩阵旁边添加低秩分解矩阵来实现微调,而无需更新原始模型参数。具体地,对于权重矩阵 $W$,LoRA添加低秩分解 $W + BA$,其中 $B \in \mathbb{R}^{d \times r}$,$A \in \mathbb{R}^{r \times d}$,$r \ll d$。这样可以显著减少可训练参数数量,同时保持模型性能。

论文在Stage 2中使用LoRA微调LLM主干网络,理解LoRA有助于评估方法的计算效率和参数开销。

研究动机

现有方法在处理多模态图(MMG)推理时存在两个核心问题。首先是弱跨模态一致性(Weak Cross-modal Consistency):在现实世界的MMG中,节点附带的图像并不总是文本的清晰视觉呈现,文本也不总是图像的忠实描述。例如,一件服装的文本描述可能侧重于材质和版型(如"柔软的棉质连帽衫,宽松廓形,适合秋季穿着"),而图像则强调颜色、风格和品牌标识,这些信息在文本中从未提及。现有方法简单地使用CLIP等VLM独立编码图像-文本对,然后将这些嵌入输入图神经网络,这种做法隐含地假设文本和视觉视图已经语义同步,但实际上这种假设在MMG中并不成立。如论文Figure 1(a)所示,冻结CLIP的文本和图像嵌入之间的余弦相似度较低,表明跨模态一致性很弱。

本文的目标是本文的具体目标是设计一个统一的框架,能够同时解决上述两个挑战:(1) 通过图拓扑引导的跨模态对齐来增强跨模态一致性,使得同一节点的文本和图像表示在潜在空间中更加接近;(2) 通过模态自适应机制,为每个节点及其邻域动态选择最具信息量的模态配置,从而充分利用多模态监督信号。量化目标包括在节点分类和链接预测任务上超越现有最先进的图模型,并在零样本迁移设置中实现高达1.6倍的性能提升。

与已有工作不同的是,本文的独特切入角度在于认识到MMG中的两个被忽视的关键现实:(1) 跨模态一致性可能是弱的,使得朴素的融合/对齐方法不可靠;(2) 模态偏好在不同节点间存在异质性。现有工作如MLaGA仅在将文本和图像融合到共享查询表示之后才进行对齐,隐式假设所有模态效用相等;Graph4MM主要针对模态缺失的MMG,忽略了每个节点同时具有文本和图像的常见场景。本文抓住了邻居节点在消歧模态语义或补充缺失信息方面的关键作用,特别是在模态仅部分重叠的情况下。如Figure 1(b)所示,在至少被一个模板正确分类的节点中,近30%仅被一个或两个模态特定模板覆盖,而非全部三个,这表明一刀切的提示策略未能充分利用可用的多模态监督信号。

核心方法

Mario框架采用两阶段架构,整体思路可以类比为「先对齐,再选择」。第一阶段类似于在嘈杂的多人会议中,先确保每个人说的话和他的动作能够对应起来(跨模态对齐),第二阶段则是在理解每个人表达的基础上,根据具体情境选择最有效的沟通方式(模态自适应)。技术路线如下:Stage 1使用图条件视觉语言模型(GVLM),通过双塔编码器架构和拓扑感知多模态混合器,在图拓扑的引导下进行细粒度的跨模态对比学习,学习到结构感知、模态一致的节点表示。Stage 2在这些对齐特征的基础上,构建多模态图上下文信号和提示模板库,训练一个轻量级的模态自适应提示路由器(MAPR),使其能够为每个节点选择最具信息量的模态配置,并与LLM联合训练以实现有效的多模态图推理。

Mario的核心创新点在于将图结构信息显式地注入到跨模态对齐和模态选择两个环节中,这是与已有方法最本质的区别。具体而言,创新体现在两个方面:第一,图条件视觉语言模型(GVLM):不同于CLIP等VLM独立处理每个节点的图像-文本对,GVLM通过拓扑感知多模态混合器将图拓扑信息注入到Transformer的token嵌入中。该混合器利用图感知位置偏置 $B_h$(基于最短路径距离桶的可学习标量)来编码节点间的结构角色,使得[CLS]摘要能够在图引导下关注所有token和patch。然后通过再注入(Reinjection)机制,将结构感知表示传播回token流,迭代精炼节点表示。第二,模态自适应提示路由器(MAPR):不同于现有GraphLLM使用固定的指令模板,MAPR是一个轻量级MLP,它融合节点的Stage 1多模态嵌入、1/2跳邻域上下文和对数度数项,输出模态选择概率 $p_v = [p_v^{(txt)}, p_v^{(vis)}, p_v^{(mm)}]^\top$。训练时采用性能加权的LLM损失加上KL正则项,将概率质量转移到损失较低的模板上,使得LLM能够利用信息丰富的模态同时降低噪声模态的权重。

方法步骤详情

Mario的完整流程包括以下步骤:(1) 模态编码:使用两个独立的L层Transformer分别处理文本和图像,初始嵌入来自预训练的CLIP编码器,[CLS]token嵌入 $h_v^{l,M}$ 作为节点表示。(2) 拓扑感知多模态混合:收集所有节点的[CLS]摘要构建节点-特征矩阵 $H_{\mathcal{G},M}^l$,通过多头注意力机制(包含图感知位置偏置 $B_h$)进行结构感知的表示学习,然后将结构感知表示 $\hat{h}_{v,M}^l$ 再注入到token流中替代原[CLS]嵌入,重复L层获得最终的结构感知嵌入 $h_v^{\text{text}}$ 和 $h_v^{\text{image}}$。(3) 跨模态对比学习:对结构感知的模态嵌入应用双向InfoNCE损失 $\mathcal{L}_{S1}$,其中同一节点的文本-图像对为正样本,跨节点组合为负样本。(4) 多模态图上下文信号构建:为每个节点 $v$ 创建特殊token $\langle GT_v \rangle$ 和 $\langle GI_v \rangle$,通过可学习投影层 $P$ 将Stage 1特征映射到LLM token嵌入空间,并选择Top-K邻居(基于余弦相似度)注入其多模态信号。(5) 提示模板库构建:为每个节点形成三种模态特定的特殊token组 $S_v^{\text{txt}}, S_v^{\text{vis}}, S_v^{\text{mm}}$,分别对应文本、视觉、多模态视图。(6) MAPR训练:路由器融合节点嵌入、邻域上下文和度数信息,输出模态选择概率 $p_v$;LLM对每个模板产生负对数损失 $\ell_v^{(k)}$,转化为性能后验 $q_v$;训练最小化复合损失 $\mathcal{L}_{S2}$,包含性能加权目标和KL正则项。(7) 推理:MAPR从软路由切换到硬策略,选择概率最高的模板 $k^* = \arg\max p_v^{(k)}$,仅将对应提示输入LLM。

技术新颖性

Mario的技术新颖性主要体现在以下几个方面:第一,提出了图条件视觉语言模型(GVLM)这一新的VLM范式,它在图拓扑引导下对齐图像和文本,产生对称的、结构感知的节点表示。这与现有VLM(如CLIP)独立处理图像-文本对的方式形成鲜明对比。GVLM通过拓扑感知多模态混合器将图结构注入到token级别的交互中,保留了细粒度的视觉细节,而GNN和MLP往往忽略了这种token级交互。第二,引入了模态自适应图指令调优这一新的调优方案,它打破了现有GraphLLMs对固定模态模板的依赖。通过轻量级路由器MAPR,为每个节点动态选择最具信息量的模态配置,实现了节点级的模态自适应。这与传统GraphLLM中所有节点共享同一指令模板的做法形成对比。第三,训练时采用概率加权的LLM损失加KL正则项的复合损失函数,实现了路由器和LLM的联合优化,路由器能够根据LLM的反馈学习到模态偏好。第四,推理时从软路由切换到硬策略,不增加额外计算开销,保持了与单模板流水线相同的推理效率。

Mario框架整体架构图
Figure 2: Mario框架整体架构图

实验结果

论文在四个多模态图数据集(Movies、Reddit、CDs、Arts)上进行了广泛的实验,涵盖节点分类(NC)和链接预测(LP)两个任务。核心发现如下:(1) 在单焦点设置下,Mario在所有数据集和两个任务上均取得最高准确率。例如,在CDs数据集上,节点分类准确率从最佳基线的56.45%提升到63.43%,提升了约7个百分点;链接预测准确率在四个数据集上平均提升4.73%。这些提升源于Stage 1的结构感知图像-文本对齐和Stage 2的模态自适应图指令调优的协同作用。(2) 与使用图像转文本增强的GraphLLM基线相比,Mario在节点分类上平均相对提升5.48%(对比LLaGA)、11.00%(对比GraphPrompter)和135.9%(对比GraphGPT),表明直接让LLM解读对齐的多模态特征比通过文本代理增强输入更有效。(3) 在混合训练设置下,Mario在所有数据集上仍保持领先,节点分类平均相对提升2.88%,链接预测平均相对提升2.57%,证明了其在联合训练场景下的泛化能力。(4) 在零样本迁移设置中,Mario展现出强大的推理能力:在Toys→Movies迁移中,节点分类准确率达到41.00%,是最佳基线MLaGA(24.95%)的1.64倍;在Toys+Movies→CDs迁移中,节点分类准确率为54.32%,是最佳基线的1.48倍;在Toys→Books迁移中,链接预测准确率为78.30%,是最佳基线的1.25倍。(5) 消融实验证实,GVLM在所有数据集上均优于GNN和MLP,在Movies数据集上相对平均增益高达5.15%,证明了细粒度结构化图像-文本对齐的重要性。(6) 训练效率方面,Mario的MAPR在Movies数据集上收敛速度比单模板变体快2.3倍,在Reddit上快1.3倍,虽然每个epoch的时间约为1.5-2倍,但整体训练时间相当。(7) 路由器选择的可视化显示,模态偏好在MMG中主要遵循同配模式,具有相同颜色(偏好模态)的节点往往形成小簇,某些区域被单一模态主导。

单焦点设置下四个MMG数据集的性能对比
Table 1: 单焦点设置下四个MMG数据集的性能对比
混合训练设置下四个多模态图数据集的性能对比
Table 2: 混合训练设置下四个多模态图数据集的性能对比
零样本迁移结果
Table 3: 零样本迁移结果
Stage 1模型消融实验
Table 4: Stage 1模型消融实验
Mario与单模板变体的训练曲线对比
Figure 3: Mario与单模板变体的训练曲线对比
Mario与不同单模态模板的性能对比
Figure 4: Mario与不同单模态模板的性能对比
路由器选择可视化
Figure 5: 路由器选择可视化
对齐多模态特征的t-SNE可视化
Figure 6: 对齐多模态特征的t-SNE可视化
投影层数量的敏感性分析
Figure 7: 投影层数量的敏感性分析
每跳邻居数量的敏感性分析
Figure 8: 每跳邻居数量的敏感性分析
查看结构化数据
任务指标本文基线提升
节点分类 Accuracy (%) 53.63 Graph4MM: 51.07 +2.56
节点分类 Accuracy (%) 95.30 LLaGA: 92.14 +3.16
节点分类 Accuracy (%) 63.43 MLaGA: 56.45 +6.98
节点分类 Accuracy (%) 92.13 LLaGA: 89.32 +2.81
链接预测 Accuracy (%) 93.90 Graph4MM: 90.24 +3.66
链接预测 Accuracy (%) 91.30 Graph4MM: 90.80 +0.50
链接预测 Accuracy (%) 92.70 LLaGA: 84.90 +7.80
链接预测 Accuracy (%) 89.96 MLaGA: 82.97 +6.99
零样本迁移 (Toys→Movies) NC Accuracy (%) 41.00 MLaGA: 24.95 1.64×
零样本迁移 (Toys+Movies→CDs) NC Accuracy (%) 54.32 MLaGA: 16.20 3.35×
零样本迁移 (Toys→Books) LP Accuracy (%) 78.30 MLaGA: 57.85 1.35×

局限与改进

尽管Mario取得了显著的性能提升,但仍存在一些局限性。首先,论文主要在电商和社交网络领域的数据集上进行实验,这些场景中的多模态数据主要是产品图片和文本描述,模态间的关联相对直接。对于更复杂的多模态场景(如医疗图像与临床报告、遥感图像与地理描述),Mario的有效性尚未得到验证。其次,Stage 1的GVLM需要对所有节点的[CLS] token进行注意力计算,复杂度为 $O(|V_s|^2 d)$,虽然论文指出通常只需1-2层即可达到对齐收敛,但对于大规模图(如Goodreads数据集包含68.5万节点),计算开销仍然可观。第三,Stage 2的MAPR在推理时采用硬路由策略,即每个节点只选择一个最优模态模板,这可能丢失其他模态中的互补信息,特别是在需要多模态证据融合的场景中。第四,论文的图上下文构建依赖于基于余弦相似度的Top-K邻居选择,这种选择策略主要依赖Stage 1学习到的特征质量,如果初始特征对齐不佳,可能影响邻居选择的准确性。第五,论文承认其方法在异配图(heterophilic graph)上的表现需要进一步验证,尽管实验显示在不同异配比例的数据集上均表现稳健,但异配比例从0.04(Reddit)到0.69(CDs)变化较大,极端异配场景下的表现仍有待研究。

独立分析的弱点

基于对论文的深入分析,我识别出以下几个弱点及其改进方向:(1) **邻居选择策略的局限性**:当前方法基于Stage 1特征的余弦相似度选择Top-K邻居,这种静态选择策略可能无法适应不同节点的局部结构特性。改进方向可以考虑引入动态邻居选择机制,例如根据节点的模态偏好和任务类型自适应调整邻居数量和选择标准。(2) **硬路由策略的信息丢失**:MAPR在推理时采用硬路由(选择概率最高的模板),这可能导致其他模态中的有用信息被丢弃。改进方向可以探索软路由与硬路由的混合策略,例如在训练后期逐渐从软路由过渡到硬路由,或者设计更精细的模态融合机制。(3) **缺乏显式的模态质量评估**:当前方法假设所有节点的文本和图像质量相当,但实际上某些节点的某个模态可能是噪声或缺失的。改进方向可以引入显式的模态质量评估模块,根据模态质量动态调整路由器的决策。(4) **计算效率的进一步优化**:虽然论文提到Stage 1只需1-2层,但对于大规模图,$O(|V_s|^2 d)$ 的复杂度仍然是瓶颈。改进方向可以考虑采用图采样技术(如ClusterGCN)或稀疏注意力机制来降低计算开销。(5) **对LLM主干的依赖性**:虽然消融实验显示Mario对不同LLM具有鲁棒性,但所有实验均基于相对较小的模型(最大13B参数)。对于更小的LLM(如3B或7B),性能可能会显著下降,这限制了方法在资源受限场景中的应用。

未来方向

论文作者和基于本文成果可延伸的未来研究方向包括:(1) **扩展到更复杂的图结构**:当前工作主要处理同构图,未来可以探索异构图(不同类型节点和边)和动态图(随时间演化的图)上的多模态推理。(2) **引入更多模态**:除了文本和图像,现实世界的数据往往包含更多模态(如音频、视频、表格数据),未来可以扩展Mario框架以支持任意数量的模态。(3) **探索更先进的路由机制**:当前MAPR是一个简单的MLP路由器,未来可以探索基于强化学习或元学习的路由策略,使路由器能够根据任务反馈动态调整。(4) **结合检索增强生成(RAG)**:将Mario与RAG技术结合,利用图结构进行更有效的上下文检索和知识增强。(5) **应用于更广泛的下游任务**:当前工作聚焦于节点分类和链接预测,未来可以扩展到图级别任务(如图分类、图匹配)和生成任务(如图生成、文本生成)。(6) **理论分析**:深入分析图结构如何影响跨模态对齐的质量,以及模态偏好与图拓扑之间的理论关系。(7) **实际部署优化**:研究如何在实际系统中高效部署Mario,包括模型压缩、量化和在线推理优化。

复现评估

论文在可复现性方面提供了较好的支持。**开源情况**:论文明确声明代码将在GitHub上公开(虽然具体链接在论文中以Mario代替,但预计会提供完整实现)。**数据集**:论文使用了6个公开可用的多模态图数据集(Movies、Toys、CDs、Arts、Reddit、Goodreads),这些数据集来自Amazon电商和社交网络领域,数据获取相对容易。**算力需求**:论文报告所有涉及LLM的实验均在两块A100-SXM4-80GB GPU上进行,Stage 1训练约需1 GPU小时,Stage 2训练约需4-6 GPU小时,总计约5-7 GPU小时,这对于学术研究和工业应用来说是可接受的计算开销。**复现难度**:中等。方法本身相对复杂(两阶段架构、多种组件),但论文提供了详细的超参数设置、训练细节和模板设计。主要挑战在于:(1) 需要部署LLaMA3.1-8B模型,需要足够的GPU显存;(2) 需要预训练的CLIP编码器提取初始特征;(3) 数据预处理和特征提取步骤较多。总体而言,具备深度学习和图学习经验的研究者应该能够在合理时间内复现论文结果。