METHOD AND SYSTEM FOR MACHINE LEARNING FROM IMBALANCED DATA WITH NOISY LABELS

A computer-implemented method for training an artificial neural network with training data including samples and corresponding labels for performing a task includes: pre-training the artificial neural network to generate matrix representations that are invariant to a predetermined set of data augmentations applied to a sample, where the artificial neural network includes an encoder module and a projection module configured to generate the matrix representations based on ones of the samples, respectively; and after the pre-training, fine-tune training the artificial neural network using a loss function, wherein fine-tuning the artificial neural network includes adjusting, based on the labels, one or more weights of the projection module while maintaining constant weights of the encoder module, and where the loss function is based on a logit adjustment loss that is based on logits that are adjusted based on a class distribution of the training data.