Systems and methods for training an artificial intelligence-based system using self-supervised learning are provided. For each respective training medical image of a set of unannotated training medical images, the following steps are performed. A first augmented image is generated by applying a first augmentation operation to the respective training medical image. A second augmented image is generated by applying a second augmentation operation to the respective training medical image. A first representation vector is created from the first augmented image using an encoder network. A second representation vector is created from the second augmented image using the encoder network. The first representation vector is mapped to first cluster codes. The second representation vector is mapped to second cluster codes. The encoder network is optimized using the first and second representation vectors and the first and second cluster codes.