• Optoelectronics Letters
  • Vol. 18, Issue 9, 547 (2022)
Enrong ZHU1, Haochen ZHAO2, and Xiaofei and HU1、*
Author Affiliations
  • 1School of Geography and Bioinformatics, Nanjing University of Posts and Telecommunications, Nanjing 210003, China
  • 2School of Communication and Information Engineering, Nanjing University of Posts and Telecommunications, Nanjing 210003, China
  • show less
    DOI: 10.1007/s11801-022-2010-0 Cite this Article
    ZHU Enrong, ZHAO Haochen, and HU Xiaofei. Semi-supervised cardiac MRI image of the left ventricle segmentation algorithm based on contrastive learning[J]. Optoelectronics Letters, 2022, 18(9): 547 Copy Citation Text show less

    Abstract

    A semi-supervised convolutional neural network segmentation method of medical images based on contrastive learning is proposed. The cardiac magnetic resonance imaging (MRI) images to be segmented are preprocessed to obtain positive and negative samples by labels. The U-Net shrinks network is applied to extract features of the positive samples, negative samples, and input samples. In addition, an unbalanced contrastive loss function is proposed, which is weighted with the binary cross-entropy loss function to obtain the total loss function. The model is pre-trained with labeled samples, and unlabeled images are predicted by the pre-trained model to generate pseudo-labels. A pseudo-label post-processing algorithm for removing disconnected regions and hole filling of pseudo-labels is proposed to guide the training process of semi-supervised networks. The results on the Sunnybrook dataset show that the segmentation results of this model are better, with a higher dice coefficient, accuracy, and recall rate.
    ZHU Enrong, ZHAO Haochen, and HU Xiaofei. Semi-supervised cardiac MRI image of the left ventricle segmentation algorithm based on contrastive learning[J]. Optoelectronics Letters, 2022, 18(9): 547
    Download Citation