arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2411.06767cs.CLcs.AIcs.LG

PDC与DM-SFT:增强LLM SQL Bug修复之路

PDC & DM-SFT: A Road for LLM SQL Bug-Fix Enhancing

Bytedance Inc.(字节跳动公司)

Yiwen Duan, Yonghong Yu, Xiaoming Zhao, Yichang Wu, Wenbo Liu

AI总结:

针对代码LLM在SQL bug修复上的不足,提出PDC数据集构建与DM-SFT微调方法,有效减少训练步数并缓解训练迷失问题,使较小模型超越更大规模的最佳模型。

摘要:

代码大型语言模型(Code LLMs),如Code llama和DeepSeek-Coder,在代码生成任务中表现优异。然而,现有模型多关注生成正确代码的能力,常在bug修复上遇到困难。我们提出了一套增强LLM SQL bug修复能力的方法,主要包括两部分:从零开始的渐进式数据集构建(PDC)和动态掩码监督微调(DM-SFT)。PDC从广度优先和深度优先两个角度提出两种数据扩展方法。DM-SFT引入高效的bug修复监督学习方法,有效减少总训练步数并缓解SQL代码bug修复训练中的“迷失方向”问题。评估中,使用这两种方法训练的代码LLM模型超越了所有当前最佳性能的更大规模模型。

英文摘要:

Code Large Language Models (Code LLMs), such as Code llama and DeepSeek-Coder, have demonstrated exceptional performance in the code generation tasks. However, most existing models focus on the abilities of generating correct code, but often struggle with bug repair. We introduce a suit of methods to enhance LLM's SQL bug-fixing abilities. The methods are mainly consisted of two parts: A Progressive Dataset Construction (PDC) from scratch and Dynamic Mask Supervised Fine-tuning (DM-SFT). PDC proposes two data expansion methods from the perspectives of breadth first and depth first respectively. DM-SFT introduces an efficient bug-fixing supervised learning approach, which effectively reduce the total training steps and mitigate the "disorientation" in SQL code bug-fixing training. In our evaluation, the code LLM models trained with two methods have exceeds all current best performing model which size is much larger.

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

↑