arXiv:2508.13037cs.CLcs.AI2025-08IJCAI被引 2

用多LoRA模拟人类双系统思维,提升小模型数学推理能力。

Can Large Models Teach Student Models to Solve Mathematical Problems Like Human Beings? A Reasoning Distillation Method via Multi-LoRA Interaction

  • 模仿人类双系统思维,分步训练知识生成与深度推理模块。
  • 在GSM8K上超越现有方法,最高提升16.1%准确率。
  • 适合想提升小模型逻辑推理能力的研究者和工程师。

近期研究显示大型语言模型(LLMs)具备强大的数学推理能力,但依赖数百亿参数。为解决小型语言模型(SLMs)推理能力弱的问题,现有方法通常利用LLMs生成海量数据进行强行训练,类似于基于经验和直觉的快速思维(系统1)。然而,人类学习还包含系统2思维,即先获取知识再通过练习强化。受此启发,我们提出一种基于多LoRA交互的数学推理蒸馏方法(LoRID)。首先,将样本问题与推理输入LLM,构建增强知识的数据集。接着,在学生模型上训练一个LoRA模块作为直觉推理器(IR),直接生成解决问题的思维链。然后,为模拟系统2思维,分别训练知识生成器(KG)和深度推理器(DR):前者接收问题后输出知识,后者使用该知识进行推理。最后,为解决IR与DR生成结果的随机性,评估其输出一致性,若不一致则迭代推理过程,通过相互反馈提升小模型的数学推理能力。实验表明,LoRID达到最优性能,尤其在GSM8K数据集上,相对于第二优方法,五种基础模型的准确率分别提升2.3%、16.1%、2.4%、12.3%和1.8%。

原文摘要 · Abstract (English)

Recent studies have demonstrated that Large Language Models (LLMs) have strong mathematical reasoning abilities but rely on hundreds of billions of parameters. To tackle the challenge of poor reasoning in Small Language Models (SLMs), existing methods typically leverage LLMs to generate massive amounts of data for cramming training. In psychology, they are akin to System 1 thinking, which resolves reasoning problems rapidly based on experience and intuition. However, human learning also requires System 2 thinking, where knowledge is first acquired and then reinforced through practice. Inspired by such two distinct modes of thinking, we propose a novel method based on the multi-LoRA Interaction for mathematical reasoning Distillation (LoRID). First, we input the question and reasoning of each sample into an LLM to create knowledge-enhanced datasets. Subsequently, we train a LoRA block on the student model as an Intuitive Reasoner (IR), which directly generates Chain-of-Thoughts for problem-solving. Then, to imitate System 2 thinking, we train the Knowledge Generator (KG) and Deep Reasoner (DR), respectively. The former outputs only knowledge after receiving problems, while the latter uses that knowledge to perform reasoning. Finally, to address the randomness in the generation of IR and DR, we evaluate whether their outputs are consistent, and the inference process needs to be iterated if not. This step can enhance the mathematical reasoning ability of SLMs through mutual feedback. Experimental results show that LoRID achieves state-of-the-art performance, especially on the GSM8K dataset, where it outperforms the second-best method by 2.3%, 16.1%, 2.4%, 12.3%, and 1.8% accuracy across the five base models, respectively.

数学推理模型蒸馏LoRA小模型

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