ENet: An Edge Enhanced Network for Accurate Liver and Tumor Segmentation on CT Scans
arXiv:2007.09791
Abstract
Developing an effective liver and liver tumor segmentation model from CT scans is very important for the success of liver cancer diagnosis, surgical planning and cancer treatment. In this work, we propose a two-stage framework for 2D liver and tumor segmentation. The first stage is a coarse liver segmentation network, while the second stage is an edge enhanced network (ENet) for more accurate liver and tumor segmentation. ENet explicitly models complementary objects (liver and tumor) and their edge information within the network to preserve the organ and lesion boundaries. We introduce an edge prediction module in ENet and design an edge distance map between liver and tumor boundaries, which is used as an extra supervision signal to train the edge enhanced network. We also propose a deep cross feature fusion module to refine multi-scale features from both objects and their edges. ENet is more easily and efficiently trained with a small labeled dataset, and it can be trained/tested on the original 2D CT slices (resolve resampling error issue in 3D models). The proposed framework has shown superior performance on both liver and liver tumor segmentation compared to several state-of-the-art 2D, 3D and 2D/3D hybrid frameworks.
References in corpus (13)
- Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation
- Res2Net: A New Multi-scale Backbone Architecture
- The Liver Tumor Segmentation Benchmark (LiTS)
- H-DenseUNet: Hybrid Densely Connected UNet for Liver and Tumor Segmentation from CT Volumes
- XLSor: A Robust and Accurate Lung Segmentor on Chest X-Rays Using Criss-Cross Attention and Customized Radiorealistic Abnormalities Generation
- Neural Network-Based Automatic Liver Tumor Segmentation With Random Forest-Based Candidate Filtering
- CT-Realistic Lung Nodule Simulation from 3D Conditional Generative Adversarial Networks for Robust Lung Segmentation
- Anatomy-Aware Siamese Network: Exploiting Semantic Asymmetry for Accurate Pelvic Fracture Detection in X-ray Images
- Semi-Automatic RECIST Labeling on CT Scans with Cascaded Convolutional Neural Networks
- Accurate Weakly-Supervised Deep Lesion Segmentation using Large-Scale Clinical Annotations: Slice-Propagated 3D Mask Generation from 2D RECIST
- Cross-Domain Medical Image Translation by Shared Latent Gaussian Mixture Model
- CT Image Enhancement Using Stacked Generative Adversarial Networks and Transfer Learning for Lesion Segmentation Improvement
- One Click Lesion RECIST Measurement and Segmentation on CT Scans