arXiv:2501.09126cs.CLcs.CY2025-01中稿 · the Second Worksho…被引 3

用LLM生成数据混合人工标注,提升文本分类模型性能。

Augmenting Human-Annotated Training Data with Large Language Model Generation and Distillation in Open-Response Assessment

  • 混合人工与LLM生成数据,蒸馏至小型BERT模型。
  • 80%合成数据+20%人工数据时效果最佳,性能提升显著。
  • 低温生成更稳定,高温带来多样性但可能降低性能。

大型语言模型(如GPT-4o)可低成本规模化自动化文本分类任务,但其输出的有效性和可靠性存疑。相比之下,人工标注虽更可靠,但难以大规模获取。本文提出一种混合方案:将人工标注数据与LLM生成的合成数据结合,用于微调经典机器学习分类器,并将其蒸馏至小型BERT模型。在人工标注测试集上评估以衡量LLM输出质量。三个实验系统性地调整了LLM生成样本的数量、多样性和一致性,依据最佳实践进行调优。结果表明,加入合成数据可提升分类器性能,最优比例为80%合成数据与20%人工数据。低温设置(0.3)生成更一致的输出,带来更稳定的性能提升,但限制了模型对新增样本的学习;高温设置(0.7及以上)则引入更大性能波动,有时导致性能下降。过滤不一致合成样本并未提升表现。结论:融合人工与LLM生成数据是提升评估中文本分类模型的可扩展解决方案,兼顾准确性与多样性。

原文摘要 · Abstract (English)

Large Language Models (LLMs) like GPT-4o can help automate text classification tasks at low cost and scale. However, there are major concerns about the validity and reliability of LLM outputs. By contrast, human coding is generally more reliable but expensive to procure at scale. In this study, we propose a hybrid solution to leverage the strengths of both. We combine human-coded data and synthetic LLM-produced data to fine-tune a classical machine learning classifier, distilling both into a smaller BERT model. We evaluate our method on a human-coded test set as a validity measure for LLM output quality. In three experiments, we systematically vary LLM-generated samples' size, variety, and consistency, informed by best practices in LLM tuning. Our findings indicate that augmenting datasets with synthetic samples improves classifier performance, with optimal results achieved at an 80% synthetic to 20% human-coded data ratio. Lower temperature settings of 0.3, corresponding to less variability in LLM generations, produced more stable improvements but also limited model learning from augmented samples. In contrast, higher temperature settings (0.7 and above) introduced greater variability in performance estimates and, at times, lower performance. Hence, LLMs may produce more uniform output that classifiers overfit to earlier or produce more diverse output that runs the risk of deteriorating model performance through information irrelevant to the prediction task. Filtering out inconsistent synthetic samples did not enhance performance. We conclude that integrating human and LLM-generated data to improve text classification models in assessment offers a scalable solution that leverages both the accuracy of human coding and the variety of LLM outputs.

文本分类LLM生成数据增强BERT蒸馏

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。