day49

@浙大疏锦行

  1. 预训练的概念

核心定义:先在大规模通用数据集上训练一个基础模型(学习通用特征,如边缘、纹理、形状),再将这个模型的参数迁移到目标任务中,进行微调(Fine-tuning)或直接使用。

本质:利用 "通用数据的特征迁移",解决目标任务数据量少、训练成本高的问题,是迁移学习的核心方式之一。

作用:大幅降低目标任务的训练时间;提升小数据集任务的模型性能;避免从零训练模型的过拟合风险。

2.经典的预训练模型

3.预训练的策略

(1)冻结预训练层(仅微调头部):

操作:冻结预训练模型的大部分层(仅保留参数不更新),只修改模型的 "分类头",仅训练分类头。

适用场景:目标任务数据量少、预训练模型特征足够通用。

(2)全量微调(Fine-tuning):

操作:解冻预训练模型的部分 / 全部层,让所有层的参数随目标任务更新。

适用场景:目标任务数据量充足,需要模型适配任务的专属特征。

(3)特征提取(Feature Extraction):

操作:直接用预训练模型的输出作为目标任务的特征,不更新预训练模型参数,仅训练新的分类器。

适用场景:目标任务与预训练任务高度相似,预训练特征可直接复用。

用预训练 ResNet18 完成 CIFAR10 分类

在CIFAR10 对比不同预训练模型(以 MobileNetV2 为例)

相关推荐
共享家95276 小时前
搭建 AI 聊天机器人:”我的人生我做主“
前端·javascript·css·python·pycharm·html·状态模式
Hgfdsaqwr6 小时前
Python在2024年的主要趋势与发展方向
jvm·数据库·python
一晌小贪欢7 小时前
Python 测试利器:使用 pytest 高效编写和管理单元测试
python·单元测试·pytest·python3·python测试
小文数模7 小时前
2026年美赛数学建模C题完整参考论文(含模型和代码)
python·数学建模·matlab
Halo_tjn7 小时前
基于封装的专项 知识点
java·前端·python·算法
Hgfdsaqwr7 小时前
掌握Python魔法方法(Magic Methods)
jvm·数据库·python
weixin_395448917 小时前
export_onnx.py_0130
pytorch·python·深度学习
s1hiyu7 小时前
使用Scrapy框架构建分布式爬虫
jvm·数据库·python
2301_763472467 小时前
使用Seaborn绘制统计图形:更美更简单
jvm·数据库·python
无垠的广袤8 小时前
【VisionFive 2 Lite 单板计算机】边缘AI视觉应用部署:缺陷检测
linux·人工智能·python·opencv·开发板