如何为现有的训练有素的深度学习模型添加新类别?

数据挖掘 深度学习 喀拉斯 美国有线电视新闻网 计算机视觉 迁移学习
2022-02-22 11:02:00

我最初用 5 个类训练了我的深度学习模型,现在我想添加另一个类,而不需要为这 5 个类重新训练整个模型。我怎样才能做到这一点?

2个回答

如果不重新训练模型的至少一部分,你就无法做到这一点。

您必须将现有的输出层替换为具有所需神经元数量的新输出层。当然,这意味着您必须至少重新训练最后一层或最后几层。但是您可以冻结所有其他层的权重。

trainable在 keras 中,您可以通过将其属性设置为 来冻结特定层的权重False

如何在没有对其他类进行完整训练的情况下向训练模型添加更多类:

  • 迁移学习两次
  • 持续学习方法
    • 正则化
    • 扩张
    • 排演