用动态文本分组提升LLM在图文图上的零样本推理能力
Dynamic Bundling with Large Language Models for Zero-Shot Inference on Text-Attributed Graphs
- 将邻近节点文本打包后提问LLM,获得整体标签
- 用批量标签监督GNN训练,显著提升分类准确率
- 适合缺乏标注数据的图文图场景,尤其关注零样本应用
大型语言模型(LLMs)在零样本学习中表现出强大的泛化能力。近期,将LLMs应用于文本属性图(TAGs)受到越来越多关注。然而,其应用面临两大挑战:图结构信息有限和响应不可靠。LLMs难以处理脱离图拓扑的孤立文本属性,且因信息不足和自身缺陷(如幻觉)导致预测不可靠。为此,本文提出一种新方法DENSE(Dynamic Text Bundling Supervision),通过向LLM查询文本包以获取包级标签,并用这些标签监督图神经网络训练。具体地,采样一组包含邻近节点及其文本的包,向LLM提问以获取每个包的标签;随后,利用包标签监督图神经网络优化,并进一步精炼包以剔除噪声项。我们还提供了该方法的理论分析。在十个数据集上的大量实验验证了所提方法的有效性。
原文摘要 · Abstract (English)
Large language models (LLMs) have been used in many zero-shot learning problems, with their strong generalization ability. Recently, adopting LLMs in text-attributed graphs (TAGs) has drawn increasing attention. However, the adoption of LLMs faces two major challenges: limited information on graph structure and unreliable responses. LLMs struggle with text attributes isolated from the graph topology. Worse still, they yield unreliable predictions due to both information insufficiency and the inherent weakness of LLMs (e.g., hallucination). Towards this end, this paper proposes a novel method named Dynamic Text Bundling Supervision (DENSE) that queries LLMs with bundles of texts to obtain bundle-level labels and uses these labels to supervise graph neural networks. Specifically, we sample a set of bundles, each containing a set of nodes with corresponding texts of close proximity. We then query LLMs with the bundled texts to obtain the label of each bundle. Subsequently, the bundle labels are used to supervise the optimization of graph neural networks, and the bundles are further refined to exclude noisy items. To justify our design, we also provide theoretical analysis of the proposed method. Extensive experiments across ten datasets validate the effectiveness of the proposed method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。