如何在深度学习中向现有分类器添加新类？

Question

我训练了一个深度学习模型，将给定的图像分为三类。现在我想在我的模型中再添加一个类。我尝试查看“在线学习”，但它似乎是在现有课程的新数据上进行训练。我是否需要在所有四个类别上再次训练我的整个模型，或者有什么方法可以在新类别上训练我的模型？

Answer 1

您可能在架构末尾的 3 个神经元密集层之后使用了 softmax，将其分为 3 个类别。添加一个类将导致在 4 个神经元密集层上执行 softmax，因此将无法在当前图中使用冻结权重容纳额外的神经元，基本上您正在修改图，因此您必须训练整个图从头开始的模型

- - -或者 - - -

一种方法是加载模型并删除最后一层，将其更改为 4 个神经元并再次训练网络！这基本上将从头开始训练最后一层的权重。我认为在添加新类时无论如何都无法保持这些（最后一层的权重）权重完整。