提出平均参数量新缩放律,统一稀疏与密集预训练的性能预测。
The Journey Matters: Average Parameter Count over Pre-training Unifies Sparse and Dense Scaling Laws
- 通过80种剪枝策略实验,发现25%~75%训练算力区间剪枝最优。
- 稀疏预训练在相同算力下达到与稠密模型相当的最终损失。
- 新缩放律适用于稀疏与密集模型,助力推理阶段大幅降耗。
剪枝可消除神经网络中冗余参数,是应对大语言模型日益增长计算需求的有前景方案。尽管多数研究聚焦于训练后剪枝,但将剪枝与预训练结合的稀疏预训练更具简洁性。本文首次系统探索了大语言模型稀疏预训练的最优配置,通过分析80种不同稀疏度与训练时长的剪枝策略,发现从总训练算力25%处开始、75%处结束剪枝,可实现接近最优的最终评估损失。进一步提出一种改进的缩放律,将Chinchilla缩放律中的最终参数量替换为预训练过程中的平均参数量。经实证与理论验证,该新缩放律能准确建模稀疏与稠密预训练模型的评估损失,统一了两类预训练范式下的缩放规律。结果表明,在相同算力预算下,稀疏预训练可达到与稠密预训练相当的模型质量,同时显著减小模型规模,为推理阶段带来巨大潜在计算节省。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。