arXiv:2501.09928cs.CLcs.AI2025-01被引 10

用知识图谱自动构建领域对话数据集,高效又省力。

Dialogue Benchmark Generation from Knowledge Graphs with Cost-Effective Retrieval-Augmented LLMs

  • 分阶段生成+检索增强,避免幻觉导致的返工
  • 在多个大模型上表现一致,超越现有最佳系统
  • 无需预处理全图,实时查询子图提升效率

对话基准对训练和评估特定领域聊天机器人至关重要。知识图谱(KG)如DBLP、DBpedia和YAGO蕴含语义丰富且结构清晰的数据。传统对话基准依赖人工创建,忽视了KG的自动化潜力;现有问答基准虽可自动生成,但不支持对话。本文提出Chatty-Gen,一种多阶段检索增强生成平台,基于KG自动构建高质量领域对话基准。该方法将生成过程分解为可控阶段,并使用断言规则实现阶段间自动验证,有效控制中间结果以防止因幻觉引发的耗时重启。通过基于上下文的高效查询检索,避免对整个KG进行预先处理。实验在多个真实大型KG上表明,Chatty-Gen显著优于当前最优系统,且在GPT-4o、Gemini 1.5、Llama 3和Mistral等多种能力模型上保持一致性能表现。

原文摘要 · Abstract (English)

Dialogue benchmarks are crucial in training and evaluating chatbots engaging in domain-specific conversations. Knowledge graphs (KGs) represent semantically rich and well-organized data spanning various domains, such as DBLP, DBpedia, and YAGO. Traditionally, dialogue benchmarks have been manually created from documents, neglecting the potential of KGs in automating this process. Some question-answering benchmarks are automatically generated using extensive preprocessing from KGs, but they do not support dialogue generation. This paper introduces Chatty-Gen, a novel multi-stage retrieval-augmented generation platform for automatically generating high-quality dialogue benchmarks tailored to a specific domain using a KG. Chatty-Gen decomposes the generation process into manageable stages and uses assertion rules for automatic validation between stages. Our approach enables control over intermediate results to prevent time-consuming restarts due to hallucinations. It also reduces reliance on costly and more powerful commercial LLMs. Chatty-Gen eliminates upfront processing of the entire KG using efficient query-based retrieval to find representative subgraphs based on the dialogue context. Our experiments with several real and large KGs demonstrate that Chatty-Gen significantly outperforms state-of-the-art systems and ensures consistent model and system performance across multiple LLMs of diverse capabilities, such as GPT-4o, Gemini 1.5, Llama 3, and Mistral.

对话生成知识图谱检索增强自动化评测

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