迁移学习和模型的微调

迁移学习

对于新的数据,需要进行分类或者回归时,常用的方法是在以个大的数据集上(ImageNet包含120万张来自1000类别的数据集)进行预训练一个CNN模型,然后用这个预训练好的模型作为特征提取部分,或者网络的初始化权重应用在特定的任务中。
实际应用中,就是在预训练模型的后面接上全连接层,然后用softmax、SVM 或者贝叶斯对网络进行分类,也可以接上几层分类器用来做回归任务。

微调(Fine-tuning)

对于新的数据与原始数据相似度较高时,可以在原始数据训练得到的模型上对新的数据进行微调。
训练的方式为:

  1. 微调所有层
  2. 固定网络前几层权重,只微调网络的后面几层,这样做可以有两个原因:
    - (1)避免因数据量小造成过拟合现象
    - (2)CNN的前几层的特征中包含更多的一般特征(例如,边缘信息,色彩信息等) ,CNN的后面几层更多时提取高层特征也就是语义特征,针对数据而言,不同的数据集后面几层学习的语义特征也是完全不同。

对模型如何进行微调

如何确定应该对新数据集执行哪种类型的迁移学,这是几个因素的函数,但是两个最重要的因素是新数据集的大小(大小),以及它与原始数据集的相似性。

  1. 新数据集很小,与原始数据集相似 由于数据量很小,会容易过度拟合,微调网络网络不好,因此可以选择冻住前几层,对微调网络的后几层,也可以重新训练网络。
  2. 新数据集很大,并且与原始数据集相似。由于我们拥有更多数据,因此可以尝试通过整个网络进行微调,则不会过拟合。
  3. 新数据集很小,但与原始数据集有很大不同。由于数据很小,因此最好只训练线性分类器。由于数据集非常不同,因此最好不要从网络顶部训练分类器,该网络包含更多特定于数据集的功能。取而代之的是,从网络中较早的某个位置进行激活来训练SVM分类器可能会更好。
  4. 新数据集很大,并且与原始数据集有很大不同。由于数据集非常大,我们可以期望我们有能力从头开始训练网络。但是,实际上,使用来自预训练模型的权重进行初始化通常仍然是有益的。在这种情况下,我们将有足够的数据和信心来调整整个网络。

总结

从上面大致可以看出迁移学习其实就是一种学习得到的知识对另一种知识的学习的影响。迁移学习它可以看作是一种学习体系,关注的核心问题是,如何将之前学习的得到的“知识”运用到其他领域中去。而fine-tune只是迁移学习的一种实现手段。通俗的讲就是,有点像武侠小说中的“吸星大法”将别人修炼好的功力直接吸收过来,然后自己稍微打坐,疏通一下经脉,把吸收过来的工作化为己用。

引用 https://blog.csdn.net/AUTO1993/article/details/78677812?utm_source=blogxgwz5

更多详细介绍

  • 1
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
迁移学习微调都是机器学习中常用的技术,它们有一些相似之处,但也存在一些区别。 迁移学习是指将在一个任务上学到的知识或模型应用到另一个相关任务上的过程。在迁移学习中,通常会使用已经训练好的模型作为初始模型,然后通过调整模型的参数或添加新的层来适应新任务。迁移学习的目标是通过利用源任务上学到的知识来提高目标任务的性能,尤其是在目标任务的数据较少或者难以获取时。 微调迁移学习中的一种常见方法,它指的是在一个预训练好的模型基础上,通过在目标任务上进行少量的训练来调整模型参数。通常情况下,微调会保留预训练模型的大部分权重,并且只对最后几层或者全连接层进行重新训练。这样可以在保留预训练模型知识的同时,针对目标任务进行一定程度的特定调整。 区别: 1. 数据集:迁移学习通常涉及两个不同的任务,而微调是在一个任务上进行的。迁移学习可以将源任务和目标任务之间的数据集进行关联,而微调只需要目标任务的数据集。 2. 参数调整:迁移学习可以对整个模型进行参数调整,包括底层特征提取器和顶层分类器等,而微调通常只对顶层分类器进行参数调整。 3. 训练方式:迁移学习通常需要更多的训练时间和计算资源,因为它需要在源任务和目标任务上进行训练。而微调相对来说更加高效,因为它只需要在目标任务上进行少量的训练。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值