A novel shape-based loss function for machine learning-based seminal
organ segmentation in medical imaging
- URL: http://arxiv.org/abs/2203.03336v1
- Date: Mon, 7 Mar 2022 12:26:30 GMT
- Title: A novel shape-based loss function for machine learning-based seminal
organ segmentation in medical imaging
- Authors: Reza Karimzadeh, Emad Fatemizadeh, Hossein Arabi
- Abstract summary: Deep convolutional neural networks have exhibited promising performance in accurate and automatic seminal segmentation.
A novel shape-based cost function is proposed which encourages/constrains the network to learn/capture the underlying shape features.
- Score: 1.0312968200748116
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: 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.
Related papers
- SINDER: Repairing the Singular Defects of DINOv2 [61.98878352956125]
Vision Transformer models trained on large-scale datasets often exhibit artifacts in the patch token they extract.
We propose a novel fine-tuning smooth regularization that rectifies structural deficiencies using only a small dataset.
arXiv Detail & Related papers (2024-07-23T20:34:23Z) - Shape Matters: Detecting Vertebral Fractures Using Differentiable
Point-Based Shape Decoding [51.38395069380457]
Degenerative spinal pathologies are highly prevalent among the elderly population.
Timely diagnosis of osteoporotic fractures and other degenerative deformities facilitates proactive measures to mitigate the risk of severe back pain and disability.
In this study, we specifically explore the use of shape auto-encoders for vertebrae.
arXiv Detail & Related papers (2023-12-08T18:11:22Z) - A Novel Information-Theoretic Objective to Disentangle Representations
for Fair Classification [46.884905701771004]
One of the main application for such disentangled representations is fair classification.
We adopt an information-theoretic view of this problem which motivates a novel family of regularizers.
The resulting set of losses, called CLINIC, is parameter free and thus, it is easier and faster to train.
arXiv Detail & Related papers (2023-10-21T12:35:48Z) - Multi-task Explainable Skin Lesion Classification [54.76511683427566]
We propose a few-shot-based approach for skin lesions that generalizes well with few labelled data.
The proposed approach comprises a fusion of a segmentation network that acts as an attention module and classification network.
arXiv Detail & Related papers (2023-10-11T05:49:47Z) - Fuzzy Attention Neural Network to Tackle Discontinuity in Airway
Segmentation [67.19443246236048]
Airway segmentation is crucial for the examination, diagnosis, and prognosis of lung diseases.
Some small-sized airway branches (e.g., bronchus and terminaloles) significantly aggravate the difficulty of automatic segmentation.
This paper presents an efficient method for airway segmentation, comprising a novel fuzzy attention neural network and a comprehensive loss function.
arXiv Detail & Related papers (2022-09-05T16:38:13Z) - Self-Supervised-RCNN for Medical Image Segmentation with Limited Data
Annotation [0.16490701092527607]
We propose an alternative deep learning training strategy based on self-supervised pretraining on unlabeled MRI scans.
Our pretraining approach first, randomly applies different distortions to random areas of unlabeled images and then predicts the type of distortions and loss of information.
The effectiveness of the proposed method for segmentation tasks in different pre-training and fine-tuning scenarios is evaluated.
arXiv Detail & Related papers (2022-07-17T13:28:52Z) - Unsupervised Domain Adaptation through Shape Modeling for Medical Image
Segmentation [23.045760366698634]
We aim at modeling shape explicitly and using it to help medical image segmentation.
Previous methods proposed Variational Autoencoder (VAE) based models to learn the distribution of shape for a particular organ.
We propose a new unsupervised domain adaptation pipeline based on a pseudo loss and a VAE reconstruction loss under a teacher-student learning paradigm.
arXiv Detail & Related papers (2022-07-06T09:16:42Z) - Learning Fuzzy Clustering for SPECT/CT Segmentation via Convolutional
Neural Networks [5.3123694982708365]
Quantitative bone single-photon emission computed tomography (QBSPECT) has the potential to provide a better quantitative assessment of bone metastasis than planar bone scintigraphy.
The segmentation of anatomical regions-of-interests (ROIs) still relies heavily on the manual delineation by experts.
This work proposes a fast and robust automated segmentation method for partitioning a QBSPECT image into lesion, bone, and background.
arXiv Detail & Related papers (2021-04-17T19:03:52Z) - An Uncertainty-Driven GCN Refinement Strategy for Organ Segmentation [53.425900196763756]
We propose a segmentation refinement method based on uncertainty analysis and graph convolutional networks.
We employ the uncertainty levels of the convolutional network in a particular input volume to formulate a semi-supervised graph learning problem.
We show that our method outperforms the state-of-the-art CRF refinement method by improving the dice score by 1% for the pancreas and 2% for spleen.
arXiv Detail & Related papers (2020-12-06T18:55:07Z) - Learning to Segment Anatomical Structures Accurately from One Exemplar [34.287877547953194]
Methods that permit to produce accurate anatomical structure segmentation without using a large amount of fully annotated training images are highly desirable.
We propose Contour Transformer Network (CTN), a one-shot anatomy segmentor including a naturally built-in human-in-the-loop mechanism.
We demonstrate that our one-shot learning method significantly outperforms non-learning-based methods and performs competitively to the state-of-the-art fully supervised deep learning approaches.
arXiv Detail & Related papers (2020-07-06T20:27:38Z) - Collaborative Boundary-aware Context Encoding Networks for Error Map
Prediction [65.44752447868626]
We propose collaborative boundaryaware context encoding networks called AEP-Net for error prediction task.
Specifically, we propose a collaborative feature transformation branch for better feature fusion between images and masks, and precise localization of error regions.
The AEP-Net achieves an average DSC of 0.8358, 0.8164 for error prediction task, and shows a high Pearson correlation coefficient of 0.9873.
arXiv Detail & Related papers (2020-06-25T12:42:01Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.