什么是上游任务、下游任务?

文章介绍了预训练模型的概念,特别是如何通过上游任务如图像分类或自监督学习来构建具备视觉表征能力的模型,如CNN或Transformer。下游任务是指针对具体问题的微调,如使用COCO或ImageNet数据集。预训练模型的最后层通常被修改以适应不同任务,而原始模型的主体部分(backbone)用于提取特征。作者建议深入学习预训练模型的微调技巧。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

①上游任务
预训练模型。一般就是利用上游数据进行预训练,以生成一个包含视觉表征能力的模型。
比如,我们想要的是一个能够提取图片特征能力的卷积神经网络或者Transformer,我们会用大量图片用图片分类这个下游任务或者其他比如自监督的方法(可以参考CLIP)去进行训练,得到一个权重合适的模型(能够很好地提取出图像的特征),那么最后我们把得到的这个模型最后一层的FC层(原本用于图片分类输出类别)去掉,这个模型就成为了一个很好的预训练模型,输入一张图,就能够提取出图像的特征,就可以用于我们的下游任务(在这个模型后面加一些诸如检测头之类的模块,处理我们想要的下游任务,或者修改下FC层,用于另外一堆类别的图像分类)。
上游任务一般称为backbone,主干网络。

②下游任务
下游任务:真正想要解决的任务。如果你想训练一个网络无论是生成任务还是检测任务,你可能会使用一些公开的数据集进行训练,例如coco,imagenet之类的公共数据集进行训练,而这些数据集可能不会很好完成你真正想完成的内容,这就意味着在解决的实际问题的数据集上,要微调这个预训练模型,而这个任务称为下游任务。

建议再学学如何进行预训练模型的微调

参考CSDN:
下游任务(downstream task)含义
【计算机视觉】上游任务和下游任务的理解

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值