Increasing Diversity While Maintaining Accuracy: Text Data Generation with Large Language Models and Human Interventions

• Published in: arXiv.org 2023-06
• Main Authors: Young Chung, John Joon, Kamar, Ece, Amershi, Saleema
• Format: Article
• Language: English
• Subjects: Accuracy; Datasets; Domains; Labels; Large language models; Model accuracy; Sampling
• ISSN: 2331-8422
• DOI: 10.48550/arxiv.2306.04140

Large language models (LLMs) can be used to generate text data for training and evaluating other models. However, creating high-quality datasets with LLMs can be challenging. In this work, we explore human-AI partnerships to facilitate high diversity and accuracy in LLM-based text data generation. We first examine two approaches to diversify text generation: 1) logit suppression, which minimizes the generation of languages that have already been frequently generated, and 2) temperature sampling, which flattens the token sampling probability. We found that diversification approaches can increase data diversity but often at the cost of data accuracy (i.e., text and labels being appropriate for the target domain). To address this issue, we examined two human interventions, 1) label replacement (LR), correcting misaligned labels, and 2) out-of-scope filtering (OOSF), removing instances that are out of the user's domain of interest or to which no considered label applies. With oracle studies, we found that LR increases the absolute accuracy of models trained with diversified datasets by 14.4%. Moreover, we found that some models trained with data generated with LR interventions outperformed LLM-based few-shot classification. In contrast, OOSF was not effective in increasing model accuracy, implying the need for future work in human-in-the-loop text data generation.