用强模型诊断弱模型缺哪些技能,针对性调整训练数据提升效果。
Skill-Targeted Adaptive Training
- 用强模型分析任务所需技能,标记每道题的技能需求。
- 根据学生答题错误情况生成缺失技能画像,动态重加权或合成新数据。
- 在MATH等数据集上提升达7.5%,且与强化学习互补增效。
语言模型在类似预训练数据的任务(如MATH)上进行传统监督微调(SFT)时往往出现性能停滞。本文提出STAT策略,利用更强的大语言模型作为教师,通过任务数据识别完成任务所需的技能,并为每个样本标注其所需技能(Didolkar et al., 2024)。教师监控学生作答,构建学生“缺失技能图谱”,记录各技能应用失败频率。据此,我们以两种方式构建改进训练集:在STAT-Sel中,对现有数据按缺失技能程度自适应重加权;在STAT-Syn中,合成包含缺失技能的新样本。在Llama和Qwen模型上实验显示,相比仅用SFT,STAT在MATH上最高提升7.5%。此外,对分布外基准(如AIME24/25、AMC23)平均提升4.6%。关键发现是:STAT与基于GRPO的强化学习具有互补性——先用STAT弥补技能缺口后,再使用GRPO仍能带来额外增益。结论表明,技能目标自适应训练可显著优化当前训练流程。代码已开源:https://github.com/princeton-pli/STAT。
原文摘要 · Abstract (English)
Language models often show little to no improvement (i.e., "saturation") when trained via vanilla supervised fine-tuning (SFT) on data similar to what they saw in their training set (e.g., MATH). We introduce a new fine-tuning strategy, STAT, to train such a student model by using the metacognition ability of a stronger large language model (LLM) as the teacher. The teacher uses the task dataset to create a list of skills needed for the task, and then labels each data point with its required skills (Didolkar et al., 2024). By monitoring the student's answers, the teacher creates a Missing-Skill-Profile for the student, tracking how often they failed to apply each skill in their responses. We use this idea to build a modified training set in one of two ways. In STAT-Sel, the teacher uses an existing set of training examples but adaptively reweights them according to the Missing-Skill-Profile. In STAT-Syn, the teacher synthesizes additional examples involving missing skills. Across extensive experiments on Llama and Qwen models, our methods yield improvements of up to 7.5% on MATH, whereas SFT provides only limited gains. Furthermore, STAT enhances performance on out-of-distribution benchmarks (e.g., AIME24/25, AMC23, etc.) by an average of 4.6%. Crucially, we find that STAT is complementary to RL via GRPO (Shao et al., 2024): after the model is improved using STAT to address skill gaps, GRPO continues to add further gains. We conclude that skill-targeted adaptive training should broadly improve current training pipelines. Our code is available at: https://github.com/princeton-pli/STAT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。