用迁移学习自动优化大模型微调配置,省去重复试错。
Transfer Learning for Finetuning Large Language Models
- 通过元学习迁移已有任务的微调经验
- 在8个合成数据集上优于零样本和默认配置
- 适合想快速部署大模型的工程师
随着大语言模型的发展,针对特定任务高效微调变得愈发重要。同时,参数高效微调方法层出不穷,使实践者在选择最优微调方案时面临复杂决策。为降低复杂度,本文研究大模型微调中的迁移学习,旨在将相关任务的配置知识迁移到新任务。我们通过新构建的元数据集,利用元学习建立性能与成本的代理模型,实现灰箱元优化。反直觉地,我们仅依赖迁移学习处理新数据集,不使用任务特异性贝叶斯优化,而是优先采用来自相关任务的知识。我们在8个合成问答数据集及包含1,800次微调运行的微软Phi-3元数据集上评估该方法,结果表明其优于零样本、默认微调及元优化基线。实验验证了微调配置的可迁移性,能更有效地适应大语言模型。
原文摘要 · Abstract (English)
As the landscape of large language models expands, efficiently finetuning for specific tasks becomes increasingly crucial. At the same time, the landscape of parameter-efficient finetuning methods rapidly expands. Consequently, practitioners face a multitude of complex choices when searching for an optimal finetuning pipeline for large language models. To reduce the complexity for practitioners, we investigate transfer learning for finetuning large language models and aim to transfer knowledge about configurations from related finetuning tasks to a new task. In this work, we transfer learn finetuning by meta-learning performance and cost surrogate models for grey-box meta-optimization from a new meta-dataset. Counter-intuitively, we propose to rely only on transfer learning for new datasets. Thus, we do not use task-specific Bayesian optimization but prioritize knowledge transferred from related tasks over task-specific feedback. We evaluate our method on eight synthetic question-answer datasets and a meta-dataset consisting of 1,800 runs of finetuning Microsoft's Phi-3. Our transfer learning is superior to zero-shot, default finetuning, and meta-optimization baselines. Our results demonstrate the transferability of finetuning to adapt large language models more effectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。