Systems and methods of automatic medical image labeling

Abstract

A computer-implemented method of automatically labeling medical images is provided. The method includes clustering training images and training labels into clusters, each cluster including a representative template having a representative image and a representative label. The method also includes training a neural network model with a training dataset that includes the training images and the training labels, and target outputs of the neural network model are labels of the medical images. The method further includes generating a suboptimal label corresponding to an unlabeled test image using the trained neural network model, and generating an optimal label corresponding to the unlabeled test image using the suboptimal label and representative templates. In addition, the method includes updating the training dataset using the test image and the optimal label, retraining the neural network model, generating a label of an unlabeled image using the retrained neural network model, and outputting the generated label.