Proactive Retrieval-based Chatbots based on Relevant Knowledge and Goals
Yutao Zhu1, Jian-Yun Nie1, Kun Zhou2, Pan Du1, Hao Jiang3, and Zhicheng Dou4 1 Université de Montréal, Montréal, Québec, Canada 2 School of Information, Renmin University of China, Beijing, China 3 Huawei Poisson Lab., Hangzhou, Zhejiang, China 4 Gaoling School of Artificial Intelligence, Renmin University of China, Beijing, China [email protected],{nie,pandu}@iro.uimontreal.ca,[email protected] [email protected],[email protected]
ABSTRACT requirements, such as play music, set a clock, or show the weather A proactive dialogue system has the ability to proactively lead the forecast. These systems are perceived just as tools by users as they conversation. Different from the general chatbots which only re- only react passively. Users may be bored quickly. The problem is act to the user, proactive dialogue systems can be used to achieve even more severe in a chit-chat style dialogue system. Proactive some goals, e.g., to recommend some items to the user. Background conversation offers a solution to this problem [11, 12, 14]. A proac- knowledge is essential to enable smooth and natural transitions in tive dialogue system can lead the dialogue proactively to achieve dialogue. In this paper, we propose a new multi-task learning frame- some goals. For example, it can drive the conversation to educate work for retrieval-based knowledge-grounded proactive dialogue. the kids about some topics, to comfort a person, to place ads un- To determine the relevant knowledge to be used, we frame knowl- obtrusively, or to recommend items to users. Various application edge prediction as a complementary task and use explicit signals scenarios are emerging, yet there has not been a standard definition to supervise its learning. The final response is selected according of what proactive conversation should be. Variations have been to the predicted knowledge, the goal to achieve, and the context. observed in the way that a goal is defined: by a sentence17 [ ] or by Experimental results show that explicit modeling of knowledge some entities that should be covered in the conversation [12], and prediction and goal selection can greatly improve the final response whether the goal should be generated dynamically [11] or prede- selection. Our code is available at https://github.com/DaoD/KPN/. fined [14]. We are still in the early stage of exploration in which people test different approaches in different settings. This study is CCS CONCEPTS a contribution to this exploration. In this work, we follow the setting given by Wu et al. [12], where • Computing methodologies → Discourse, dialogue and prag- the goal is specified by a set of entities (topics) and the background matics. knowledge about these entities is provided in a knowledge graph. KEYWORDS The goal is to lead the conversation smoothly to mention the re- quired entities. The knowledge graph helps to generate paths of Proactive Dialogue; Retrieval-based Chatbot; Multi-task Learning conversation that appear natural. Despite its simplicity, this setting ACM Reference Format: has many potential applications in practice, in particular in conver- Yutao Zhu, Jian-Yun Nie, Kun Zhou, Pan Du, Hao Jiang, and Zhicheng Dou. sational recommendation systems [4, 5, 9], where some items can 2021. Proactive Retrieval-based Chatbots based on Relevant Knowledge be set in advance for recommendation. An example of proactive and Goals. In Proceedings of the 44th International ACM SIGIR Conference conversation in the movie domain is shown in Figure 1. The goal on Research and Development in Information Retrieval (SIGIR ’21), July 11– is defined by two entities (topics): the movie McDull: Rise of the 15, 2021, Virtual Event, Canada. ACM, New York, NY, USA, 5 pages. https: Rice Cooker and the star Bo Peng. The system is asked to cover //doi.org/10.1145/3404835.3463011 both entities during the conversation. By exploiting the knowledge graph, the system aims to naturally transit from one conversation 1 INTRODUCTION topic to another and eventually fulfill the pre-defined goal. From Microsoft Xiaoice, Apple Siri, to Google Assistant, dialogue In this work, we focus on the retrieval-based method due to systems have been widely applied in our daily life. In general, these its higher fluency. Although knowledge has been incorporated in systems are designed to make responses in reaction to the user’s some existing approaches to proactive conversation [12], it has been simply embedded in the response selection process [1, 6, 7], which Permission to make digital or hard copies of all or part of this work for personal or is optimized globally according to the loss on the final response classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation selection. Although the end-to-end training could be reasonable on the first page. Copyrights for components of this work owned by others than the with a very large amount of training data, in practice, the limited author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or training data may lead to sub-optimal solutions: when a wrong republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected]. response is selected by the system, it is hard to tell if it is due to a SIGIR ’21, July 11–15, 2021, Virtual Event, Canada poor knowledge prediction or a bad response selection, thus hard © 2021 Copyright held by the owner/author(s). Publication rights licensed to ACM. to optimize. ACM ISBN 978-1-4503-8037-9/21/07...$15.00 https://doi.org/10.1145/3404835.3463011 Context Knowledge (�) START McDull: Rise of the Rice Cooker Bo Peng Goal ! " {� } {� } Goal � Knowledge (�) (3rd turn) Representative Max Pooling th 6.9 Work Heze City {� } (5 turn) th Rating Native Place (7 turn) Nice, very 1 − ! Comment Lack of explosive × � ’ funny McDull: Rise of the Rice Cooker Bo Peng � ⋯ performance Comment Is released Blood type � �’ Tracking Yes Star Type O # !" Conversation (�, �) � ! ⋯ � #!"#$% � $& � (1) Bot Do you usually spend your weekend watching movies? (2) User Of course, could you recommend good films for me? I can watch it on this weekend. Cosine Cosine Cosine (3) Bot You can watch Rise of the Rice Cooker, which is rated 6.9. � " ℒ (4) User Sounds great, thanks. Task Knowledge Prediction $ � (5) Bot It is very nice and funny. Do you know who is the star of this movie? Predicting � & × " (KP Loss) (6) User I don’t know. Who is that? Context Predicted Knowledge Response Updated Goal (7) Bot It’s Bo Peng. Do you know him? People said that he was a kind of lacking explosive performance. # {� ! } {� " } (8) User I will know more about him later. � � LSTM
Figure 1: An example of proactive dialogue. The system is # ! " � {� } {� } {� } {� } � � � asked to exploit the background knowledge to lead the dia- Goal-Response � Matching logue and accomplish the goal. Score ⋯ ⋯