S3FT选择性自监督微调:通过智能选择训练样本提升大模型整体表现

简介: 选择性自我监督微调(S3FT)是一种创新的大语言模型微调方法,通过语义等价性判断器识别模型生成的正确响应,并结合标准答案优化模型。相比传统监督微调,S3FT在特定任务上表现更优,显著提升跨域泛化能力,有效缓解灾难性遗忘问题。实验结果显示,S3FT在多个基准测试中表现出色,尤其在阅读理解和领域外任务上优势明显。

选择性自我监督微调(Selective Self-to-Supervised Fine-Tuning,S3FT)是一种创新的大语言模型微调方法,该方法通过部署专门的语义等价性判断器来识别训练集中模型自身生成的正确响应。在微调过程中,S3FT策略性地结合这些正确响应与剩余样本的标准答案(或其释义版本)来优化模型。与传统监督微调(SFT)相比,S3FT不仅在特定任务上表现出更优的性能,还显著提升了模型的跨域泛化能力。通过充分利用模型自身生成的高质量响应,S3FT有效减缓了微调阶段中常见的模型过度专门化问题。

S3FT技术原理与实现机制

选择性自我监督微调(S3FT)旨在解决大型语言模型(LLM)特定任务微调过程中的一个核心挑战:如何在提升模型在目标任务上的表现的同时,最大程度地保留其通用能力。这一问题在标准监督微调(SFT)中尤为突出。S3FT的设计基于两项关键发现:

自然语言处理任务通常存在多种有效响应现象,即对同一输入可能存在多个语义等价但表述不同的正确答案。此外,利用模型自身生成的语言形式进行训练有助于保持模型原始分布特性,从而减轻灾难性遗忘现象(即模型丢失先前获取的知识)。

初始预测阶段:

S3FT首先针对训练样本(输入xi与标准答案yi)让基础模型Mθ0(已经过预训练和指令调整)生成预测结果ˆyi = Mθ0(xi)。

等价性评估阶段:

系统随后评估生成的预测ˆyi与标准答案yi之间的语义等价性。这一评估可通过两种方式实现:一是采用启发式方法,如关键信息比对或整体一致性验证;二是调用更强大的语言模型作为判断器,对ˆyi和yi之间的语义等价性进行评估。

训练数据选择策略:

当ˆyi与yi语义等价时,系统将采用(xi, ˆyi)对作为训练样本,这种方式强化了模型现有知识结构,并有助于维持其原始分布特性。当ˆyi与yi不等价时,基础模型Mθ0会对标准答案yi进行自主释义,生成˜yi = Mθ0([xi; yi]),这一步骤旨在缩小标准答案与模型自身语言风格之间的差距。

二次等价性验证:

对于需要释义的情况,系统会再次验证˜yi是否与yi语义等价。

最终训练数据确定:

如果˜yi与yi语义等价,则使用(xi, ˜yi)对进行训练,这种方法在传授模型所需输出的同时,保持了其自身的"语言风格",最大限度减少与原始分布的偏离。如果˜yi与yi不等价,则回退到标准SFT方式,使用原始(xi, yi)对进行训练,这是当模型无法生成合适释义时的兜底策略。

实验中采用Mistral-instruct-v2 (7B)同时作为基础模型和判断模型。所有微调实验均采用低秩适应(Low-Rank Adaptation,LoRA)技术,其中秩设为8,缩放因子为16,dropout率为0.1。

性能评估与实验结果

上图展示了不同微调技术在准确率(%)指标下的性能对比。

领域内性能提升: 实验结果表明,S3FT在领域内数据集(如GSM8K、MBPP和NQ)上的表现显著优于基础模型和传统SFT方法。特别是在阅读理解任务(NQ数据集)上,S3FT达到了与SDFT相当的性能水平。

上图展示了各方法在其他基准测试上的泛化能力。

灾难性遗忘缓解效果: 与传统SFT相比,S3FT展现出更强的泛化能力,在微调后的领域外基准测试中性能下降幅度明显减小。相比之下,SFT在这些基准上出现了显著的性能降低,表明存在严重的灾难性遗忘问题。

上图为Mistral-7B-Instruct-v0.2对标准响应、标准响应释义和模型自身预测分配的对数概率直方图。

标准响应释义的影响分析: 研究发现,将模型自身的正确响应作为训练目标(如S3FT中采用的策略)能带来更优的性能和泛化能力。这一现象可归因于模型生成的响应通常比标准响应甚至释义后的标准响应更接近模型自身的分布特性。直接训练标准响应会导致模型分布发生改变,从而对泛化能力产生负面影响。

总结

S3FT(选择性自监督微调)代表了一种解决大语言模型微调中固有问题的创新方法。通过智能地选择和整合模型自身生成的高质量响应,S3FT成功地在两个看似矛盾的目标之间取得了平衡:提升特定任务的性能,同时保留模型的泛化能力。实验结果清晰地表明,与传统监督微调相比,S3FT不仅在目标领域内取得了更好的性能,还显著减轻了灾难性遗忘现象,维持了模型在领域外任务上的表现。

这种方法的核心优势在于尊重模型原有的语言分布特性,使微调过程更加和谐,避免了强制模型适应可能与其内部表征不一致的外部标准答案。此外,S3FT的实现相对简单,不需要复杂的架构修改或额外的训练阶段,这使其成为一种实用且有效的微调策略。

未来工作可以探索更高效的等价性判断机制,以及S3FT在更广泛任务类型和更大规模模型上的应用效果。此外,将S3FT与其他微调技术(如参数高效微调方法)结合的潜力也值得研究。总体而言,S3FT为大语言模型的微调提供了一种平衡特定任务性能和通用能力的新范式,为AI系统的实际应用提供了重要价值。

论文地址:

https://avoid.overfit.cn/post/da816d0257eb4600a132a6da935b3cd9

目录
相关文章
|
11月前
|
存储 机器学习/深度学习 人工智能
大模型微调技术:LoRA原理与实践
本文深入解析大语言模型微调中的关键技术——低秩自适应(LoRA)。通过分析全参数微调的计算瓶颈,详细阐述LoRA的数学原理、实现机制和优势特点。文章包含完整的PyTorch实现代码、性能对比实验以及实际应用场景,为开发者提供高效微调大模型的实践指南。
3481 3
|
XML JSON 数据库
大模型不听话?试试提示词微调
想象一下,你向大型语言模型抛出问题,满心期待精准回答,得到的却是答非所问,是不是让人抓狂?在复杂分类场景下,这种“大模型不听话”的情况更是常见。
640 9
|
负载均衡 并行计算 异构计算
大模型训练推理优化(5): FlexLink —— NVLink 带宽无损提升27%
本期我们将介绍蚂蚁集团ASystem团队在大模型通信优化上的新工作FlexLink,旨在通过动态聚合多路通信(NVLink,PCIe,RDMA),在H800等典型硬件上将典型通信算子如(AllReduce, All Gather)吞吐提升最高达27%,尤其适合大模型长序列推理(Prefill阶段),及训练等通信密集的带宽bound场景。方案对精度无影响。
|
11月前
|
数据采集 人工智能 搜索推荐
智能新纪元:多模态大模型如何重塑人机交互
智能新纪元:多模态大模型如何重塑人机交互
461 113
|
机器学习/深度学习 人工智能 算法
GSPO:Qwen让大模型强化学习训练告别崩溃,解决序列级强化学习中的稳定性问题
这是7月份的一篇论文,Qwen团队提出的群组序列策略优化算法及其在大规模语言模型强化学习训练中的技术突破
2157 0
GSPO:Qwen让大模型强化学习训练告别崩溃,解决序列级强化学习中的稳定性问题
|
12月前
|
人工智能 自然语言处理 测试技术
有没有可能不微调也能让大模型准确完成指定任务?(少样本学习)
对于我这种正在从0到1构建AI产品的一人公司来说,Few Shots学习的最大价值在于:用最少的资源获得最大的效果。我不需要大量的标注数据,不需要复杂的模型训练,只需要精心设计几个示例,就能让大模型快速理解我的业务场景。
639 43
|
12月前
|
机器学习/深度学习 算法 数据可视化
从零开始训练推理模型:GRPO+Unsloth改造Qwen实战指南
推理型大语言模型兴起,通过先思考再作答提升性能。本文介绍GRPO等强化学习算法,详解其原理并动手用Qwen2.5-3B训练推理模型,展示训练前后效果对比,揭示思维链生成的实现路径。
1428 2
从零开始训练推理模型:GRPO+Unsloth改造Qwen实战指南
|
机器学习/深度学习 算法
Proximal SFT:用PPO强化学习机制优化SFT,让大模型训练更稳定
本文介绍了一种改进的监督微调方法——Proximal Supervised Fine-Tuning (PSFT),旨在解决传统SFT易过拟合、泛化能力差及导致“熵坍塌”的问题。受PPO强化学习算法启发,PSFT通过引入参数更新的稳定性机制,防止模型在训练中变得过于确定,从而提升探索能力与后续强化学习阶段的表现。实验表明,PSFT在数学推理、模型对齐及泛化能力方面均优于传统SFT。
1033 3
Proximal SFT:用PPO强化学习机制优化SFT,让大模型训练更稳定
|
机器学习/深度学习 数据采集 人工智能
微调之后还能做什么?大模型后训练全链路技术解析
本文探讨了后训练的重要性、方法以及最新进展。文章将包含理论分析与实际操作指南,适合希望深入了解并应用这些技术的开发者。
3249 19
微调之后还能做什么?大模型后训练全链路技术解析
|
12月前
|
人工智能 Java API
Java与大模型集成实战:构建智能Java应用的新范式
随着大型语言模型(LLM)的API化,将其强大的自然语言处理能力集成到现有Java应用中已成为提升应用智能水平的关键路径。本文旨在为Java开发者提供一份实用的集成指南。我们将深入探讨如何使用Spring Boot 3框架,通过HTTP客户端与OpenAI GPT(或兼容API)进行高效、安全的交互。内容涵盖项目依赖配置、异步非阻塞的API调用、请求与响应的结构化处理、异常管理以及一些面向生产环境的最佳实践,并附带完整的代码示例,助您快速将AI能力融入Java生态。
1656 12