arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.02418cs.CLcs.AIcs.LG
收藏

通过合成损坏解码数据质量:代码数据的嵌入引导剪枝

Decoding Data Quality via Synthetic Corruptions: Embedding-guided Pruning of Code Data

UC Los Angeles(加州大学洛杉矶分校);FAIR at Meta(Meta 基础人工智能研究院);DatologyAI

Yu Yang, Aaditya K. Singh, Mostafa Elhoushi, Anas Mahmoud, Kushal Tirumala, Fabian Gloeckle, Baptiste Rozière, Carole-Jean Wu, Ari S. Morcos, Newsha Ardalani

AI总结:

本文提出 SCIP 方法,利用合成损坏分析低质量代码的嵌入特征并设计剪枝指标,在 HumanEval 和 MBPP 上优于现有嵌入方法,最高提升 3%。

摘要:

代码数据集通常从 GitHub 等多样且不受控的来源收集,可能存在质量问题,从而影响为代码生成优化的大型语言模型(LLM)的性能和训练效率。以往研究表明利用嵌入空间进行数据剪枝是有益的,但它们主要关注去重或增加多样性,并且集中在图像等其他模态上。我们的工作聚焦于使用嵌入来识别并移除“低质量”代码数据。首先,我们通过使用合成损坏来探索嵌入空间中“低质量”代码的特征。基于这些认识,我们设计了在嵌入空间中运行的新型剪枝指标,用于识别并移除 Stack 数据集中的低质量条目。我们在成熟的 HumanEval 和 MBPP 基准上展示了这种合成损坏引导剪枝(SCIP)方法的收益,其性能优于现有的基于嵌入的方法。重要的是,与不进行剪枝相比,我们实现了最高 3% 的性能提升,从而表明来自合成损坏的洞见在数据剪枝方面具有广阔前景。

英文摘要:

Code datasets, often collected from diverse and uncontrolled sources such as GitHub, potentially suffer from quality issues, thereby affecting the performance and training efficiency of Large Language Models (LLMs) optimized for code generation. Previous studies demonstrated the benefit of using embedding spaces for data pruning, but they mainly focused on duplicate removal or increasing variety, and in other modalities, such as images. Our work focuses on using embeddings to identify and remove "low-quality" code data. First, we explore features of "low-quality" code in embedding space, through the use of synthetic corruptions. Armed with this knowledge, we devise novel pruning metrics that operate in embedding space to identify and remove low-quality entries in the Stack dataset. We demonstrate the benefits of this synthetic corruption informed pruning (SCIP) approach on the well-established HumanEval and MBPP benchmarks, outperforming existing embedding-based methods. Importantly, we achieve up to a 3% performance improvement over no pruning, thereby showing the promise of insights from synthetic corruptions for data pruning.

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

↑