VersaTune让大模型多领域能力提升35%,还能避免遗忘旧知识。
VersaTune: An Efficient Data Composition Framework for Training Multi-Capability LLMs
- 根据模型已有知识分布动态组合训练数据,提升多领域学习效率。
- 相比均匀分配,多能力综合性能提升35.21%,超越GPT-4o等主流模型。
- 支持灵活扩展特定领域,其他领域退化减少38.77%,适合实用部署。
如GPT、Claude系列等专有大语言模型所示,单一模型具备跨法律、医学、金融、科学、编程等多领域卓越能力,这些能力在监督微调(SFT)阶段进一步增强。然而,现有方法多聚焦于特定领域的优化,面临跨领域知识灾难性遗忘的挑战。本文提出**VersaTune**,一种新型数据组合框架,用于训练过程中全面提升大模型的多领域能力。首先检测基础模型中各领域知识的分布,随后按此分布构建训练数据。训练期间,依据领域可学潜力与遗忘程度动态调整权重。实验表明,相较于均匀分配,整体多能力表现提升35.21%。此外,Qwen-2.5-32B + VersaTune在多项指标上超越GPT-4o、Claude3.5-Sonnet和DeepSeek-V3,分别领先0.86%、4.76%和4.60%。当需灵活扩展某一领域时,其他领域性能下降减少38.77%,同时保持目标领域训练效率。
原文摘要 · Abstract (English)
As demonstrated by the proprietary Large Language Models (LLMs) such as GPT and Claude series, LLMs have the potential to achieve remarkable proficiency across a wide range of domains, including law, medicine, finance, science, code, etc., all within a single model. These capabilities are further augmented during the Supervised Fine-Tuning (SFT) phase. Despite their potential, existing work mainly focuses on domain-specific enhancements during fine-tuning, the challenge of which lies in catastrophic forgetting of knowledge across other domains. In this study, we introduce **VersaTune**, a novel data composition framework designed for enhancing LLMs' overall multi-domain capabilities during training. We begin with detecting the distribution of domain-specific knowledge within the base model, followed by the training data composition that aligns with the model's existing knowledge distribution. During the subsequent training process, domain weights are dynamically adjusted based on their learnable potential and forgetting degree. Experimental results indicate that VersaTune is effective in multi-domain fostering, with an improvement of 35.21\% in the overall multi-ability performances compared to uniform domain weights. Furthermore, we find that Qwen-2.5-32B + VersaTune even surpasses frontier models, including GPT-4o, Claude3.5-Sonnet and DeepSeek-V3 by 0.86\%, 4.76\% and 4.60\%. Additionally, in scenarios where flexible expansion of a specific domain is required, VersaTune reduces the performance degradation in other domains by 38.77\%, while preserving the training efficacy of the target domain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。