Fighting Class Imbalance with Contrastive Learning

Medical Image Computing and Computer-Assisted Intervention (MICCAI), 2021
Abstract: Medical image datasets are hard to collect, expensive to label, and often highly imbalanced. The last issue is underestimated, as typical average metrics hardly reveal that the often very important minority classes have a very low accuracy. In this paper, we address this problem by a feature embedding that balances the classes using contrastive learning as an alternative to the common cross-entropy loss. The approach is largely orthogonal to existing sampling methods and can be easily combined with those. We show on the challenging ISIC2018 and APTOS2019 datasets that the approach improves especially the accuracy of minority classes without negatively affecting the majority ones.

Images and movies


BibTex reference

  author       = "Y. Marrakchi and O. Makansi and T. Brox",
  title        = "Fighting Class Imbalance with Contrastive Learning",
  booktitle    = "Medical Image Computing and Computer-Assisted Intervention (MICCAI)",
  month        = " ",
  year         = "2021",
  url          = "http://lmb.informatik.uni-freiburg.de/Publications/2021/MMB21"

Other publications in the database