UTAG: Leveraging LLM as a Unified Embedding Generator for Text-Attributed Graphs
Mingqian Ding, Jianjun Li, Zhiyuan Ma, Liwei Zhang, Wenqi Yang
Abstract
Large Language Models (LLMs) have shown significant potential in handling text-attributed graphs (TAGs). Nevertheless, existing LLM-enhanced approaches for TAG learning mainly utilize LLMs as text augmenters, which not only incur substantial computational overhead but may also introduce noise. In contrast, their capacity as embedding generators remains underutilized. Furthermore, most current methods are designed for specific TAG instances and lack cross-domain transferability. To overcome these limitations, we propose UTAG, a novel framework that leverages LLMs as unified embedding generators. First, we design a dual-view contrastive fine-tuning approach for cross-domain pretraining. The in-domain view promotes local consistency via adaptive positive sample selection, while the cross-domain view establishes alignment through subspace clustering-based cross-domain prototype aggregation. By jointly pretraining on multiple TAG datasets from diverse domains, we obtain a unified LLM encoder that produces transferable representations. Subsequently, we introduce a graph-specific adaptation training technique that implements two-stage dimensionality reduction with carefully designed semantic and Laplacian reconstruction constraints. This allows effective incorporation of structural signals while preserving semantic information to the greatest extent. Additionally, we employ cross-space alignment to further enhance consistency. Extensive experiments demonstrate that UTAG substantially surpasses existing LLM-enhanced methods, achieving an average improvement of 5.9% on cross-domain transfer tasks, which verifies its effectiveness and superiority.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Your agent calls
Lunesearch_papers
Free to start. No credit card required.
Terminal
Install the CLIlune papers get f6b9d6ad-3345-415e-9fe5-04dcf5f14e78Related papers
- GraphCLIP: Enhancing Transferability in Graph Foundation Models for Text-Attributed GraphsYun Zhu, Haizhou Shi, Xiaotang Wang, Yongchao Liu et al.WWW 2025 · 54 citations
- SSTAG: Structure-Aware Self-Supervised Learning Method for Text-Attributed GraphsRuyue Liu, Rong Yin, Xiangzhen Bo, Xiaoshuai Hao et al.NeurIPS 2025 · 5 citations
- Large Language Model Meets Graph Neural Network in Knowledge DistillationShengxiang Hu, Guobing Zou, Song Yang, Shiyi Lin et al.AAAI 2025 · 19 citations
- Quantizing Text-attributed Graphs for Semantic-Structural IntegrationJianyuan Bo, Hao Wu, Yuan FangKDD 2025
- Generalization Principles for Inference over Text-Attributed Graphs with Large Language ModelsHaoyu Peter Wang, Shikun Liu, Rongzhe Wei, Pan LiICML 2025
