Do Larger Language Models Imply Better Reasoning? A Pretraining Scaling Law for Reasoning

LLM PTSAD Latent Reasoning
大型语言模型 (LLMs) 在需要复杂推理的广泛任务中展现了卓越的能力。然而,规模扩展对其推理能力的影响仍不够清晰。在本文中,我们引入了一种合成的多步推理环境,旨在紧密复制现实世界大规模知识图谱的结构和分布。我们的推理任务涉及填补知识图谱中的缺失边,这需要高级的多步推理能力,并且模拟了现实世界的推理场景。为此,我们从头开始预训练语言模型 (LMs),仅使用不完整图谱中的三元组数据,并评估这些模型推断缺失边的能力。有趣的是,我们观察到过度参数化可能会因过多的记忆化而损害推理性能。我们研究了影响这种 U 型损失曲线的各种因素,包括图结构、模型大小和训练步数。为了预测适用于特定知识图谱的最佳模型大小,我们发现了一种经验性的缩放规律,该规律可以将知识图谱的搜索熵线性映射到最佳模型大小。本研究为规模扩展与 LLM 推理能力之间的关系提供了新的见解,并揭示了优化其在推理任务中性能的可能方法。
Large Language Models (LLMs) have demonstrated remarkable capabilities across a wide range of tasks requiring complex reasoning. However, the effects of scaling on their reasoning abilities remain insufficiently understood. In this paper, we introduce a synthetic multihop reasoning environment designed to closely replicate the structure and distribution of real-world large-scale knowledge graphs. Our reasoning task involves completing missing edges in the graph, which requires advanced multi-hop reasoning and mimics real-world reasoning scenarios. To evaluate this, we pretrain language models (LMs) from scratch solely on triples from the incomplete graph and assess their ability to infer the missing edges. Interestingly, we observe that overparameterization can impair reasoning performance due to excessive memorization. We investigate different factors that affect this U-shaped loss curve, including graph structure, model size, and training steps. To predict the optimal model size for a specific knowledge graph, we find an empirical scaling that linearly maps the knowledge graph search entropy to the optimal model size. This work provides new insights into the relationship between scaling and reasoning in LLMs, shedding light on possible ways to optimize their performance for reasoning tasks.
许愿