让大模型学会特定技能,不用调数据比例还能保持语言能力。
Agent Skill Acquisition for Large Language Models via CycleQD
- 用循环切换任务目标的方式训练,专注单个任务提升性能。
- 在代码、系统、数据库任务上超越传统微调,接近GPT-3.5水平。
- 保留强语言能力,适用于多类模型,通用性好。
训练大语言模型掌握特定技能仍具挑战性。传统方法常受数据分布不均和目标函数与任务表现不匹配的影响。为此,我们提出CycleQD,基于质量多样性框架,通过算法循环适应、模型合并交叉和SVD变异实现。在CycleQD中,每个任务的性能指标轮流作为质量度量,其余任务作为行为特征,使模型集中精力于单一任务,无需调整数据比例,简化目标函数设计。实验结果表明,将CycleQD应用于基于LLAMA3-8B-INSTRUCT的模型,在AgentBench测试中,于编码、操作系统和数据库任务上超越传统微调方法,并在这些领域达到与参数更多、性能更强的GPT-3.5-TURBO相当的水平。关键的是,该方法在提升专项能力的同时,仍保持良好的语言能力,体现在广泛使用的语言基准测试中的优异表现。我们详细分析了核心设计选择及其有效性。此外,该方法具备通用性,可拓展至图像分割模型,展现跨领域适用潜力。
原文摘要 · Abstract (English)
Training large language models to acquire specific skills remains a challenging endeavor. Conventional training approaches often struggle with data distribution imbalances and inadequacies in objective functions that do not align well with task-specific performance. To address these challenges, we introduce CycleQD, a novel approach that leverages the Quality Diversity framework through a cyclic adaptation of the algorithm, along with a model merging based crossover and an SVD-based mutation. In CycleQD, each task's performance metric is alternated as the quality measure while the others serve as the behavioral characteristics. This cyclic focus on individual tasks allows for concentrated effort on one task at a time, eliminating the need for data ratio tuning and simplifying the design of the objective function. Empirical results from AgentBench indicate that applying CycleQD to LLAMA3-8B-INSTRUCT based models not only enables them to surpass traditional fine-tuning methods in coding, operating systems, and database tasks, but also achieves performance on par with GPT-3.5-TURBO, which potentially contains much more parameters, across these domains. Crucially, this enhanced performance is achieved while retaining robust language capabilities, as evidenced by its performance on widely adopted language benchmark tasks. We highlight the key design choices in CycleQD, detailing how these contribute to its effectiveness. Furthermore, our method is general and can be applied to image segmentation models, highlighting its applicability across different domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。