Bayesian Generative Models for Knowledge Transfer in MRI Semantic Segmentation Problems
arXiv:2005.12639
Abstract
Automatic segmentation methods based on deep learning have recently demonstrated state-of-the-art performance, outperforming the ordinary methods. Nevertheless, these methods are inapplicable for small datasets, which are very common in medical problems. To this end, we propose a knowledge transfer method between diseases via the Generative Bayesian Prior network. Our approach is compared to a pre-train approach and random initialization and obtains the best results in terms of Dice Similarity Coefficient metric for the small subsets of the Brain Tumor Segmentation 2018 database (BRATS2018).
arXiv admin note: substantial text overlap with arXiv:1908.05480
References in corpus (13)
- A Survey on Deep Learning in Medical Image Analysis
- Auto-Encoding Variational Bayes
- Anatomically Constrained Neural Networks (ACNN): Application to Cardiac Image Enhancement and Segmentation
- A large annotated medical image dataset for the development and evaluation of segmentation algorithms
- Variational Dropout and the Local Reparameterization Trick
- DeepNAT: Deep Convolutional Neural Network for Segmenting Neuroanatomy
- Multi-source Transfer Learning with Convolutional Neural Networks for Lung Pattern Analysis
- nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentation
- Structured Bayesian Pruning via Log-Normal Multiplicative Noise
- Adversarial Networks for the Detection of Aggressive Prostate Cancer
- Domain and Geometry Agnostic CNNs for Left Atrium Segmentation in 3D Ultrasound
- Adversarial Reprogramming of Neural Networks
- On the Bernstein-Von Mises Theorem for High Dimensional Nonlinear Bayesian Inverse Problems