arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.03250cs.LGcs.AIcs.CL
收藏

DiffLM:通过扩散语言模型实现可控合成数据生成

DiffLM: Controllable Synthetic Data Generation via Diffusion Language Models

University of Chinese Academy of Sciences(中国科学院大学);Chinese Information Processing Laboratory, Institute of Software, Chinese Academy of Sciences(中国科学院软件研究所信息处理实验室);ByteDance Inc.(字节跳动公司)

Ying Zhou, Xinyao Wang, Yulei Niu, Yaojie Shen, Lexin Tang, Fan Chen, Ben He, Le Sun, Longyin Wen

AI总结:

本文提出基于VAE与潜变量扩散的DiffLM框架,通过即插即用特征注入解耦分布学习与LLM生成目标,提升结构化合成数据质量。

摘要:

近期,大语言模型(LLM)的进展显著增强了其知识与生成能力,引发了利用LLM进行高质量数据合成的浓厚兴趣。然而,由于LLM对目标数据分布的理解有限,且提示工程复杂,尤其是对于结构化格式数据,通过提示LLM进行合成数据生成仍具挑战。为解决这些问题,我们提出DiffLM,这是一个基于变分自编码器(VAE)的可控数据合成框架,它进一步:(1)利用扩散模型在学习到的潜变量分布中保留更多原始分布和格式结构的信息;(2)通过即插即用的潜变量特征注入模块,将目标分布知识的学习与LLM的生成目标解耦。由于我们观察到VAE的潜变量表示与真实数据分布之间存在显著差异,因此在框架中引入潜变量扩散模块,以学习表达能力充分的潜变量分布。在七个包含结构化格式数据(即表格、代码和工具数据)的真实世界数据集上的评估表明,DiffLM能够生成高质量数据,在某些情况下,其下游任务性能比真实数据高出2%-7%。数据和代码可在https://github.com/bytedance/DiffLM获取。

英文摘要:

Recent advancements in large language models (LLMs) have significantly enhanced their knowledge and generative capabilities, leading to a surge of interest in leveraging LLMs for high-quality data synthesis. However, synthetic data generation via prompting LLMs remains challenging due to LLMs' limited understanding of target data distributions and the complexity of prompt engineering, especially for structured formatted data. To address these issues, we introduce DiffLM, a controllable data synthesis framework based on variational autoencoder (VAE), which further (1) leverages diffusion models to reserve more information of original distribution and format structure in the learned latent distribution and (2) decouples the learning of target distribution knowledge from the LLM's generative objectives via a plug-and-play latent feature injection module. As we observed significant discrepancies between the VAE's latent representations and the real data distribution, the latent diffusion module is introduced into our framework to learn a fully expressive latent distribution. Evaluations on seven real-world datasets with structured formatted data (i.e., Tabular, Code, and Tool data) demonstrate that DiffLM generates high-quality data, with performance on downstream tasks surpassing that of real data by 2%-7% in certain cases. Data and code are available at https://github.com/bytedance/DiffLM.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑