An End-to-end Approach to Semantic Segmentation with 3D CNN and Posterior-CRF in Medical Images
arXiv:1811.03549
Abstract
Fully-connected Conditional Random Field (CRF) is often used as post-processing to refine voxel classification results by encouraging spatial coherence. In this paper, we propose a new end-to-end training method called Posterior-CRF. In contrast with previous approaches which use the original image intensity in the CRF, our approach applies 3D, fully connected CRF to the posterior probabilities from a CNN and optimizes both CNN and CRF together. The experiments on white matter hyperintensities segmentation demonstrate that our method outperforms CNN, post-processing CRF and different end-to-end training CRF approaches.
arXiv commentAccepted in Medical Imaging meets NIPS Workshop, NIPS 2018
References in corpus (5)
- Efficient Multi-Scale 3D CNN with Fully Connected CRF for Accurate Brain Lesion Segmentation
- Efficient Inference in Fully Connected CRFs with Gaussian Edge Potentials
- DeepNAT: Deep Convolutional Neural Network for Segmenting Neuroanatomy
- Fully Connected Deep Structured Networks
- Conditional Random Fields as Recurrent Neural Networks for 3D Medical Imaging Segmentation