HyperAIHyperAI

Command Palette

Search for a command to run...

基于Transformer的上下文知识库构建用于端到端任务导向对话系统

Yanjie Gou Yinjie Lei Lingqiao Liu Yong Dai Chunxu Shen

摘要

将知识库(Knowledge Base, KB)融入端到端任务导向对话系统具有挑战性,因为需要对知识库中的实体进行恰当表示,而该表示需同时融合实体所关联的知识库上下文与对话上下文。现有方法仅基于实体的部分知识库上下文进行表示,容易因信息丢失而导致表征效果不佳,进而对知识库推理与回复生成产生不利影响。为解决这一问题,本文提出通过动态感知所有相关实体及对话历史,实现对实体表征的充分上下文化。为此,我们提出一种上下文感知的记忆增强型Transformer框架(COntext-aware Memory Enhanced Transformer, COMET)。该框架将知识库视为一个序列,并引入一种新颖的记忆掩码(Memory Mask)机制,使实体在表征过程中仅关注与其相关的实体及对话历史,同时有效避免无关实体的干扰。通过大量实验验证,结果表明,所提出的COMET框架在性能上显著优于现有最先进方法。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供