day49

@浙大疏锦行

  1. 预训练的概念

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

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

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

2.经典的预训练模型

3.预训练的策略

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

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

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

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

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

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

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

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

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

用预训练 ResNet18 完成 CIFAR10 分类

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

相关推荐
cup116 小时前
[技术复盘] Windows Python 打包实战:Nuitka 环境踩坑总结与 CI 自动化构建全指南
python·ai·环境变量·ci·nuitka·skill
aqi008 小时前
15天学会AI应用开发(七)有了大模型为什么还要引入RAG
人工智能·python·大模型·ai编程·ai应用
金銀銅鐵10 小时前
用 Python 实现 Take-Away 游戏
python·游戏
copyer_xyf11 小时前
Agent 流程编排
后端·python·agent
copyer_xyf12 小时前
Agent RAG
后端·python·agent
copyer_xyf12 小时前
【RAG】向量数据库:milvus
后端·python·agent
copyer_xyf12 小时前
Agent 记忆管理
后端·python·agent
星云穿梭1 天前
用Python写一个带图形界面的学生管理系统——完整教程
python
金銀銅鐵1 天前
用 Pygame 实现 15 puzzle
python·数学·游戏