arXiv:2502.00806cs.LG2025-02被引 61

统一多模态图嵌入空间,提升跨模态图学习效果

UniGraph2: Learning a Unified Embedding Space to Bind Multimodal Graphs

  • 用多模态编码器+GNN学习统一低维嵌入空间
  • 跨域多图预训练实现多任务性能显著提升
  • 适合多模态图数据的通用表示学习场景

现有基础模型如CLIP致力于在多模态数据上学习统一嵌入空间,支持搜索、推荐等下游应用。然而,这些模型常忽略多模态数据中的固有图结构,其中实体及其关系至关重要。多模态图(MMGs)指节点关联多源特征、边刻画实体间关系的图结构。现有图基础模型主要针对文本属性图(TAGs),难以处理MMGs复杂性。为此,我们提出UniGraph2,一种新型跨域图基础模型,可实现对MMGs的通用表征学习,提供统一嵌入空间。UniGraph2采用模态特异性编码器与图神经网络(GNN)结合,学习融合多模态信息与图结构的统一低维嵌入空间。设计了一种大规模跨域多图预训练算法,确保跨不同图域与模态的有效迁移。此外,引入专家混合(MoE)组件,对齐不同域与模态特征,保障嵌入的一致性与鲁棒性。大量实验证明,UniGraph2在表示学习、迁移学习及多模态生成任务中显著优于当前最优模型,为MMGs学习提供可扩展、灵活的解决方案。

原文摘要 · Abstract (English)

Existing foundation models, such as CLIP, aim to learn a unified embedding space for multimodal data, enabling a wide range of downstream web-based applications like search, recommendation, and content classification. However, these models often overlook the inherent graph structures in multimodal datasets, where entities and their relationships are crucial. Multimodal graphs (MMGs) represent such graphs where each node is associated with features from different modalities, while the edges capture the relationships between these entities. On the other hand, existing graph foundation models primarily focus on text-attributed graphs (TAGs) and are not designed to handle the complexities of MMGs. To address these limitations, we propose UniGraph2, a novel cross-domain graph foundation model that enables general representation learning on MMGs, providing a unified embedding space. UniGraph2 employs modality-specific encoders alongside a graph neural network (GNN) to learn a unified low-dimensional embedding space that captures both the multimodal information and the underlying graph structure. We propose a new cross-domain multi-graph pre-training algorithm at scale to ensure effective transfer learning across diverse graph domains and modalities. Additionally, we adopt a Mixture of Experts (MoE) component to align features from different domains and modalities, ensuring coherent and robust embeddings that unify the information across modalities. Extensive experiments on a variety of multimodal graph tasks demonstrate that UniGraph2 significantly outperforms state-of-the-art models in tasks such as representation learning, transfer learning, and multimodal generative tasks, offering a scalable and flexible solution for learning on MMGs.

多模态图统一嵌入GNN跨域学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。