当前位置:网站首页>一、迁移学习与fine-tuning有什么区别?
一、迁移学习与fine-tuning有什么区别?
2022-07-29 05:22:00 【MY头发乱了】
一、迁移学习
举个例子,假设boss让你做一下目标检测,这个数据集是关于光纤箱喷码字符检测的。问题是,数据集很少(只有1000张数据),里面有多干扰的信息,你发现从零训练开始训练yolo的效果很差,很容易过拟合。怎么办呢,于是你想到了使用Transfer Learning,用别人已经训练好的Imagenet的模型来做。
二、迁移学习有哪些手法
- 把Alexnet里卷积层最后一层输出的特征拿出来,然后直接用SVM分类。这是Transfer Learning,因为你用到了Alexnet中已经学到了的“知识”。
- Vggnet卷积层最后的输出拿出来,用贝叶斯分类器分类。思想基本同上。
- 至你可以把Alexnet、Vggnet的输出拿出来进行组合,自己设计一个分类器分类。这个过程中你不仅用了Alexnet的“知识”,也用了Vggnet的“知识”。
- 最后,你也可以直接使用fine-tune这种方法,在Alexnet的基础上,重新加上全连接层,再去训练网络。
三、fine-tune使用策略
决定如何使用迁移学习的因素有很多,这是最重要的只有两个:新数据集的大小、以及新数据和原数据集的相似程度。有一点一定记住:网络前几层学到的是通用特征,后面几层学到的是与类别相关的特征。这里有使用的四个场景:
- 新数据集比较小且和原数据集相似。因为新数据集比较小,如果fine-tune可能会过拟合;又因为新旧数据集类似,我们期望他们高层特征类似,可以使用预训练网络当做特征提取器,用提取的特征训练线性分类器。
- 新数据集大且和原数据集相似。因为新数据集足够大,可以fine-tune整个网络。
- **新数据集小且和原数据集不相似。**新数据集小,最好不要fine-tune,和原数据集不类似,最好也不使用高层特征。这时可是使用前面层的特征来训练SVM分类器。
- **新数据集大且和原数据集不相似。**因为新数据集足够大,可以重新训练。但是实践中fine-tune预训练模型还是有益的。新数据集足够大,可以fine-tine整个网络。
四、总结
综上,Transfer Learning关心的问题是:什么是“知识”以及如何更好地运用之前得到的“知识”。这可以有很多方法和手段。而fine-tune只是其中的一种手段。
边栏推荐
- Spring, summer, autumn and winter with Miss Zhang (4)
- Valuable blog and personal experience collection (continuous update)
- clion+opencv+aruco+cmake配置
- 并发编程学习笔记 之 原子操作类AtomicReference、AtomicStampedReference详解
- Flutter正在被悄悄放弃?浅析Flutter的未来
- PyTorch基础知识(可入门)
- 迁移学习—— Transfer Feature Learning with Joint Distribution Adaptation
- 研究生新生培训第三周:ResNet+ResNeXt
- anaconda中移除旧环境、增加新环境、查看环境、安装库、清理缓存等操作命令
- 【语义分割】语义分割综述
猜你喜欢

并发编程学习笔记 之 工具类Semaphore(信号量)

【目标检测】Generalized Focal Loss V1

简单聊聊 PendingIntent 与 Intent 的区别

clion+opencv+aruco+cmake配置

迁移学习—— Transfer Feature Learning with Joint Distribution Adaptation

GAN:生成对抗网络 Generative Adversarial Networks
![[semantic segmentation] setr_ Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformer](/img/aa/daccb45d5b6d4a3f7e8517dd5bd7d2.png)
[semantic segmentation] setr_ Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformer

【pycharm】pycharm远程连接服务器

ANR优化:导致 OOM 崩溃及相对应的解决方案
![[semantic segmentation] Introduction to mapillary dataset](/img/3d/49b1b6007c1f97b1ba8ac438097590.png)
[semantic segmentation] Introduction to mapillary dataset
随机推荐
GAN:生成对抗网络 Generative Adversarial Networks
个人学习网站
性能优化之趣谈线程池:线程开的越多就越好吗?
【go】defer的使用
PHP write a diaper to buy the lowest price in the whole network
Spring, summer, autumn and winter with Miss Zhang (2)
有价值的博客、面经收集(持续更新)
[go] use of defer
FFmpeg创作GIF表情包教程来了!赶紧说声多谢乌蝇哥?
深入理解MMAP原理,让大厂都爱不释手的技术
这些你一定要知道的进程知识
【bug】XLRDError: Excel xlsx file; not supported
Markdown syntax
【Transformer】SegFormer:Simple and Efficient Design for Semantic Segmentation with Transformers
GA-RPN:引导锚点的建议区域网络
Ribbon学习笔记二
Detailed explanation of MySQL statistical function count
Flutter 绘制技巧探索:一起画箭头(技巧拓展)
DataX installation
Windos下安装pyspider报错:Please specify --curl-dir=/path/to/built/libcurl解决办法