迁移学习基础模型的选择

数据挖掘 深度学习 图像分类 迁移学习
2022-02-23 21:37:10

是否有一条黄金法则可以直观地说明需要使用哪个基本模型来解决给定的图像分类问题。

大多数文章都提供了以下详细信息,说明了如何根据数据集训练模型。

微调模型的规则(取自 analyticsvidya)

但是,我无法找到选择基本模型的良好参考。

谢谢

1个回答

没有与迁移学习的基本模型选择相关的特定规则。它通常是模型精度和资源分配之间的权衡。随着层数的增加,参数的数量增加,模型变得越来越重,但更深的模型往往比浅层模型具有更好的准确性。这是一个比较:

转换拱门

除此之外还要参考这个问题:哪个是最快的图像预训练模型?如果模型的速度很重要。