Yu-Kai Lee
2023
Story Co-telling Dialogue Generation via Reinforcement Learning and Knowledge Graph
Yu-Kai Lee
|
Chia-Hui Chang
Proceedings of the 35th Conference on Computational Linguistics and Speech Processing (ROCLING 2023)
Story Co-telling Dialogue Generation based on Multi-Agent Reinforcement Learning and Story Highlights
Yu-Kai Lee
|
Chia-Hui Chang
Proceedings of the 21st Annual Workshop of the Australasian Language Technology Association
Retelling a story is one way to develop narrative skills in students, but it may present some challenges for English as Second Language (ESL) students who are learning new stories and vocabularies at the same time. The goal of this research is to develop a dialogue module for story co-telling for ESL students in order to help students to co-narrate an English story and enhance their narrative skills. However, story co-telling is a relatively underexplored and novel task. In order to understand the story content and select the right plot to continue the story co-telling based on the current dialogue, we utilize open domain information extraction techniques to construct a knowledge graph, and adopt multi-agent reinforcement learning methods to train two agents to select relevant facts from the knowledge graph and generate responses, jointly accomplishing the task of story co-telling. Compared to models that reply on chronological order, our model improves the performance from 67.0% to 70.8% through self-training with reward evaluation, achieving an increase of approximately 3.8%.
Search