Model training with different class size

How to train deep learning model, if one class has large data compared to other classes?


There are many ways to deal with this situation and actually this is an area of research. So, generally, here are few options you can select:

  • Using sampling in the way that in each batch, same size of classes exists.
    How to handle imbalanced classes
  • Weighting loss function in way that biased toward rare class. How to handle imbalanced classes
  • Using ensemble of ANNs using Boosting approaches for generating synthesized data or sampling not biased data such as RBBoost, SMOTEBoost.

There are other ways for sure which you need to find which one suits your case the best.


1 Like

thanks , I will try that.