Supervised contrastive learning