在哪里可以找到用于迁移学习的预训练模型

机器算法验证 机器学习 分类 神经网络 迁移学习
2022-03-09 09:11:34

我是机器学习领域的新手,但我想尝试用 Keras 实现一个简单的分类算法。不幸的是,我的数据集非常少,所以我想尝试将迁移学习应用到这个问题上;但是,我在网上找不到任何东西,所以我想了解哪些是寻找预训练神经网络的最佳地点。在这方面您有什么建议吗?哪个网站最适合了解如何启动机器学习项目?

2个回答

Keras 本身提供了一些在 ImageNet 上预训练的成功的图像处理神经网络: https ://keras.io/applications/

其他深度学习库也提供了一些预训练模型,特别是:

还可以在https://www.gradientzoo.com找到许多适用于各种平台的预训练模型。

此外,如果您对某些特定的网络架构感兴趣,作者有时会自己提供预训练模型,例如ResNeXt

由于问题标题是通用的(而不是特定于计算机视觉),我也会给出与NLP相关的答案,以防它对偶然发现预训练向量嵌入的人有所帮助:

可以在这些链接上找到两个最流行的预训练向量嵌入:

还有一些不太受欢迎和/或更近期的: