arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.01771cs.CLcs.AIcs.NI
收藏

<SOG_k>: 一个LLM令牌用于显式图结构理解

<SOG_k>: One LLM Token for Explicit Graph Structural Understanding

Shanghai Jiao Tong University(上海交通大学);IGSNRR, Chinese Academy of Sciences(中科院信息融合国家重点实验室)

Jingyao Wu, Bin Lu, Zijun Di, Xiaoying Gan, Meng Jin, Luoyi Fu, Xinbing Wang, Chenghu Zhou

AI总结:

本文提出<SOG_k>令牌,通过统一令牌空间实现图结构的显式表示,提升LLM对图结构的理解、生成和推理能力。

摘要:

大型语言模型在无结构数据理解方面显示出巨大潜力,但仍然面临图结构的显著挑战,即结构幻觉。现有方法主要要么将图结构转化为自然语言,导致过度的令牌消耗和分散的注意力,要么将图结构转化为可训练的连续嵌入(即软提示),但表现出与原始文本令牌严重不一致。为了解决这个问题,我们提出在统一的令牌空间中引入一个特殊令牌<SOG_k>,以完全表示图结构内的结构,从而促进显式拓扑输入和结构信息共享。具体而言,我们提出了一种拓扑感知的结构分词器,将每个图拓扑映射到一个高度选择性的单个令牌。之后,我们构建了一组混合结构问题-答案语料库,以将新的结构令牌与现有的文本令牌对齐。通过这种方法,<SOG_k>使LLM能够以简洁和准确的方式理解和生成图结构。在五个图级别基准上的广泛实验表明,我们的方法优于现有方法,性能提升了9.9%到41.4%,并且具有可解释性和一致性。此外,我们的方法为节点级别任务提供了灵活的扩展,能够实现全局和局部结构理解。代码库可在https://github.com/Jingyao-Wu/SOG上公开获取。

英文摘要:

Large language models show great potential in unstructured data understanding, but still face significant challenges with graphs due to their structural hallucination. Existing approaches mainly either verbalize graphs into natural language, which leads to excessive token consumption and scattered attention, or transform graphs into trainable continuous embeddings (i.e., soft prompt), but exhibit severe misalignment with original text tokens. To solve this problem, we propose to incorporate one special token <SOG_k> to fully represent the Structure Of Graph within a unified token space, facilitating explicit topology input and structural information sharing. Specifically, we propose a topology-aware structural tokenizer that maps each graph topology into a highly selective single token. Afterwards, we construct a set of hybrid structure Question-Answering corpora to align new structural tokens with existing text tokens. With this approach, <SOG_k> empowers LLMs to understand, generate, and reason in a concise and accurate manner. Extensive experiments on five graph-level benchmarks demonstrate the superiority of our method, achieving a performance improvement of 9.9% to 41.4% compared to the baselines while exhibiting interpretability and consistency. Furthermore, our method provides a flexible extension to node-level tasks, enabling both global and local structural understanding. The codebase is publicly available at https://github.com/Jingyao-Wu/SOG.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑