arXiv:2410.18574cs.AI2024-10ACL被引 7

让小模型学会多策略解题,自选最优路径提升数学推理能力

SIKeD: Self-guided Iterative Knowledge Distillation for mathematical reasoning

  • 大模型教小模型用多种解题策略,小模型自动生成反馈选择最佳策略
  • 在多个数学推理数据集上,小模型性能显著优于传统蒸馏方法
  • 适合需要灵活推理的小模型部署场景,如移动设备或低算力环境

大型语言模型(LLMs)可通过传授解决多步推理任务所需的中间推理过程,将推理能力迁移至小型模型。尽管LLMs能通过多种策略准确完成推理任务,甚至无需微调,但小型模型在蒸馏过程中无法充分拟合LLMs在所有策略上的分布,往往倾向于依赖单一策略,导致在面对其偏好策略不适用的任务时表现受限。为此,我们提出自指导迭代知识蒸馏方法SIKeD,使LLM指导小型模型采用不同策略解题,并让小型模型利用自身生成的在线策略输出,自主选择最适合当前任务的策略。训练以自指导迭代方式进行,每轮动态决定如何融合LLM数据与自生成输出。与传统蒸馏方法不同,SIKeD使小型模型在持续学习多策略解题的同时,具备任务-策略匹配能力。在多个数学推理数据集上的实验表明,SIKeD在不同尺寸的小型模型上均显著优于传统蒸馏技术。代码已公开于:https://github.com/kumar-shridhar/SIKeD

原文摘要 · Abstract (English)

Large Language Models (LLMs) can transfer their reasoning skills to smaller models by teaching them to generate the intermediate reasoning process required to solve multistep reasoning tasks. While LLMs can accurately solve reasoning tasks through a variety of strategies, even without fine-tuning, smaller models are not expressive enough to fit the LLMs distribution on all strategies when distilled and tend to prioritize one strategy over the others. This reliance on one strategy poses a challenge for smaller models when attempting to solve reasoning tasks that may be difficult with their preferred strategy. To address this, we propose a distillation method SIKeD (Self-guided Iterative Knowledge Distillation for mathematical reasoning), where the LLM teaches the smaller model to approach a task using different strategies and the smaller model uses its self-generated on-policy outputs to choose the most suitable strategy for the given task. The training continues in a self-guided iterative manner, where for each training iteration, a decision is made on how to combine the LLM data with the self-generated outputs. Unlike traditional distillation methods, SIKeD allows the smaller model to learn which strategy is suitable for a given task while continuously learning to solve a task using different strategies. Our experiments on various mathematical reasoning datasets show that SIKeD significantly outperforms traditional distillation techniques across smaller models of different sizes. Our code is available at: https://github.com/kumar-shridhar/SIKeD

知识蒸馏数学推理自指导小模型

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