Optoelectronics Letters, Volume. 18, Issue 9, 547(2022)

Semi-supervised cardiac MRI image of the left ventricle segmentation algorithm based on contrastive learning

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

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Jan. 20, 2022

    Accepted: May. 7, 2022

    Published Online: Jan. 20, 2023

    The Author Email: Xiaofei and HU (huxf@njupt.edu.cn)

    DOI:10.1007/s11801-022-2010-0

    Topics