arXiv:2502.08661cs.CLcs.AI2025-02中稿 · www 2025被引 12

用分布匹配方法生成更贴近真实数据的合成文本,提升小模型性能。

Few-shot LLM Synthetic Data with Distribution Matching

  • 通过高斯过程追踪不确定性,动态选示范数据以增强多样性。
  • 合成数据经最大均值差异优化权重,使分布与真实数据一致。
  • 适合需要高质量合成数据的小模型训练场景。

随着大语言模型(LLMs)的发展,其上下文学习和少样本生成能力显著提升,促使人们利用LLM生成高质量合成数据以增强小型模型(如在线检索器或弱LLM)的表现。然而,LLM生成的合成数据在关键语言属性(如风格、语调、内容比例等)上常与真实数据存在差异,直接混合可能扭曲原始数据分布,反而阻碍性能提升。为此,我们提出SynAlign:一种基于关键属性分布匹配的合成数据生成与过滤框架。生成前,SynAlign使用高斯过程模型代理的不确定性追踪器,迭代选择与已选数据簇不同的数据簇作为新数据合成的示范,高效探索真实数据的多样性;随后,采用潜在属性推理方法,让LLM总结示范数据的语言属性并据此生成新数据,确保合成数据具备真实数据中的语言特征。生成后,使用最大均值差异(Maximum Mean Discrepancy)作为目标函数,学习每条合成数据的采样权重,确保其分布与真实数据匹配。多项文本预测任务实验表明性能显著提升,且在在线检索器上进行了A/B测试,验证了SynAlign的有效性。

原文摘要 · Abstract (English)

As large language models (LLMs) advance, their ability to perform in-context learning and few-shot language generation has improved significantly. This has spurred using LLMs to produce high-quality synthetic data to enhance the performance of smaller models like online retrievers or weak LLMs. However, LLM-generated synthetic data often differs from the real data in key language attributes (e.g., styles, tones, content proportions, etc.). As a result, mixing these synthetic data directly with real data may distort the original data distribution, potentially hindering performance improvements. To solve this, we introduce SynAlign: a synthetic data generation and filtering framework based on key attribute distribution matching. Before generation, SynAlign employs an uncertainty tracker surrogated by the Gaussian Process model to iteratively select data clusters distinct from selected ones as demonstrations for new data synthesis, facilitating the efficient exploration diversity of the real data. Then, a latent attribute reasoning method is employed: the LLM summarizes linguistic attributes of demonstrations and then synthesizes new data based on them. This approach facilitates synthesizing diverse data with linguistic attributes that appear in real data.After generation, the Maximum Mean Discrepancy is used as the objective function to learn the sampling weight of each synthetic data, ensuring distribution matching with the real data. Our experiments on multiple text prediction tasks show significant performance improvements. We also conducted an online A/B test on an online retriever to demonstrate SynAlign's effectiveness.

合成数据分布匹配少样本LLM

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