Deep Implicit Statistical Shape Models for 3D Medical Image Delineation
- URL: http://arxiv.org/abs/2104.02847v1
- Date: Wed, 7 Apr 2021 01:15:06 GMT
- Title: Deep Implicit Statistical Shape Models for 3D Medical Image Delineation
- Authors: Ashwin Raju, Shun Miao, Chi-Tung Cheng, Le Lu, Mei Han, Jing Xiao,
Chien-Hung Liao, Junzhou Huang and Adam P. Harrison
- Abstract summary: 3D delineation of anatomical structures is a cardinal goal in medical imaging analysis.
Prior to deep learning, statistical shape models that imposed anatomical constraints and produced high quality surfaces were a core technology.
We present deep implicit statistical shape models (DISSMs), a new approach to delineation that marries the representation power of CNNs with the robustness of SSMs.
- Score: 47.78425002879612
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: 3D delineation of anatomical structures is a cardinal goal in medical imaging
analysis. Prior to deep learning, statistical shape models that imposed
anatomical constraints and produced high quality surfaces were a core
technology. Prior to deep learning, statistical shape models that imposed
anatomical constraints and produced high quality surfaces were a core
technology. Today fully-convolutional networks (FCNs), while dominant, do not
offer these capabilities. We present deep implicit statistical shape models
(DISSMs), a new approach to delineation that marries the representation power
of convolutional neural networks (CNNs) with the robustness of SSMs. DISSMs use
a deep implicit surface representation to produce a compact and descriptive
shape latent space that permits statistical models of anatomical variance. To
reliably fit anatomically plausible shapes to an image, we introduce a novel
rigid and non-rigid pose estimation pipeline that is modelled as a Markov
decision process(MDP). We outline a training regime that includes inverted
episodic training and a deep realization of marginal space learning (MSL).
Intra-dataset experiments on the task of pathological liver segmentation
demonstrate that DISSMs can perform more robustly than three leading FCN
models, including nnU-Net: reducing the mean Hausdorff distance (HD) by
7.7-14.3mm and improving the worst case Dice-Sorensen coefficient (DSC) by
1.2-2.3%. More critically, cross-dataset experiments on a dataset directly
reflecting clinical deployment scenarios demonstrate that DISSMs improve the
mean DSC and HD by 3.5-5.9% and 12.3-24.5mm, respectively, and the worst-case
DSC by 5.4-7.3%. These improvements are over and above any benefits from
representing delineations with high-quality surface.
Related papers
- Ground-truth effects in learning-based fiber orientation distribution estimation in neonatal brains [8.545738983975639]
We train a state-of-the-art model based on the U-Net architecture on both MSMT-CSD and single-shell three-tissue constrained spherical deconvolution (SS3T-CSD)
Our results suggest that SS3T-CSD might be more suited for neonatal brains, given that the ratio between single and multiple fiber-estimated voxels with SS3T-CSD is more realistic compared to MSMT-CSD.
In an age domain-shift setting, SS3T-CSD maintains robust performance across age groups, indicating its potential for more accurate neonatal brain imaging.
arXiv Detail & Related papers (2024-09-02T12:05:03Z) - Probabilistic 3D Correspondence Prediction from Sparse Unsegmented Images [1.2179682412409507]
We propose SPI-CorrNet, a unified model that predicts 3D correspondences from sparse imaging data.
Experiments on the LGE MRI left atrium dataset and Abdomen CT-1K liver datasets demonstrate that our technique enhances the accuracy and robustness of sparse image-driven SSM.
arXiv Detail & Related papers (2024-07-02T03:56:20Z) - On the Localization of Ultrasound Image Slices within Point Distribution
Models [84.27083443424408]
Thyroid disorders are most commonly diagnosed using high-resolution Ultrasound (US)
Longitudinal tracking is a pivotal diagnostic protocol for monitoring changes in pathological thyroid morphology.
We present a framework for automated US image slice localization within a 3D shape representation.
arXiv Detail & Related papers (2023-09-01T10:10:46Z) - ADASSM: Adversarial Data Augmentation in Statistical Shape Models From
Images [0.8192907805418583]
This paper introduces a novel strategy for on-the-fly data augmentation for the Image-to-SSM framework by leveraging data-dependent noise generation or texture augmentation.
Our approach achieves improved accuracy by encouraging the model to focus on the underlying geometry rather than relying solely on pixel values.
arXiv Detail & Related papers (2023-07-06T20:21:12Z) - S3M: Scalable Statistical Shape Modeling through Unsupervised
Correspondences [91.48841778012782]
We propose an unsupervised method to simultaneously learn local and global shape structures across population anatomies.
Our pipeline significantly improves unsupervised correspondence estimation for SSMs compared to baseline methods.
Our method is robust enough to learn from noisy neural network predictions, potentially enabling scaling SSMs to larger patient populations.
arXiv Detail & Related papers (2023-04-15T09:39:52Z) - Feature robustness and sex differences in medical imaging: a case study
in MRI-based Alzheimer's disease detection [1.7616042687330637]
We compare two classification schemes on the ADNI MRI dataset.
We do not find a strong dependence of model performance for male and female test subjects on the sex composition of the training dataset.
arXiv Detail & Related papers (2022-04-04T17:37:54Z) - Mesh convolutional neural networks for wall shear stress estimation in
3D artery models [7.7393800633675465]
We propose to use mesh convolutional neural networks that directly operate on the same finite-element surface mesh as used in CFD.
We show that our flexible deep learning model can accurately predict 3D wall shear stress vectors on this surface mesh.
arXiv Detail & Related papers (2021-09-10T11:32:05Z) - Revisiting 3D Context Modeling with Supervised Pre-training for
Universal Lesion Detection in CT Slices [48.85784310158493]
We propose a Modified Pseudo-3D Feature Pyramid Network (MP3D FPN) to efficiently extract 3D context enhanced 2D features for universal lesion detection in CT slices.
With the novel pre-training method, the proposed MP3D FPN achieves state-of-the-art detection performance on the DeepLesion dataset.
The proposed 3D pre-trained weights can potentially be used to boost the performance of other 3D medical image analysis tasks.
arXiv Detail & Related papers (2020-12-16T07:11:16Z) - Probabilistic 3D surface reconstruction from sparse MRI information [58.14653650521129]
We present a novel probabilistic deep learning approach for concurrent 3D surface reconstruction from sparse 2D MR image data and aleatoric uncertainty prediction.
Our method is capable of reconstructing large surface meshes from three quasi-orthogonal MR imaging slices from limited training sets.
arXiv Detail & Related papers (2020-10-05T14:18:52Z) - Modelling the Distribution of 3D Brain MRI using a 2D Slice VAE [66.63629641650572]
We propose a method to model 3D MR brain volumes distribution by combining a 2D slice VAE with a Gaussian model that captures the relationships between slices.
We also introduce a novel evaluation method for generated volumes that quantifies how well their segmentations match those of true brain anatomy.
arXiv Detail & Related papers (2020-07-09T13:23:15Z)
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.