TransWikia.com

How to add a new category to a existing trained deep learning model?

Data Science Asked by Subham Tiwari on May 11, 2021

i have trained my deep learning model initially with 5 classes now i want to add another class without training the whole model over again for those 5 classes. How can I do that?

2 Answers

You cannot do that without re-training at least part of the model.

You will have to replace the existing output layer with a new output layer that has the desired number of neurons. That, of course, means that you will have to retrain at least the last layer or the last few layers. But you can freeze the weights of all the other layers.

In keras you can freeze the weights of a specific layer by setting its trainable property to False.

Answered by georg-un on May 11, 2021

How to add more class to trained model without complete training for other classes:

  • Transfer Learning Twice
  • Continual learning approaches
    • Regularization
    • Expansion
    • Rehearsal

Answered by Farshid PirahanSiah on May 11, 2021

Add your own answers!

Ask a Question

Get help from others!

© 2024 TransWikia.com. All rights reserved. Sites we Love: PCI Database, UKBizDB, Menu Kuliner, Sharing RPP