arXiv:2504.10647cs.CL2025-04被引 2

通过推理蒸馏提升语言模型的归纳推理能力,显著改进少样本学习效果。

Improving In-Context Learning with Reasoning Distillation

  • 设计推理蒸馏框架ReDis,结合数据增强与对齐优化,强化模型对输入输出规律的理解。
  • 在1D-ARC、ACRE、MiniSCAN等任务上,性能超越GPT-4o,最高提升66.6%。
  • 适合关注少样本推理、模型可解释性与高效微调的研究者与开发者。

语言模型依赖语义先验进行上下文学习,导致在需要归纳推理的任务上表现不佳。基于模仿学习的指令微调方法虽能表面提升少样本学习性能,但难以真正改善模型对输入输出间底层规则的理解。本文提出ReDis——一种推理蒸馏技术,通过精心组合数据增强、过滤、监督微调与对齐策略,在1D-ARC、List Function、ACRE和MiniSCAN等多种任务上实现显著性能提升。实验表明,三种语言模型骨干均在所有任务上优于基线少样本提示,部分甚至超越教师模型GPT-4o。基于LLaMA-3的ReDis在1D-ARC、ACRE和MiniSCAN上相对GPT-4o分别提升23.2%、2.8%和66.6%,且在相近假设搜索空间内完成。代码、数据集与模型检查点将公开于https://github.com/NafisSadeq/reasoning-distillation.git。

原文摘要 · Abstract (English)

Language models rely on semantic priors to perform in-context learning, which leads to poor performance on tasks involving inductive reasoning. Instruction-tuning methods based on imitation learning can superficially enhance the in-context learning performance of language models, but they often fail to improve the model's understanding of the underlying rules that connect inputs and outputs in few-shot demonstrations. We propose ReDis, a reasoning distillation technique designed to improve the inductive reasoning capabilities of language models. Through a careful combination of data augmentation, filtering, supervised fine-tuning, and alignment, ReDis achieves significant performance improvements across a diverse range of tasks, including 1D-ARC, List Function, ACRE, and MiniSCAN. Experiments on three language model backbones show that ReDis outperforms equivalent few-shot prompting baselines across all tasks and even surpasses the teacher model, GPT-4o, in some cases. ReDis, based on the LLaMA-3 backbone, achieves relative improvements of 23.2%, 2.8%, and 66.6% over GPT-4o on 1D-ARC, ACRE, and MiniSCAN, respectively, within a similar hypothesis search space. The code, dataset, and model checkpoints will be made available at https://github.com/NafisSadeq/reasoning-distillation.git.

推理蒸馏少样本学习语言模型归纳推理

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