Abstract:Supervised learning-based medical image segmentation solutions usually require sufficient labeled training data. Insufficient available labeled training data often leads to the limitations of model performances, such as over-fitting, low accuracy, and poor generalization ability. However, this dilemma may worsen in the field of medical image analysis. Medical image annotation is usually labor-intensive and professional work. In this work, we propose a novel shape and boundary-aware deep learning model for medical image segmentation based on semi-supervised learning. The model makes good use of labeled data and also enables unlabeled data to be well applied by using task consistency loss. Firstly, we adopt V-Net for Pixel-wise Segmentation Map (PSM) prediction and Signed Distance Map (SDM) regression. In addition, we multiply multi-scale features, extracted by Pyramid Pooling Module (PPM) from input X, with 2 - |SDM| to enhance the features around the boundary of the segmented target, and then feed them into the Feature Fusion Module (FFM) for fine segmentation. Besides boundary loss, the high-level semantics implied in SDM facilitate the accurate segmentation of boundary regions. Finally, we get the ultimate result by fusing coarse and boundary-enhanced features. Last but not least, to mine unlabeled training data, we impose consistency constraints on the three core outputs of the model, namely PSM1, SDM, and PSM3. Through extensive experiments over three representative but challenging medical image datasets (LA2018, BraTS2019, and ISIC2018) and comparisons with the existing representative methods, we validate the practicability and superiority of our model.

Shape Description Losses for Medical Image Segmentation

Segmentation of Cardiac Tissues and Organs for CCTA Images Based on a Deep Learning Model

Dual-Term Loss Function for Shape-Aware Medical Image Segmentation.

Loss odyssey in medical image segmentation

Adaptive Region-Specific Loss for Improved Medical Image Segmentation

Shape-aware Semi-supervised 3D Semantic Segmentation for Medical Images

A Modified Level Set Algorithm Based on Point Distance Shape Constraint for Lesion and Organ Segmentation

Shape-Sensitive Loss for Catheter and Guidewire Segmentation

Residual Moment Loss for Medical Image Segmentation

Boundary-wise Loss for Medical Image Segmentation Based on Fuzzy Rough Sets

Offset Curves Loss for Imbalanced Problem in Medical Segmentation

Geometric Loss For Deep Multiple Sclerosis Lesion Segmentation

Enhancing Foreground Boundaries for Medical Image Segmentation

Unsupervised Domain Adaptation through Shape Modeling for Medical Image Segmentation

Deformable Segmentation Via Sparse Shape Representation.

Deep Level Set with Confidence Map and Boundary Loss for Medical Image Segmentation

Shape Prior-Constrained Deep Learning Network for Medical Image Segmentation

Shape and boundary-aware multi-branch model for semi-supervised medical image segmentation

Shape-intensity knowledge distillation for robust medical image segmentation

Shape Constrained CNN for Cardiac MR Segmentation with Simultaneous Prediction of Shape and Pose Parameters

Contour-weighted loss for class-imbalanced image segmentation