引言:大模型微调中的挑战
在人工智能的发展过程中,大型语言模型(LLM)的微调(fine-tuning)始终是提升模型在特定任务上性能的关键。然而,微调过程中常面临一个主要挑战:知识遗忘。这指的是在模型进行特定任务学习时,可能会丢失其原有的预训练知识。为应对这一挑战,香港大学的研究团队联合腾讯ARC实验室,提出了一种新颖的微调方法——Block Expansion,并基于此方法开发了新型模型LLaMA Pro。
-
Huggingface模型下载:https://huggingface.co/TencentARC/LLaMA-Pro-8B
-
AI快站模型免费加速下载:https://aifasthub.com/models/TencentARC
微调传统方法的局限性
传统的大模型微调方法主要分为两类:一是部分冻结模型的参数,二是更新所有参数。虽然这些方法能在一定程度上提升模型在特定任务上的性能,但它们通常会导致模型在学习新任务时忘记原有的知识。这种“知识遗忘”现象限制了大型模型在多任务和持续学习场景下的应用。
Block Expansion:一种创新的解决方案
香港大学的研究团队提出的Block Expansion方法为解决这一问题提供了新思路。该方法的核心思想是在保持预训练模型参数不变的基础上,增加新的模块来适应新的训练任务。这些新加入的模块与原有模块协同工作,既保留了模型原有的知识,又能够适应新的训练数据和任务需求。
LLaMA Pro模型的构建与特性
在LLaMA2-7B模型(70亿参数)的基础上,研究团队通过增加8个新模块,构建了LLaMA Pro模型,使其参数量达到83亿。这一过程中,原有的模型参数保持不变,新模块在微调过程中逐渐适应新的任务。LLaMA Pro模型不仅在代码理解和数学推理方面表现出色,而且在语言理解任务上也有所提升。
LLaMA Pro的实验评测与结果分析
为验证LLaMA Pro的效果,研究团队在不同的数据集上进行了广泛测试,包括代码数据集和指令微调数据集。实验结果显示,LLaMA Pro在多个领域的任务上均取得了显著进步,特别是在代码和数学推理方面的表现尤为突出。相比原始LLaMA2-7B模型,LLaMA Pro在保持通用知识的同时,实现了在特定任务上的显著提升。
与传统微调方法的对比
进一步地,研究人员将Block Expansion方法与传统的有监督微调方法进行了对比。实验结果表明,LLaMA Pro在Backward Transfer(BWT)和Overall Performance两个重要指标上均优于传统方法,显示出优异的持续学习和多任务处理能力。
结论与展望
LLaMA Pro模型的成功展示了Block Expansion方法在解决大模型微调过程中知识遗忘问题上的创新性和有效性。这一新方法不仅能有效缓解大模型在学习新任务时的知识遗忘问题,而且在保持模型原有能力的同时,显著提升了模型在特定领域任务的表现。LLaMA Pro模型的开发和实验结果为大型语言模型的微调提供了新的视角,预示着未来可能成为替代传统微调方法的新选择,尤其是在需要模型持续学习和处理多任务的应用场景中。
模型下载
Huggingface模型下载
https://huggingface.co/TencentARC/LLaMA-Pro-8B
AI快站模型免费加速下载
https://aifasthub.com/models/TencentARC