Distance Map Loss Penalty Term for Semantic Segmentation
Convolutional neural networks for semantic segmentation suffer from low performance at object boundaries. In medical imaging, accurate representation of tissue surfaces and volumes is important for tracking of disease biomarkers such as tissue morphology and shape features. In this work, we propose a novel distance map derived loss penalty term for semantic segmentation. We propose to use distance maps, derived from ground truth masks, to create a penalty term, guiding the network's focus towards hard-to-segment boundary regions. We investigate the effects of this penalizing factor against cross-entropy, Dice, and focal loss, among others, evaluating performance on a 3D MRI bone segmentation task from the publicly available Osteoarthritis Initiative dataset. We observe a significant improvement in the quality of segmentation, with better shape preservation at bone boundaries and areas affected by partial volume. We ultimately aim to use our loss penalty term to improve the extraction of shape biomarkers and derive metrics to quantitatively evaluate the preservation of shape.
Code (0)
등록된 구현이 없습니다.
Tasks
SegmentationSemantic SegmentationSimilar Papers 제목 키워드 기반
Incorporating Boundary Uncertainty into loss functions for biomedical image segmentation
Manual segmentation is used as the gold-standard for evaluating neural networks on automated image segmentation tasks. Due to considerable heterogeneity in shapes, colours and textures, demarcating object boundaries is p…
Image SegmentationSegmentationSemantic SegmentationDeep Distance Map Regression Network with Shape-aware Loss for Imbalanced Medical Image Segmentation
Small object segmentation, like tumor segmentation, is a difficult and critical task in the field of medical image analysis. Although deep learning based methods have achieved promising performance, they are restricted t…
Image SegmentationMedical Image AnalysisMedical Image Segmentationregression+3Topology-Aware Focal Loss for 3D Image Segmentation
The efficacy of segmentation algorithms is frequently compromised by topological errors like overlapping regions, disrupted connections, and voids. To tackle this problem, we introduce a novel loss function, namely Topol…
Brain Tumor SegmentationImage SegmentationSegmentationSemantic Segmentation+1Leveraging point annotations in segmentation learning with boundary loss
This paper investigates the combination of intensity-based distance maps with boundary loss for point-supervised semantic segmentation. By design the boundary loss imposes a stronger penalty on the false positives the fa…
Computational EfficiencyHeart SegmentationOrgan SegmentationSegmentation+1Learning Non-Unique Segmentation with Reward-Penalty Dice Loss
Semantic segmentation is one of the key problems in the field of computer vision, as it enables computer image understanding. However, most research and applications of semantic segmentation focus on addressing unique se…
Medical Image SegmentationSegmentationSemantic Segmentation