arXiv:2410.11451cs.CL2024-10EMNLP被引 8

小模型训练后期性能下降,因激活收敛慢且参数有效秩低。

Tending Towards Stability: Convergence Challenges in Small Language Models

  • 分析不同规模模型的注意力与MLP激活收敛过程。
  • 大模型20%训练内激活稳定,小模型收敛慢且不稳。
  • 参数有效秩低是小模型学习效率差的关键原因。

增大参数量是提升语言模型性能的常见策略,但小模型因运行成本低仍具价值。尽管如此,小模型在同等数据和算力下常表现不佳,尤其在预训练后期性能下降。这通常归因于表征能力弱,但具体原因尚不明确。本文利用Pythia模型系列分析训练动态,考察不同模型规模下注意力与MLP激活的收敛情况,以及参数有效秩对收敛的影响。结果发现,大模型各层在前20%训练中即趋于稳定,而小模型收敛更慢且不稳定,尤其当参数有效秩较低时。通过将激活收敛与参数有效秩关联,本研究为改进小模型学习效率提供了方向。

原文摘要 · Abstract (English)

Increasing the number of parameters in language models is a common strategy to enhance their performance. However, smaller language models remain valuable due to their lower operational costs. Despite their advantages, smaller models frequently underperform compared to their larger counterparts, even when provided with equivalent data and computational resources. Specifically, their performance tends to degrade in the late pretraining phase. This is anecdotally attributed to their reduced representational capacity. Yet, the exact causes of this performance degradation remain unclear. We use the Pythia model suite to analyse the training dynamics that underlie this phenomenon. Across different model sizes, we investigate the convergence of the Attention and MLP activations to their final state and examine how the effective rank of their parameters influences this process. We find that nearly all layers in larger models stabilise early in training - within the first 20% - whereas layers in smaller models exhibit slower and less stable convergence, especially when their parameters have lower effective rank. By linking the convergence of layers' activations to their parameters' effective rank, our analyses can guide future work to address inefficiencies in the learning dynamics of small models.

小模型训练动态参数有效秩收敛

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。