Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Large Language Model as Attributed Training Data Generator: A Tale of Diversity and Bias

About

Large language models (LLMs) have been recently leveraged as training data generators for various natural language processing (NLP) tasks. While previous research has explored different approaches to training models using generated data, they generally rely on simple class-conditional prompts, which may limit the diversity of the generated data and inherit systematic biases of LLM. Thus, we investigate training data generation with diversely attributed prompts (e.g., specifying attributes like length and style), which have the potential to yield diverse and attributed generated data. Our investigation focuses on datasets with high cardinality and diverse domains, wherein we demonstrate that attributed prompts outperform simple class-conditional prompts in terms of the resulting model's performance. Additionally, we present a comprehensive empirical study on data generation encompassing vital aspects like bias, diversity, and efficiency, and highlight three key observations: firstly, synthetic datasets generated by simple prompts exhibit significant biases, such as regional bias; secondly, attribute diversity plays a pivotal role in enhancing model performance; lastly, attributed prompts achieve the performance of simple class-conditional prompts while utilizing only 5\% of the querying cost of ChatGPT associated with the latter. The data and code are available on \url{https://github.com/yueyu1030/AttrPrompt}.

Yue Yu, Yuchen Zhuang, Jieyu Zhang, Yu Meng, Alexander Ratner, Ranjay Krishna, Jiaming Shen, Chao Zhang• 2023

Related benchmarks

TaskDatasetResultRank
Sentiment ClassificationSST2 (test)
Accuracy83.64
233
Sentiment ClassificationIMDB (test)--
144
Topic ClassificationAG News (test)
Accuracy82.74
128
Sentiment ClassificationYelp (test)
Accuracy90.6
46
Untargeted Reconstruction AttackChatDoctor 250 adversarial prompts
Repeat Prompt0.00e+0
42
Retrieval-based Question AnsweringChatDoctor
ROUGE-L7.3
23
Synthetic Data GenerationIMDB (test)
MAUVE0.566
20
Topic ClassificationNYT (test)
Accuracy81.3
18
Synthetic Data GenerationSST-2 (test)
FID3.525
7
Synthetic Data GenerationAGNews (test)
FID2.193
7
Showing 10 of 19 rows

Other info

Code

Follow for update