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

Stable-DiffCoder:推动代码扩散大语言模型的前沿

Stable-DiffCoder: Pushing the Frontier of Code Diffusion Large Language Model

Huazhong University of Science and Technology(华中科技大学)

Chenghao Fan, Wen Heng, Bo Li, Sichen Liu, Yuxuan Song, Jing Su, Xiaoye Qu, Kai Shen, Wei Wei

AI总结:

Stable-DiffCoder通过基于扩散的训练方法,在代码生成任务中超越了传统自回归模型,提升了代码建模的质量和效率。

摘要:

基于扩散的语言模型(DLLMs)相比自回归(AR)模型提供了非序列、分块生成和更丰富的数据重用,但现有的代码DLLMs在可比预算下仍落后于强大的AR基线。我们在此设置中进行了受控研究,并引入了Stable-DiffCoder,这是一种基于分块扩散的代码模型,重新利用Seed-Coder架构、数据和训练流程。为了实现高效的知识学习和稳定的训练,我们结合了一个增强的分块扩散持续预训练(CPT)阶段,该阶段通过定制的预热和分块裁剪噪声调度进行增强。在相同的数据和架构下,Stable-DiffCoder在广泛的代码基准测试中整体优于其AR counterpart。此外,仅依靠CPT和监督微调阶段,Stable-DiffCoder在广泛的~8B AR和DLLMs上实现了更强的性能,证明了基于扩散的训练可以超越仅AR训练的代码建模质量。此外,基于扩散的任意顺序建模提高了结构化代码的编辑和推理建模,并通过数据增强,受益于低资源编程语言。

英文摘要:

Diffusion-based language models (DLLMs) offer non-sequential, block-wise generation and richer data reuse compared to autoregressive (AR) models, but existing code DLLMs still lag behind strong AR baselines under comparable budgets. We revisit this setting in a controlled study and introduce Stable-DiffCoder, a block diffusion code model that reuses the Seed-Coder architecture, data, and training pipeline. To enable efficient knowledge learning and stable training, we incorporate a block diffusion continual pretraining (CPT) stage enhanced by a tailored warmup and block-wise clipped noise schedule. Under the same data and architecture, Stable-DiffCoder overall outperforms its AR counterpart on a broad suite of code benchmarks. Moreover, relying only on the CPT and supervised fine-tuning stages, Stable-DiffCoder achieves stronger performance than a wide range of \~8B ARs and DLLMs, demonstrating that diffusion-based training can improve code modeling quality beyond AR training alone. Moreover, diffusion-based any-order modeling improves structured code modeling for editing and reasoning, and through data augmentation, benefits low-resource coding languages.

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

↑