Controllable and Diverse Data Augmentation with Large Language Model for Low-Resource Open-Domain Dialogue Generation

Zhenhua Liu,Tong Zhu,Jianxiang Xiang,Wenliang Chen
2024-03-30
Abstract:Data augmentation (DA) is crucial to mitigate model training instability and over-fitting problems in low-resource open-domain dialogue generation. However, traditional DA methods often neglect semantic data diversity, restricting the overall quality. Recently, large language models (LLM) have been used for DA to generate diversified dialogues. However, they have limited controllability and tend to generate dialogues with a distribution shift compared to the seed dialogues. To maximize the augmentation diversity and address the controllability problem, we propose \textbf{S}ummary-based \textbf{D}ialogue \textbf{A}ugmentation with LLM (SDA). Our approach enhances the controllability of LLM by using dialogue summaries as a planning tool. Based on summaries, SDA can generate high-quality and diverse dialogue data even with a small seed dataset. To evaluate the efficacy of data augmentation methods for open-domain dialogue, we designed a clustering-based metric to characterize the semantic diversity of the augmented dialogue data. The experimental results show that SDA can augment high-quality and semantically diverse dialogues given a small seed dataset and an LLM, and the augmented data can boost the performance of open-domain dialogue models.
Computation and Language
What problem does this paper attempt to address?
The paper attempts to address the issue of how data augmentation methods can improve the diversity of dialogue data while maintaining the quality and consistency of the dialogue distribution in low-resource open-domain dialogue generation. Traditional data augmentation methods often overlook the diversity of semantic data, limiting overall quality; whereas existing large language models (LLMs) can generate diverse dialogues, their controllability is limited, and the distribution of generated dialogues may deviate from the seed dialogues. Therefore, this study proposes a dialogue augmentation method based on dialogue summaries (SDA), aiming to maximize the diversity of augmented data and address controllability issues. By using dialogue summaries as planning tools, SDA can generate high-quality and diverse dialogue data on a small-scale seed dataset. Additionally, to evaluate the effectiveness of the data augmentation method, the researchers designed a clustering-based metric to characterize the semantic diversity of the augmented dialogue data. Experimental results show that SDA can effectively enhance high-quality and semantically diverse dialogues given a small-scale seed dataset and a large language model, thereby improving the performance of open-domain dialogue generation models.