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

训练过程至关重要:平均预训练参数计数统一了稀疏和密集的扩展规律

The Journey Matters: Average Parameter Count over Pre-training Unifies Sparse and Dense Scaling Laws

MIT CSAIL(MIT 计算科学与人工智能实验室);Rice University(稻大学);Google Research(谷歌研究);Google DeepMind(谷歌DeepMind);Google(谷歌);IST Austria(奥地利科学院)

Tian Jin, Ahmed Imtiaz Humayun, Utku Evci, Suvinay Subramanian, Amir Yazdanbakhsh, Dan Alistarh, Gintare Karolina Dziugaite

AI总结:

本文通过研究80种不同的剪枝计划,发现预训练过程中在25%和75%的计算量启动和结束剪枝可获得最佳评估损失,提出新的扩展规律统一了稀疏和密集预训练的扩展规律。

摘要:

剪枝通过消除神经网络中不必要的参数,为大型语言模型(LLMs)日益增长的计算需求提供了一个有前途的解决方案。虽然许多研究关注训练后的剪枝,但将剪枝和预训练结合到一个阶段的稀疏预训练提供了一个更简单的替代方案。在本文中,我们通过研究80种不同的剪枝计划,探讨了不同稀疏度和训练持续时间下的最优稀疏预训练配置。我们发现,在总训练计算量的25%处启动剪枝并在75%处结束可获得接近最优的最终评估损失。这些发现为高效且有效的LLMs稀疏预训练提供了有价值的见解。此外,我们提出了一种新的扩展规律,修改了Chinchilla扩展规律以使用预训练期间的平均参数计数。通过实证和理论验证,我们证明了这种修改后的扩展规律能够准确地建模稀疏和密集预训练LLMs的评估损失,统一了预训练范式的扩展规律。我们的发现表明,虽然稀疏预训练在等效计算预算下能获得与密集预训练相同的最终模型质量,但通过减少模型大小,它在推理过程中提供了显著的计算节省潜力。

英文摘要:

Pruning eliminates unnecessary parameters in neural networks; it offers a promising solution to the growing computational demands of large language models (LLMs). While many focus on post-training pruning, sparse pre-training--which combines pruning and pre-training into a single phase--provides a simpler alternative. In this work, we present the first systematic exploration of optimal sparse pre-training configurations for LLMs through an examination of 80 unique pruning schedules across different sparsity levels and training durations. We find that initiating pruning at 25% of total training compute and concluding at 75% achieves near-optimal final evaluation loss. These findings provide valuable insights for efficient and effective sparse pre-training of LLMs. Furthermore, we propose a new scaling law that modifies the Chinchilla scaling law to use the average parameter count over pre-training. Through empirical and theoretical validation, we demonstrate that this modified scaling law accurately models evaluation loss for both sparsely and densely pre-trained LLMs, unifying scaling laws across pre-training paradigms. Our findings indicate that while sparse pre-training achieves the same final model quality as dense pre-training for equivalent compute budgets, it provides substantial benefits through reduced model size, enabling significant potential computational savings during inference.

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

↑