Improving LLM Reasoning with Homophily-aware Structural and Semantic Text-Attributed Graph Compression
Zijun Di, Bin Lu, Huquan Kang, Luoyi Fu, Jiaxin Ding, Xiaoying Gan, Lei Zhou, Xinbing Wang, Chenghu Zhou
TL;DR
This work introduces HS2C, a homophily-aware compression framework for text-attributed graphs (TAGs) to enhance large language model (LLM) reasoning under strict input contexts. It combines an entropy-guided, hierarchical structure partition to uncover cohesive, homophilic communities with a semantic-aggregation step that distills background text per community, producing a compact yet informative graph for downstream tasks. Empirical results across ten node-level benchmarks show substantial improvements in compression efficiency and accuracy, and generalization to seven graph-level tasks, highlighting scalability and robustness across model families and graph domains. By aligning structural and semantic information with target nodes, HS2C offers a principled, scalable path to reliable LLM-based reasoning on large TAGs, reducing input redundancy while preserving critical context.
Abstract
Large language models (LLMs) have demonstrated promising capabilities in Text-Attributed Graph (TAG) understanding. Recent studies typically focus on verbalizing the graph structures via handcrafted prompts, feeding the target node and its neighborhood context into LLMs. However, constrained by the context window, existing methods mainly resort to random sampling, often implemented via dropping node/edge randomly, which inevitably introduces noise and cause reasoning instability. We argue that graphs inherently contain rich structural and semantic information, and that their effective exploitation can unlock potential gains in LLMs reasoning performance. To this end, we propose Homophily-aware Structural and Semantic Compression for LLMs (HS2C), a framework centered on exploiting graph homophily. Structurally, guided by the principle of Structural Entropy minimization, we perform a global hierarchical partition that decodes the graph's essential topology. This partition identifies naturally cohesive, homophilic communities, while discarding stochastic connectivity noise. Semantically, we deliver the detected structural homophily to the LLM, empowering it to perform differentiated semantic aggregation based on predefined community type. This process compresses redundant background contexts into concise community-level consensus, selectively preserving semantically homophilic information aligned with the target nodes. Extensive experiments on 10 node-level benchmarks across LLMs of varying sizes and families demonstrate that, by feeding LLMs with structurally and semantically compressed inputs, HS2C simultaneously enhances the compression rate and downstream inference accuracy, validating its superiority and scalability. Extensions to 7 diverse graph-level benchmarks further consolidate HS2C's task generalizability.
