Knowledge Commons of Institute of Automation,CAS
Heterogeneous Relational Graph Neural Networks with Adaptive Objective for End-to-End Task-Oriented Dialogue | |
其他题名 | Heterogeneous Relational Graph Neural Networks with Adaptive Objective for End-to-End Task-Oriented Dialogue |
Liu, Qingbin1,2; Bai, Guirong1,2; He, Shizhu1,2; Liu, Cao3; Liu, Kang1,2; Zhao, Jun1,2 | |
发表期刊 | KNOWLEDGE-BASED SYSTEMS |
ISSN | 0950-7051 |
2021-09-05 | |
卷号 | 227期号:2021页码:107186 |
摘要 | End-to-end task-oriented dialogue systems, which provide a natural and informative way for human- computer interaction, are gaining more and more attention. The main challenge of such dialogue systems is how to effectively incorporate external knowledge bases into the learning framework. However, existing approaches usually overlook the natural graph structure information in the knowledge base and the relevant information between the knowledge base and the dialogue history, which makes them deficient in handling the above challenge. Besides, existing methods ignore the entity imbalance problem and treat different entities in system responses indiscriminately, which limits the learning of hard target entities. To address the two challenges, we propose Heterogeneous Relational Graph Neural Networks with Adaptive Objective (HRGNN-AO) for end-to-end task-oriented dialogue systems. In the method, we explore effective heterogeneous relational graphs to jointly capture multi perspective graph structure information from the knowledge base and the dialogue history, which ultimately facilitates the generation of informative responses. Moreover, we design two components, shared-private parameterization and hierarchical attention mechanism, to solve the overfitting and confusion problems in the heterogeneous relational graph, respectively. To handle the entity imbalance problem, we propose an adaptive objective, which dynamically adjusts the weights of different target entities during the training process. The experimental results show that HRGNN-AO is effective in generating informative responses and outperforms state-of-the-art dialogue systems on the SMD and extended Multi-WOZ 2.1 datasets. (c) 2021 Elsevier B.V. All rights reserved. |
关键词 | End-to-end task-oriented dialogue Heterogeneous relational graph neural networks Shared-private parameterization Hierarchical attention mechanism Adaptive objective |
DOI | 10.1016/j.knosys.2021.107186 |
收录类别 | SCI |
语种 | 英语 |
资助项目 | National Key Research and Development Program of China[2020AAA0106400] ; National Natural Science Foundation of China[61922085] ; National Natural Science Foundation of China[61976211] ; Beijing Academy of Artificial Intelligence[BAAI2019QN0301] ; Key Research Program of the Chinese Academy of Sciences[ZDBS-SSW-JSC006] ; National Laboratory of Pattern Recognition, China ; Youth Innovation Promotion Association CAS, China ; Meituan-Dianping Group |
项目资助者 | National Key Research and Development Program of China ; National Natural Science Foundation of China ; Beijing Academy of Artificial Intelligence ; Key Research Program of the Chinese Academy of Sciences ; National Laboratory of Pattern Recognition, China ; Youth Innovation Promotion Association CAS, China ; Meituan-Dianping Group |
WOS研究方向 | Computer Science |
WOS类目 | Computer Science, Artificial Intelligence |
WOS记录号 | WOS:000679379400017 |
出版者 | ELSEVIER |
七大方向——子方向分类 | 自然语言处理 |
引用统计 | |
文献类型 | 期刊论文 |
条目标识符 | http://ir.ia.ac.cn/handle/173211/45582 |
专题 | 多模态人工智能系统全国重点实验室_自然语言处理 |
通讯作者 | Zhao, Jun |
作者单位 | 1.National Laboratory of Pattern Recognition, Institute of Automation, Chinese Academy of Sciences, Beijing, 100190, China 2.School of Artificial Intelligence, University of Chinese Academy of Sciences, Beijing, 100049, China 3.Meituan, Beijing, 100102, China |
第一作者单位 | 模式识别国家重点实验室 |
通讯作者单位 | 模式识别国家重点实验室 |
推荐引用方式 GB/T 7714 | Liu, Qingbin,Bai, Guirong,He, Shizhu,et al. Heterogeneous Relational Graph Neural Networks with Adaptive Objective for End-to-End Task-Oriented Dialogue[J]. KNOWLEDGE-BASED SYSTEMS,2021,227(2021):107186. |
APA | Liu, Qingbin,Bai, Guirong,He, Shizhu,Liu, Cao,Liu, Kang,&Zhao, Jun.(2021).Heterogeneous Relational Graph Neural Networks with Adaptive Objective for End-to-End Task-Oriented Dialogue.KNOWLEDGE-BASED SYSTEMS,227(2021),107186. |
MLA | Liu, Qingbin,et al."Heterogeneous Relational Graph Neural Networks with Adaptive Objective for End-to-End Task-Oriented Dialogue".KNOWLEDGE-BASED SYSTEMS 227.2021(2021):107186. |
条目包含的文件 | ||||||
文件名称/大小 | 文献类型 | 版本类型 | 开放类型 | 使用许可 | ||
KBS-ours.pdf(1021KB) | 期刊论文 | 出版稿 | 开放获取 | CC BY-NC-SA | 浏览 下载 |
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。
修改评论