arXiv:2601.05254cs.IRcs.CL2026-01ACL被引 2

用标签构建分层知识图谱,让小模型也能高效精准地回答领域问题。

TagRAG: Tag-guided Hierarchical Knowledge Graph Retrieval-Augmented Generation

  • 通过文档提取对象标签与关系,形成分层标签链结构化知识
  • 在跨领域数据集上平均胜率78.36%,构建效率比GraphRAG高14.6倍
  • 适合资源有限的场景,支持知识增量更新,小模型也可用

检索增强生成通过引入外部知识提升语言模型的回答质量。传统RAG依赖片段级检索,难以处理聚焦摘要的查询。GraphRAG采用图结构实现全局推理,但存在信息抽取低效、资源消耗大、难适应增量更新等问题。为此,我们提出TagRAG——一种标签引导的分层知识图谱RAG框架,兼顾高效全局推理与可扩展图谱维护。其核心包括:(1) 标签知识图谱构建,从文档中提取对象标签及其关系,组织为分层领域标签链以实现结构化表示;(2) 标签引导的检索增强生成,在推理时检索领域中心的标签链,定位并融合相关知识。该设计显著适配小型语言模型,提升检索粒度,并支持高效知识更新。在涵盖农业、计算机科学、法律及跨领域设置的UltraDomain数据集上的实验表明,TagRAG相较基线平均胜率达78.36%,且构建效率约为GraphRAG的14.6倍,检索效率提升1.9倍。

原文摘要 · Abstract (English)

Retrieval-Augmented Generation enhances language models by retrieving external knowledge to support informed and grounded responses. However, traditional RAG methods rely on fragment-level retrieval, limiting their ability to address query-focused summarization queries. GraphRAG introduces a graph-based paradigm for global knowledge reasoning, yet suffers from inefficiencies in information extraction, costly resource consumption, and poor adaptability to incremental updates. To overcome these limitations, we propose TagRAG, a tag-guided hierarchical knowledge graph RAG framework designed for efficient global reasoning and scalable graph maintenance. TagRAG introduces two key components: (1) Tag Knowledge Graph Construction, which extracts object tags and their relationships from documents and organizes them into hierarchical domain tag chains for structured knowledge representation, and (2) Tag-Guided Retrieval-Augmented Generation, which retrieves domain-centric tag chains to localize and synthesize relevant knowledge during inference. This design significantly adapts to smaller language models, improves retrieval granularity, and supports efficient knowledge increment. Extensive experiments on UltraDomain datasets spanning Agriculture, Computer Science, Law, and cross-domain settings demonstrate that TagRAG achieves an average winning rate of 78.36% against baselines while maintaining about 14.6x construction and 1.9x retrieval efficiency compared with GraphRAG.

知识图谱RAG标签引导小模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。