Search papers, labs, and topics across Lattice.
This paper introduces OMG-VLM, a unified framework that utilizes vision-language models (VLMs) to learn from attributed graphs across diverse modalities, addressing the limitations of existing models that require separate architectures for different attribute types. By employing structure-aware graph adapters, OMG-VLM effectively integrates neighborhood information while leveraging the pretrained VLM's embedding space, enabling it to handle text-only, image-only, and mixed attribute graphs within a single model. Experimental results demonstrate that OMG-VLM outperforms state-of-the-art graph neural network (GNN) and large language model (LLM) baselines in tasks such as node classification and link prediction, showcasing its robust generalization capabilities across varying graph structures.
A single model can effectively learn from heterogeneous attributed graphs, outperforming specialized approaches in critical graph learning tasks.
Vision-language models (VLMs) provide a unified representation space for textual and visual information, yet their potential as general-purpose backbones for graph-structured data remains largely unexplored. In practice, attributed graphs exhibit substantial modality heterogeneity: some graphs contain only textual node attributes, others only visual attributes, while still others provide both. Existing graph learning approaches are typically designed for fixed modality schemas, requiring separate models for different settings and limiting scalability and cross-graph generalization. To bridge this gap, we present OMG-VLM (One Model, Many Graphs with Vision-Language Models), a unified framework for learning over attributed graphs across heterogeneous modality schemas. OMG-VLM leverages a pretrained VLM as a shared backbone and introduces structure-aware graph adapters that integrate neighborhood information while remaining compatible with the VLM's native embedding space. This design enables effective learning over text-attributed, image-attributed, and multi-attributed graphs within a single model. Extensive experiments across diverse domains show that OMG-VLM consistently outperforms state-of-the-art GNN- and LLM-based baselines on attributed graph learning tasks such as node classification and link prediction, while exhibiting strong generalization to unseen graphs and varying modality schemas. The source code is available at https://github.com/Jo-eyang/OMG-VLM.
