arXiv:2410.16736cs.CL2024-10被引 6

用让大模型出错的提问生成训练数据,提升其真实表现

Forewarned is Forearmed: Leveraging LLMs for Data Synthesis through Failure-Inducing Exploration

  • 训练专用生成器,主动设计让模型答错的问题
  • 在安全、诚实、数学任务上效果优于人工或普通生成数据
  • 适合想提升模型鲁棒性的研究者和工程师

大型语言模型(LLMs)得益于多样且高质量的任务特定数据,在诸多下游应用中表现出色。当前方法通常依赖人工标注数据或预设任务模板来引导大模型生成相关训练数据,但这种对人工设计成分的依赖可能限制生成数据的范围,忽略关键边缘情况或新型挑战场景。本文提出一种新方法 ReverseGen,可自动生成能暴露大模型弱点的有效训练样本。具体而言,我们训练一个专用生成器,专门生成导致目标模型产生不理想回答的查询。这些诱发失败的查询被用于构建训练数据,有助于弥补模型缺陷并提升整体性能。该方法灵活适配不同规模模型(3B、7B、8B)。我们在三个核心应用(安全、诚实、数学)上评估 ReverseGen,结果表明生成数据兼具高效性与多样性。经 ReverseGen 数据微调的模型始终优于使用人工标注或通用模型生成数据训练的模型,为任务特定大模型增强提供了新思路。

原文摘要 · Abstract (English)

Large language models (LLMs) have significantly benefited from training on diverse, high-quality task-specific data, leading to impressive performance across a range of downstream applications. Current methods often rely on human-annotated data or predefined task templates to direct powerful LLMs in synthesizing task-relevant data for effective model training. However, this dependence on manually designed components may constrain the scope of generated data, potentially overlooking critical edge cases or novel scenarios that could challenge the model. In this paper, we present a novel approach, ReverseGen, designed to automatically generate effective training samples that expose the weaknesses of LLMs. Specifically, we introduce a dedicated proposer trained to produce queries that lead target models to generate unsatisfactory responses. These failure-inducing queries are then used to construct training data, helping to address the models' shortcomings and improve overall performance. Our approach is flexible and can be applied to models of various scales (3B, 7B, and 8B). We evaluate ReverseGen on three key applications (safety, honesty, and math), demonstrating that our generated data is both highly effective and diverse. Models fine-tuned with ReverseGen-generated data consistently outperform those trained on human-annotated or general model-generated data, offering a new perspective on data synthesis for task-specific LLM enhancement.

数据合成模型鲁棒性大模型训练

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