arXiv:2410.05248cs.CLcs.AI2024-10ACL被引 4

用混合方法提升大模型指令微调效果,无需高质量数据集

SFTMix: Elevating Language Model Instruction Tuning with Mixup Recipe

  • 基于置信度差异混合训练样本,动态调节学习过程
  • 在多个任务和模型上均实现稳定性能提升
  • 适合数据少或算力受限场景,兼容性强

为获得指令遵循能力,大语言模型(LLMs)通过指令-响应对进行指令微调,使用下一个词预测(NTP)训练。现有改进多依赖高质量监督微调(SFT)数据集,通常需使用专有大模型筛选或人工标注。本文提出SFTMix,一种基于Mixup的新型微调方案,无需依赖精心构建的数据集即可提升指令微调效果。我们发现,大模型在语义空间中置信度分布不均:高置信样本易过拟合,低置信样本更难泛化。据此,SFTMix利用训练动态识别不同置信度样本,并通过插值缩小置信差距,引入Mixup正则化以增强对合成样本的学习。我们在指令跟随及医疗领域SFT任务中验证了SFTMix的有效性,在多种模型族与数据集规模、质量下均表现一致提升。六项深入分析表明其兼容数据筛选、适应算力受限环境,并具备广泛扩展潜力。

原文摘要 · Abstract (English)

To acquire instruction-following capabilities, large language models (LLMs) undergo instruction tuning, where they are trained on instruction-response pairs using next-token prediction (NTP). Efforts to improve instruction tuning often focus on higher-quality supervised fine-tuning (SFT) datasets, typically requiring data filtering with proprietary LLMs or human annotation. In this paper, we take a different approach by proposing SFTMix, a novel Mixup-based recipe that elevates LLM instruction tuning without relying on well-curated datasets. We observe that LLMs exhibit uneven confidence across the semantic representation space. We argue that examples with different confidence levels should play distinct roles in instruction tuning: Confident data is prone to overfitting, while unconfident data is harder to generalize. Based on this insight, SFTMix leverages training dynamics to identify examples with varying confidence levels. We then interpolate them to bridge the confidence gap and apply a Mixup-based regularization to support learning on these additional, interpolated examples. We demonstrate the effectiveness of SFTMix in both instruction-following and healthcare-specific SFT tasks, with consistent improvements across LLM families and SFT datasets of varying sizes and qualities. Extensive analyses across six directions highlight SFTMix's compatibility with data selection, adaptability to compute-constrained scenarios, and scalability to broader applications.

指令微调Mixup大模型数据效率

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