A novel shape-based loss function for machine learning-based seminal organ segmentation in medical imaging

03/07/2022
by   Reza Karimzadeh, et al.
30

Automated medical image segmentation is an essential task to aid/speed up diagnosis and treatment procedures in clinical practices. Deep convolutional neural networks have exhibited promising performance in accurate and automatic seminal segmentation. For segmentation tasks, these methods normally rely on minimizing a cost/loss function that is designed to maximize the overlap between the estimated target and the ground-truth mask delineated by the experts. A simple loss function based on the degrees of overlap (i.e., Dice metric) would not take into account the underlying shape and morphology of the target subject, as well as its realistic/natural variations; therefore, suboptimal segmentation results would be observed in the form of islands of voxels, holes, and unrealistic shapes or deformations. In this light, many studies have been conducted to refine/post-process the segmentation outcome and consider an initial guess as prior knowledge to avoid outliers and/or unrealistic estimations. In this study, a novel shape-based cost function is proposed which encourages/constrains the network to learn/capture the underlying shape features in order to generate a valid/realistic estimation of the target structure. To this end, the Principal Component Analysis (PCA) was performed on a vectorized training dataset to extract eigenvalues and eigenvectors of the target subjects. The key idea was to use the reconstruction weights to discriminate valid outcomes from outliers/erroneous estimations.

READ FULL TEXT

page 7

page 11

page 12

01/07/2022

Effect of Prior-based Losses on Segmentation Performance: A Benchmark

Today, deep convolutional neural networks (CNNs) have demonstrated state...
02/27/2022

Topology-Preserving Segmentation Network: A Deep Learning Segmentation Framework for Connected Component

Medical image segmentation, which aims to automatically extract anatomic...
02/11/2019

Psi-Net: Shape and boundary aware joint multi-task deep network for medical image segmentation

Medical image segmentation is a primary task in many applications, and t...
06/21/2018

Star Shape Prior in Fully Convolutional Networks for Skin Lesion Segmentation

Semantic segmentation is an important preliminary step towards automatic...
11/16/2020

High-level Prior-based Loss Functions for Medical Image Segmentation: A Survey

Today, deep convolutional neural networks (CNNs) have demonstrated state...
06/27/2021

Residual Moment Loss for Medical Image Segmentation

Location information is proven to benefit the deep learning models on ca...
11/20/2020

Segmentation overlapping wear particles with few labelled data and imbalance sample

Ferrograph image segmentation is of significance for obtaining features ...