本文将从微调的本质ingFang SC", "Hiragino Sans GB", "Microsoft YaHei UI", "Microsoft YaHei", Arial, sans-serif;letter-spacing: 0.578px;color: rgb(255, 129, 36);font-size: 16px;text-align: center;">、微调的原理、微调的应用三个方面,带您一文搞懂模型微调 Fine-tuning。
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei UI", "Microsoft YaHei", Arial, sans-serif;letter-spacing: 0.034em;text-indent: 0em;text-wrap: wrap;outline: 0px;text-align: start;font-size: var(--articleFontsize);visibility: visible;">
Fine-tuning模型微调
一、微调的本质
迁移学习和微调
减少对新数据的需求:从头开始训练一个大型神经网络通常需要大量的数据和计算资源,而在实际应用中,我们可能只有有限的数据集。通过微调预训练模型,我们可以利用预训练模型已经学到的知识,减少对新数据的需求,从而在小数据集上获得更好的性能。
降低训练成本:由于我们只需要调整预训练模型的部分参数,而不是从头开始训练整个模型,因此可以大大减少训练时间和所需的计算资源。这使得微调成为一种高效且经济的解决方案,尤其适用于资源有限的环境。
二、微调的原理
PEFT仅针对部分参数进行微调,从而显著减少了训练时间和成本,尤其适用于数据量有限或计算资源受限的场景。
PEFT包含了多种不同的技术,例如Prefix Tuning、Prompt Tuning、Adapter Tuning和LoRA等,每种技术都有其独特的方法和特点,可以根据具体的任务和模型需求灵活选择。
PEFT的分类
Prefix Tuning:通过在模型的输入前添加可学习的虚拟令牌(virtual tokens)作为前缀来实现微调。在训练过程中,仅更新这些前缀参数,而模型的其余部分保持不变。这种方法减少了需要更新的参数数量,从而提高了训练效率。
Prompt Tuning:在输入层加入prompt tokens,可以看作是Prefix Tuning的简化版,它不需要额外的多层感知机(MLP)调整。随着模型规模的增大,Prompt Tuning的效果逐渐接近全量微调。
Prompt Tuning
Adapter Tuning:则是通过在模型中设计并嵌入Adapter结构来进行微调。这些Adapter结构通常是小型网络模块,可以添加到模型的特定层中。在训练过程中,仅对这些新增的Adapter结构进行微调,而原模型的参数保持不变。这种方法保持了模型的高效性,同时引入的额外参数数量相对较少。
Adapter Tuning
LoRA(Low-Rank Adaptation):通过在模型的矩阵相乘模块中引入低秩矩阵来模拟全量微调的效果。它主要更新语言模型中的关键低秩维度,从而实现高效的参数调整并降低计算复杂度。
LoRA
三、微调的应用
几种微调CNN模型的方法:
方法一:仅修改最后一层(全连接层)
策略:保持预训练CNN模型中除最后一层外的所有层不变,仅替换或修改最后的全连接层以适应新任务的类别数。根据需要,可以选择是否冻结靠近输入的层。
效果:快速适应新任务的分类需求,同时保留预训练模型学到的有用特征。
方法二:修改最后几层
策略:除了最后一层外,还修改倒数第二层或更前面的几层,以适应新任务的特征需求。在修改过程中,可以根据需要选择是否冻结部分层。
效果:使模型学习更多与新任务相关的特征表示,提高在新任务上的性能。但需要注意过拟合的风险。
方法三:微调整个模型
策略:对预训练模型的所有层进行参数更新,以适应新任务的需求。在微调过程中,可以根据需要选择是否冻结部分层,以平衡新特征学习和保留有用特征的需求。
效果:使模型在新任务上达到更好的性能,但需要更多的计算资源和时间,并容易过拟合。
Transformer block with adapters:一种有效的参数高效微调技术,它通过在预训练的 Transformer 模型的主干网络中添加额外的适配器(adapters)或残差块,来实现对特定任务的微调。这些适配器通常是可训练的参数,而模型的其他部分则保持固定。
Transformer block with adapters 的微调方法通常包括以下步骤:
加载预训练模型:首先,加载已经在大规模语料库上预训练好的 Transformer 模型。
添加适配器:在模型的主干网络中,为每个 Transformer 层添加适配器。这些适配器可以是简单的线性层或更复杂的结构,具体取决于任务的需求。
初始化适配器参数:为添加的适配器设置初始参数。这些参数通常是随机初始化的,也可以采用其他初始化策略。
进行微调:使用特定任务的数据集对模型进行微调。在微调过程中,仅更新适配器的参数,而保持模型的其他部分不变。
评估性能:在微调完成后,使用验证集评估模型的性能。根据评估结果,可以进一步调整适配器的结构或参数,以优化模型的性能。
| 欢迎光临 链载Ai (http://www.lianzai.com/) | Powered by Discuz! X3.5 |