用大模型提升异构图建模效率,兼顾精度与速度。
Towards Efficient LLM-aware Heterogeneous Graph Learning
- 用大模型编码多跳多类型关系,捕捉复杂语义
- 提出分层图变换器,计算复杂度从指数降为线性
- 适配大模型的细粒度提示工程,提升任务匹配度
异构图广泛存在于真实世界的复杂网络中,节点和关系类型的多样性带来了丰富但复杂的语义。现有方法受限于预定义语义依赖和监督信号稀缺,而基于图结构的自监督预训练虽缓解此问题,却引入了任务间语义鸿沟。大语言模型(LLMs)凭借强大的文本推理能力,在解决关系与任务语义问题上具有潜力,但其在异构图中的应用受制于计算复杂度。为此,本文提出高效的大模型感知框架ELLA。通过设计大模型感知的关系标记器,利用LLM编码多跳、多类型关系;进一步引入分层关系图变换器,将大模型感知关系推理的复杂度从指数级降低至线性。为弥合预训练与微调任务间的语义差距,提出细粒度任务感知的文本思维链(CoT)提示。在四个异构图数据集上的实验表明,ELLA在性能与效率上均优于现有方法,可扩展至130亿参数的LLM,相比现有基于大模型的方法提速最高达4倍。代码已开源:https://github.com/l-wd/ELLA。
原文摘要 · Abstract (English)
Heterogeneous graphs are widely present in real-world complex networks, where the diversity of node and relation types leads to complex and rich semantics. Efforts for modeling complex relation semantics in heterogeneous graphs are restricted by the limitations of predefined semantic dependencies and the scarcity of supervised signals. The advanced pre-training and fine-tuning paradigm leverages graph structure to provide rich self-supervised signals, but introduces semantic gaps between tasks. Large Language Models (LLMs) offer significant potential to address the semantic issues of relations and tasks in heterogeneous graphs through their strong reasoning capabilities in textual modality, but their incorporation into heterogeneous graphs is largely limited by computational complexity. Therefore, in this paper, we propose an Efficient LLM-Aware (ELLA) framework for heterogeneous graphs, addressing the above issues. To capture complex relation semantics, we propose an LLM-aware Relation Tokenizer that leverages LLM to encode multi-hop, multi-type relations. To reduce computational complexity, we further employ a Hop-level Relation Graph Transformer, which help reduces the complexity of LLM-aware relation reasoning from exponential to linear. To bridge semantic gaps between pre-training and fine-tuning tasks, we introduce the fine-grained task-aware textual Chain-of-Thought (CoT) prompts. Extensive experiments on four heterogeneous graphs show that our proposed ELLA outperforms state-of-the-art methods in the performance and efficiency. In particular, ELLA scales up to 13b-parameter LLMs and achieves up to a 4x speedup compared with existing LLM-based methods. Our code is publicly available at https://github.com/l-wd/ELLA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。