轻量级多模态大模型提升知识图谱补全效果
ELMM: Efficient Lightweight Multimodal Large Language Models for Multimodal Knowledge Graph Completion
- 用多视角图像压缩器减少视觉冗余,避免模态冲突
- 通过注意力剪枝降低计算开销,线性投影补偿性能损失
- 在四个数据集上达到顶尖效果,适合资源受限场景
多模态知识图谱(MKG)通过融合视觉与文本模态,实现更丰富的实体表征。然而现有MKG常存在不完整性,影响下游任务表现。因此,多模态知识图谱补全(MKGC)受到越来越多关注。尽管大语言模型(LLMs)在知识图谱补全(KGC)中表现优异,其在多模态场景下的应用仍较少。将多模态大语言模型(MLLMs)用于MKGC面临两大挑战:(1) 每个实体的图像标记数量庞大,导致语义噪声和模态冲突;(2) 处理大量输入标记带来高计算成本。为此,我们提出高效轻量级多模态大语言模型(ELMM)。ELMM设计基于多头注意力机制的多视图视觉标记压缩器(MVTC),可自适应地从文本与视觉视图中压缩图像标记,有效减少冗余并保留必要信息,避免模态冲突。此外,我们提出注意力剪枝策略,移除冗余注意力层以显著降低推理开销,并引入线性投影补偿剪枝带来的性能下降。在四个基准数据集上的广泛实验表明,ELMM实现了最先进的性能。
原文摘要 · Abstract (English)
Multimodal Knowledge Graphs (MKGs) extend traditional knowledge graphs by incorporating visual and textual modalities, enabling richer and more expressive entity representations. However, existing MKGs often suffer from incompleteness, which hinder their effectiveness in downstream tasks. Therefore, multimodal knowledge graph completion (MKGC) task is receiving increasing attention. While large language models (LLMs) have shown promise for knowledge graph completion (KGC), their application to the multimodal setting remains underexplored. Moreover, applying Multimodal Large Language Models (MLLMs) to the task of MKGC introduces significant challenges: (1) the large number of image tokens per entity leads to semantic noise and modality conflicts, and (2) the high computational cost of processing large token inputs. To address these issues, we propose Efficient Lightweight Multimodal Large Language Models (ELMM) for MKGC. ELMM proposes a Multi-view Visual Token Compressor (MVTC) based on multi-head attention mechanism, which adaptively compresses image tokens from both textual and visual views, thereby effectively reducing redundancy while retaining necessary information and avoiding modality conflicts. Additionally, we design an attention pruning strategy to remove redundant attention layers from MLLMs, thereby significantly reducing the inference cost. We further introduce a linear projection to compensate for the performance degradation caused by pruning. Extensive experiments on four benchmark datasets demonstrate that ELMM achieves state-of-the-art performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。