Anatomical and Diagnostic Bayesian Segmentation in Prostate MRI
$-$Should Different Clinical Objectives Mandate Different Loss Functions?
- URL: http://arxiv.org/abs/2110.12889v1
- Date: Mon, 25 Oct 2021 12:44:52 GMT
- Title: Anatomical and Diagnostic Bayesian Segmentation in Prostate MRI
$-$Should Different Clinical Objectives Mandate Different Loss Functions?
- Authors: Anindo Saha, Joeran Bosma, Jasper Linmans, Matin Hosseinzadeh, Henkjan
Huisman
- Abstract summary: Probable voxel-level classification of anatomy and malignancy in prostate MRI requires different loss functions for optimal performance.
We investigate distribution, region and boundary-based loss functions for both tasks across 200 patient exams from the publicly-available ProstateX dataset.
- Score: 0.3149883354098941
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: We hypothesize that probabilistic voxel-level classification of anatomy and
malignancy in prostate MRI, although typically posed as near-identical
segmentation tasks via U-Nets, require different loss functions for optimal
performance due to inherent differences in their clinical objectives. We
investigate distribution, region and boundary-based loss functions for both
tasks across 200 patient exams from the publicly-available ProstateX dataset.
For evaluation, we conduct a thorough comparative analysis of model predictions
and calibration, measured with respect to multi-class volume segmentation of
the prostate anatomy (whole-gland, transitional zone, peripheral zone), as well
as, patient-level diagnosis and lesion-level detection of clinically
significant prostate cancer. Notably, we find that distribution-based loss
functions (in particular, focal loss) are well-suited for diagnostic or
panoptic segmentation tasks such as lesion detection, primarily due to their
implicit property of inducing better calibration. Meanwhile, (with the
exception of focal loss) both distribution and region/boundary-based loss
functions perform equally well for anatomical or semantic segmentation tasks,
such as quantification of organ shape, size and boundaries.
Related papers
- Comprehensive framework for evaluation of deep neural networks in detection and quantification of lymphoma from PET/CT images: clinical insights, pitfalls, and observer agreement analyses [0.9958347059366389]
This study addresses critical gaps in automated lymphoma segmentation from PET/CT images.
Deep learning has been applied for lymphoma lesion segmentation, but few studies incorporate out-of-distribution testing.
We show that networks perform better on large, intense lesions with higher metabolic activity.
arXiv Detail & Related papers (2023-11-16T06:58:46Z) - 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) - Segmentation of glioblastomas in early post-operative multi-modal MRI
with deep neural networks [33.51490233427579]
Two state-of-the-art neural network architectures for pre-operative segmentation were trained for the task.
The best performance achieved was a 61% Dice score, and the best classification performance was about 80% balanced accuracy.
The predicted segmentations can be used to accurately classify the patients into those with residual tumor, and those with gross total resection.
arXiv Detail & Related papers (2023-04-18T10:14:45Z) - Towards Reliable Medical Image Segmentation by utilizing Evidential Calibrated Uncertainty [52.03490691733464]
We introduce DEviS, an easily implementable foundational model that seamlessly integrates into various medical image segmentation networks.
By leveraging subjective logic theory, we explicitly model probability and uncertainty for the problem of medical image segmentation.
DeviS incorporates an uncertainty-aware filtering module, which utilizes the metric of uncertainty-calibrated error to filter reliable data.
arXiv Detail & Related papers (2023-01-01T05:02:46Z) - Reliable Joint Segmentation of Retinal Edema Lesions in OCT Images [55.83984261827332]
In this paper, we propose a novel reliable multi-scale wavelet-enhanced transformer network.
We develop a novel segmentation backbone that integrates a wavelet-enhanced feature extractor network and a multi-scale transformer module.
Our proposed method achieves better segmentation accuracy with a high degree of reliability as compared to other state-of-the-art segmentation approaches.
arXiv Detail & Related papers (2022-12-01T07:32:56Z) - A Deep Learning Approach to Predicting Collateral Flow in Stroke
Patients Using Radiomic Features from Perfusion Images [58.17507437526425]
Collateral circulation results from specialized anastomotic channels which provide oxygenated blood to regions with compromised blood flow.
The actual grading is mostly done through manual inspection of the acquired images.
We present a deep learning approach to predicting collateral flow grading in stroke patients based on radiomic features extracted from MR perfusion data.
arXiv Detail & Related papers (2021-10-24T18:58:40Z) - Prostate Gland Segmentation in Histology Images via Residual and
Multi-Resolution U-Net [1.244681179922733]
This work is a prostate gland segmentation based on U-Net convolutional neural network architectures modified with residual and multi-resolution blocks, trained using data augmentation techniques.
The residual configuration outperforms in the test subset the previous state-of-the-art approaches in an image-level comparison, reaching an average Dice Index of 0.77.
arXiv Detail & Related papers (2021-05-21T20:11:36Z) - 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) - Enhancing Foreground Boundaries for Medical Image Segmentation [13.954358685870886]
We propose a boundary enhancement loss to enforce additional constraints on optimizing machine learning models.
Our experimental results validate that our loss function are better than, or at least comparable to, other state-of-the-art loss functions in terms of segmentation accuracy.
arXiv Detail & Related papers (2020-05-29T00:50:08Z) - Detecting Scatteredly-Distributed, Small, andCritically Important
Objects in 3D OncologyImaging via Decision Stratification [23.075722503902714]
We focus on the detection and segmentation of oncology-significant (or suspicious cancer metastasized) lymph nodes.
We propose a divide-and-conquer decision stratification approach that divides OSLNs into tumor-proximal and tumor-distal categories.
We present a novel global-local network (GLNet) that combines high-level lesion characteristics with features learned from localized 3D image patches.
arXiv Detail & Related papers (2020-05-27T23:12:11Z) - VerSe: A Vertebrae Labelling and Segmentation Benchmark for
Multi-detector CT Images [121.31355003451152]
Large Scale Vertebrae Challenge (VerSe) was organised in conjunction with the International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI) in 2019 and 2020.
We present the the results of this evaluation and further investigate the performance-variation at vertebra-level, scan-level, and at different fields-of-view.
arXiv Detail & Related papers (2020-01-24T21:09:18Z)
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.