SHADE: Speaker-History-Aware Dialog Generation Through Contrastive and Prompt Learning
Futian Wang,Xuyang Zhao,Xiao Sun
DOI: https://doi.org/10.1109/tcss.2023.3297051
2024-01-01
IEEE Transactions on Computational Social Systems
Abstract:Traditional persona-based dialog generation models typically rely on textual descriptions of persona information, but this approach is limited by this and its inability to capture all aspects of a persona's speaking style. To solve this problem, we propose a novel dialog generation model called Speaker-History-Aware Dialogue GEneration (SHADE) through Contrastive and Prompt Learning, which utilizes contrastive learning to model speaking style from historical conversation, resulting in more personalized and distinctive response. Based on the idea of, Prompt learning, on the one hand, we embed small-scale trainable parameters according to Continuous Prompts to stimulate the knowledge of the pre-trained language model, so as to improve the feature extraction ability of the model. On the other hand, we add speaker tag to the decoder input according to Discrete Prompts to improve the consistency of speaking style. We evaluate our model on two popular text-only datasets, DailyDialog and PersonaChat. The results show that SHADE outperforms other multiturn hierarchical dialog generation models in terms of response consistency, logic, and diversity. SHADE is optimal for metrics PPL, NIST, ROUGEL, METEOR, BLUE2, etc. Our proposed SHADE offers a promising direction for persona-based dialog generation, addressing the limitation of existing approaches and paving the way for more personalized and engaging conversational agents in various applications.