arrow
Return

Towards context-aware graph representation learning: Adaptive node aggregation with LLMs

delete2026-09-01
delete0
PRE
AI
S
Songwei Zhao
Y
Yuan Jiang *
Z
Zhang, Sinuo
J
Jifeng Hu
P
Philip S. Yu
陈贺昌 cover
陈贺昌 (Hechang Chen) *
DOI:10.1016/j.artint.2026.104584delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Representation learning on graph-structured data is a crucial research area in graph machine learning, with widespread application in real-world domains such as social networks and recommender systems. Traditional graph neural networks (GNNs) typically rely on static graph aggregation functions (e.g., mean, sum) applied over shallow textual features, which limits their ability to model complex semantic interactions among nodes. Recent advances have introduced large language models (LLMs) to enhance node features through deep semantic encoding. However, most existing work treats LLMs as static encoders, overlooking their potential to reason about inter-node relationships. In this work, we propose LLM-based Tailored Node Aggregation (LTNA), a novel approach that leverages the in-context reasoning capabilities of LLMs to generate adaptive, semantics-driven aggregation strategies for each node. Rather than relying on a fixed rule or global function, LTNA prompts the LLM to infer the relational meaning between a node and its neighbors (e.g., supportive, critical, background) and outputs a customed aggregation strategy based on the inferred semantics. This enables message passing in GNNs to move beyond numeric aggregation and become a context-sensitive semantic operation. By jointly leveraging node textual attributes and graph structure, LTNA enhances the expressiveness of node representations and improves performance across downstream tasks. Extensive experiments on five benchmark datasets demonstrate that LTNA consistently outperforms state-of-the-art GNN models, achieving an average improvement of 7.95% in classification accuracy and 9.90% in Micro-F1 score.
Keywords:
Text-attributed graph
Graph neural network
Large language model
Context-aware aggregation

Journal

A
Artificial Intelligence
IF:
4.6
Papers:
103
Citations:
1

Organization

U
university of illinois chicago hospital
Scholars:
1.1W
Papers: 8.8K
Citations: 16
N
nanyang technological university
Scholars:
969
Papers: 488
Citations: 0
U
University of Illinois Chicago
Scholars:
345
Papers: 153
Citations: 0
researcher View more organizations
Cited Papers

Cited Papers

No cited papers available