Imbalanced datasets
millanp95 opened this issue · 1 comments
millanp95 commented
Hi,
Thank you for this implementation. It is my understanding that some contrastive frameworks build upon entropy maximization which leads to inapplicability in the contexts of imbalanced datasets. I don't see the direct connection between the ProtoNCE loss and the entropy so I was wondering, does this method support imbalanced datasets?
Thanks
LiJunnan1992 commented
Hi, thanks for your question.
ProtoNCE should be able to support imbalanced datasets, since it is an unsupervised learning method which is unaware of the class distribution.