I am working on image segmentation task with image and 7 [2, 5] labels given in .npy format. I checked the class frequency in each by taking the sum of ones since the labels are readily one-hot encoded. Based on suggested approach, I using the weighted (median sampling frequency) dice+CrossEntropy loss but the performance is not improving.
Are there other approach that can be used to handle such imbalance for improved performance?