如何正确使用迁移学习

在使用迁移学习时,关键在于数据规模和与预训练模型数据的相似度。小数据且相似时,适合特征提取;大数据且相似时,可微调网络。数据不同但规模小时,应从低层提取特征;大数据且不同,从头训练或微调预训练模型作对比。
摘要由CSDN通过智能技术生成

在使用迁移学习之前,有两点需要特别注意的地方:

  • 你的数据的尺寸大小
  • 你的数据和预训练好网络结构数据(典型的比如说是ImageNet)的相似程度

关注上面的问题可以帮助我们确定何种数据迁移策略。我们以表格的形式来展示一下。

  相似数据 不同数据
小数据 使用全连接进行特征提取+分类 使用高层卷积特征提取+分类
大数据 可以尝试微调网络
也可以尝试从头训练自己的网络
微调网络可能有效果也可能没有效果
可以尝试从头训练自己的网络
原始样本数据是小数据并且和要迁移模型数据相似

因为数据较小,所以很有可能没有办法从头获取CNN特征而不发生过拟合,微调网络也有可能会遇到这样的问题。

原始数据和预训练模型的数据很像,用来进行特征提取通常是一个不错的办法。然后使用一个简单的机器学习分类器对这些特征进行分类。通常建议可以从更深的网络结构中(更高的层)提取特征,因为这些特征更丰富,更能代表中原始数据中学到的模式。

原始样本数据是大数据并且和要迁移模型的数据相似

对于大数据,微调网络通常不会发生过拟合现象。你也可以尝试自己从头训练一个模型(至少这个想法值得做实验尝试一下)。因为原始样本数据和要迁移模型数据十分相似,那么迁移的预训练模型可能已经有了足够好的特征,因此使用微调网络会很不错。

原始样本是小数据并且和要迁移模型数据不同

这是最糟糕的一种情况。
为了防止过拟合,我们不可能从新训

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Einstellung

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值