CUAMT: A MRI semi-supervised medical image segmentation framework based on contextual information and mixed uncertainty.

Authors

Xiao H,Wang Y,Xiong S,Ren Y,Zhang H

Affiliations (5)

  • School of Artificial Intelligence, Chongqing University of Technology, Chongqing 401135, China. Electronic address: [email protected].
  • School of Artificial Intelligence, Chongqing University of Technology, Chongqing 401135, China. Electronic address: [email protected].
  • School of Artificial Intelligence, Chongqing University of Technology, Chongqing 401135, China. Electronic address: [email protected].
  • School of Artificial Intelligence, Chongqing University of Technology, Chongqing 401135, China. Electronic address: [email protected].
  • School of Artificial Intelligence, Chongqing University of Technology, Chongqing 401135, China. Electronic address: [email protected].

Abstract

Semi-supervised medical image segmentation is a class of machine learning paradigms for segmentation model training and inference using both labeled and unlabeled medical images, which can effectively reduce the data labeling workload. However, existing consistency semi-supervised segmentation models mainly focus on investigating more complex consistency strategies and lack efficient utilization of volumetric contextual information, which leads to vague or uncertain understanding of the boundary between the object and the background by the model, resulting in ambiguous or even erroneous boundary segmentation results. For this reason, this study proposes a hybrid uncertainty network CUAMT based on contextual information. In this model, a contextual information extraction module CIE is proposed, which learns the connection between image contexts by extracting semantic features at different scales, and guides the model to enhance learning contextual information. In addition, a hybrid uncertainty module HUM is proposed, which guides the model to focus on segmentation boundary information by combining the global and local uncertainty information of two different networks to improve the segmentation performance of the networks at the boundary. In the left atrial segmentation and brain tumor segmentation dataset, validation experiments were conducted on the proposed model. The experiments show that our model achieves 89.84%, 79.89%, and 8.73 on the Dice metric, Jaccard metric, and 95HD metric, respectively, which significantly outperforms several current SOTA semi-supervised methods. This study confirms that the CIE and HUM strategies are effective. A semi-supervised segmentation framework is proposed for medical image segmentation.

Topics

Magnetic Resonance ImagingImage Processing, Computer-AssistedJournal Article

Ready to Sharpen Your Edge?

Join hundreds of your peers who rely on RadAI Slice. Get the essential weekly briefing that empowers you to navigate the future of radiology.

We respect your privacy. Unsubscribe at any time.