likes
comments
collection
share

腾讯LLaMA Pro大模型:突破大模型微调的知识遗忘难题

作者站长头像
站长
· 阅读数 7

引言:大模型微调中的挑战

在人工智能的发展过程中,大型语言模型(LLM)的微调(fine-tuning)始终是提升模型在特定任务上性能的关键。然而,微调过程中常面临一个主要挑战:知识遗忘。这指的是在模型进行特定任务学习时,可能会丢失其原有的预训练知识。为应对这一挑战,香港大学的研究团队联合腾讯ARC实验室,提出了一种新颖的微调方法——Block Expansion,并基于此方法开发了新型模型LLaMA Pro。

腾讯LLaMA Pro大模型:突破大模型微调的知识遗忘难题

微调传统方法的局限性

传统的大模型微调方法主要分为两类:一是部分冻结模型的参数,二是更新所有参数。虽然这些方法能在一定程度上提升模型在特定任务上的性能,但它们通常会导致模型在学习新任务时忘记原有的知识。这种“知识遗忘”现象限制了大型模型在多任务和持续学习场景下的应用。

Block Expansion:一种创新的解决方案

香港大学的研究团队提出的Block Expansion方法为解决这一问题提供了新思路。该方法的核心思想是在保持预训练模型参数不变的基础上,增加新的模块来适应新的训练任务。这些新加入的模块与原有模块协同工作,既保留了模型原有的知识,又能够适应新的训练数据和任务需求。

腾讯LLaMA Pro大模型:突破大模型微调的知识遗忘难题

LLaMA Pro模型的构建与特性

在LLaMA2-7B模型(70亿参数)的基础上,研究团队通过增加8个新模块,构建了LLaMA Pro模型,使其参数量达到83亿。这一过程中,原有的模型参数保持不变,新模块在微调过程中逐渐适应新的任务。LLaMA Pro模型不仅在代码理解和数学推理方面表现出色,而且在语言理解任务上也有所提升。

LLaMA Pro的实验评测与结果分析

为验证LLaMA Pro的效果,研究团队在不同的数据集上进行了广泛测试,包括代码数据集和指令微调数据集。实验结果显示,LLaMA Pro在多个领域的任务上均取得了显著进步,特别是在代码和数学推理方面的表现尤为突出。相比原始LLaMA2-7B模型,LLaMA Pro在保持通用知识的同时,实现了在特定任务上的显著提升。

腾讯LLaMA Pro大模型:突破大模型微调的知识遗忘难题

与传统微调方法的对比

进一步地,研究人员将Block Expansion方法与传统的有监督微调方法进行了对比。实验结果表明,LLaMA Pro在Backward Transfer(BWT)和Overall Performance两个重要指标上均优于传统方法,显示出优异的持续学习和多任务处理能力。

结论与展望

LLaMA Pro模型的成功展示了Block Expansion方法在解决大模型微调过程中知识遗忘问题上的创新性和有效性。这一新方法不仅能有效缓解大模型在学习新任务时的知识遗忘问题,而且在保持模型原有能力的同时,显著提升了模型在特定领域任务的表现。LLaMA Pro模型的开发和实验结果为大型语言模型的微调提供了新的视角,预示着未来可能成为替代传统微调方法的新选择,尤其是在需要模型持续学习和处理多任务的应用场景中。

模型下载

Huggingface模型下载

huggingface.co/TencentARC/…

AI快站模型免费加速下载

aifasthub.com/models/Tenc…