针对小模型数学推理,自适应选择是否引入技能提示,避免信息过载。
AdaptMI: Adaptive Skill-based In-context Math Instruction for Small Language Models
- 根据模型表现动态决定是否使用技能引导示例
- 在5个主流数学基准上提升准确率最高达6%
- 适合资源有限但需提升推理能力的小模型使用者
上下文学习(ICL)使语言模型在提供合适上下文信息时提升解题能力。近期研究发现,利用前沿大模型的元认知能力预测解题所需技能,并据此构造上下文示例,可显著提升大模型的ICL性能。然而该方法在小语言模型(SLMs)上的增益有限,暴露出其在ICL能力上的差距。我们分析发现,对简单问题强行引入技能提示会增加不必要的信息,导致认知过载,反而降低小模型表现。为此,我们提出AdaptMI,一种基于认知负荷理论的自适应策略:仅当模型表现不佳时才引入技能相关示例。进一步提出AdaptMI+,额外加入针对性弥补模型缺失技能的示例。在包含5个数学基准和5种1B–7B规模(Qwen、Llama)SLM的5次采样评估中,AdaptMI+相比朴素技能引导策略最高提升准确率6%。
原文摘要 · Abstract (English)
In-context learning (ICL) allows a language model to improve its problem-solving capability when provided with suitable information in context. Since the choice of in-context information can be determined based on the problem itself, in-context learning is analogous to human learning from teachers in a classroom. Recent works (Didolkar et al., 2024a; 2024b) show that ICL performance can be improved by leveraging a frontier large language model's (LLM) ability to predict required skills to solve a problem, popularly referred to as an LLM's metacognition, and using the recommended skills to construct necessary in-context examples. While this skill-based strategy boosts ICL performance in larger models, its gains on small language models (SLMs) have been minimal, highlighting a performance gap in ICL capabilities. We investigate this gap and show that skill-based prompting can hurt SLM performance on easy questions by introducing unnecessary information, akin to cognitive overload. To address this, we introduce AdaptMI, an adaptive approach to selecting skill-based in-context Math Instructions for SLMs. Inspired by cognitive load theory from human pedagogy, our method only introduces skill-based examples when the model performs poorly. We further propose AdaptMI+, which adds examples targeted to the specific skills missing from the model's responses. On 5-shot evaluations across popular math benchmarks and five SLMs (1B--7B; Qwen, Llama), AdaptMI+ improves accuracy by up to 6% over naive skill-based strategies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。