arXiv:2411.00863cs.CLcs.AI2024-11被引 2

优化数学证明数据顺序,可显著提升大模型生成能力

Next-Token Prediction Task Assumes Optimal Data Ordering for LLM Training in Proof Generation

  • 采用直觉顺序排列证明步骤,让中间监督信息提前出现
  • 顺序优化使证明成功率提升11%,效果差异显著
  • 发现17.3%的高阶数学定理存在顺序缺陷,适合形式化验证研究者

在基于大语言模型(LLM)的证明生成领域,尽管在ArXiv等大规模数据集上训练,模型对中等难度证明任务的表现仍不理想。我们认为这部分源于训练数据中证明步骤排序不优。现有证明常按纯逻辑顺序排列,利于验证但不利于学习发现过程。我们提出‘直觉顺序’——每个证明步骤前应包含其所需中间监督信息。在直觉命题逻辑定理证明和数字乘法任务上验证,模型在最优顺序下训练效果最佳。实验显示,最优与最差顺序间证明成功率相差达11%。此外,我们识别出高等数学证明中一种常见顺序问题,在一本广泛使用的研究生教材前两章中,17.3%的非平凡定理存在此问题,附录提供详细列表。

原文摘要 · Abstract (English)

In the field of large language model (LLM)-based proof generation, despite extensive training on large datasets such as ArXiv, LLMs still exhibit only modest performance on proving tasks of moderate difficulty. We believe that this is partly due to the widespread presence of suboptimal ordering within the data for each proof used in training. For example, published proofs often follow a purely logical order, where each step logically proceeds from the previous steps based on the deductive rules. This order is designed to facilitate the verification of the proof's soundness, rather than to help people and models learn the discovery process of the proof. In proof generation, we argue that the optimal order for one training data sample occurs when the relevant intermediate supervision for a particular proof step in the proof is always positioned to the left of that proof step. We call such order the intuitively sequential order. We validate our claims using two tasks: intuitionistic propositional logic theorem-proving and digit multiplication. Our experiments verify the order effect and provide support for our explanations. We demonstrate that training is most effective when the proof is in the intuitively sequential order. Moreover, the order effect and the performance gap between models trained on different data orders can be substantial -- with an 11 percent improvement in proof success rate observed in the propositional logic theorem-proving task, between models trained on the optimal order compared to the worst order. Lastly, we define a common type of order issue in advanced math proofs and find that 17.3 percent of theorems with nontrivial proofs in the first two chapters of a widely used graduate-level mathematics textbook suffer from this issue. A detailed list of those proofs is provided in the appendix.

证明生成数据顺序大模型训练数学推理

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