数据少时,扩散模型比自回归模型更优。
Diffusion Beats Autoregressive in Data-Constrained Settings
- 用随机掩码训练扩散模型,重复利用有限数据。
- 计算资源多时,扩散模型验证损失更低,下游表现更好。
- 适合数据稀缺但算力充足的场景,如小样本任务。
自回归(AR)模型长期主导大语言模型发展,而基于扩散的语言模型作为新兴替代方案,其优势尚未充分探索。本文系统研究了在数据受限设置下,通过多次遍历有限数据进行训练的掩码扩散模型,发现当计算资源充足但数据稀缺时,扩散模型显著优于自回归模型。扩散模型能更高效利用重复数据,实现更低的验证损失和更优的下游性能。我们发现了扩散模型的新缩放规律,并推导出扩散模型超越自回归模型的临界计算阈值的闭式表达式。原因在于,扩散模型的随机掩码目标隐式地在丰富的词元排列分布上进行训练,起到了隐式数据增强的作用,这是自回归模型固定左右顺序因式分解所不具备的。结果表明,在数据而非算力成为瓶颈时,扩散模型是自回归范式的有力替代方案。代码已公开:https://diffusion-scaling.github.io。
原文摘要 · Abstract (English)
Autoregressive (AR) models have long dominated the landscape of large language models, driving progress across a wide range of tasks. Recently, diffusion-based language models have emerged as a promising alternative, though their advantages over AR models remain underexplored. In this paper, we systematically study masked diffusion models in data-constrained settings where training involves repeated passes over limited data and find that they significantly outperform AR models when compute is abundant but data is scarce. Diffusion models make better use of repeated data, achieving lower validation loss and superior downstream performance. We find new scaling laws for diffusion models and derive a closed-form expression for the critical compute threshold at which diffusion begins to outperform AR. Finally, we explain why diffusion models excel in this regime: their randomized masking objective implicitly trains over a rich distribution of token orderings, acting as an implicit data augmentation that AR's fixed left-to-right factorization lacks. Our results suggest that when data, not compute, is the bottleneck, diffusion models offer a compelling alternative to the standard AR paradigm. Our code is available at: https://diffusion-scaling.github.io.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。