CE-Net: Context Encoder Network for 2D Medical Image Segmentation

Abstract

Medical image segmentation is an important step in medical image analysis. With the rapid development of convolutional neural network in image processing, deep learning has been used for medical image segmentation, such as optic disc segmentation, blood vessel detection, lung segmentation, cell segmentation, etc. Previously, U-net based approaches have been proposed. However, the consecutive pooling and strided convolutional operations lead to the loss of some spatial information. In this paper, we propose a context encoder network (referred to as CE-Net) to capture more high-level information and preserve spatial information for 2D medical image segmentation. CE-Net mainly contains three major components: a feature encoder module, a context extractor and a feature decoder module. We use pretrained ResNet block as the fixed feature extractor. The context extractor module is formed by a newly proposed dense atrous convolution (DAC) block and residual multi-kernel pooling (RMP) block. We applied the proposed CE-Net to different 2D medical image segmentation tasks. Comprehensive results show that the proposed method outperforms the original U-Net method and other state-of-the-art methods for optic disc segmentation, vessel detection, lung segmentation, cell contour segmentation and retinal optical coherence tomography layer segmentation.

accepted by IEEE transcations on medical imaging, (TMI)

References in corpus (5)

• Efficient Multi-Scale 3D CNN with Fully Connected CRF for Accurate Brain Lesion Segmentation
• Joint Optic Disc and Cup Segmentation Based on Multi-label Deep Network and Polar Transformation
• Optic Disc and Cup Segmentation Methods for Glaucoma Detection with Modification of U-Net Convolutional Neural Network
• InfiNet: Fully Convolutional Networks for Infant Brain MRI Segmentation
• Sparse Range-constrained Learning and Its Application for Medical Image Grading

Cited by in corpus (41)

• U-Net and its variants for medical image segmentation: theory and applications
• Medical Image Segmentation Using Deep Learning: A Survey
• Multi-stage Attention ResU-Net for Semantic Segmentation of Fine-Resolution Remote Sensing Images
• Boundary-aware Context Neural Network for Medical Image Segmentation
• Modality specific U-Net variants for biomedical image segmentation: A survey
• Evaluation of Retinal Image Quality Assessment Networks in Different Color-spaces
• RTNet: Relation Transformer Network for Diabetic Retinopathy Multi-lesion Segmentation
• DoFE: Domain-oriented Feature Embedding for Generalizable Fundus Image Segmentation on Unseen Datasets
• Boundary-aware Transformers for Skin Lesion Segmentation
• MACU-Net for Semantic Segmentation of Fine-Resolution Remotely Sensed Images
• MFSNet: A Multi Focus Segmentation Network for Skin Lesion Segmentation
• AADG: Automatic Augmentation for Domain Generalization on Retinal Image Segmentation
• Bi-directional Dermoscopic Feature Learning and Multi-scale Consistent Decision Fusion for Skin Lesion Segmentation
• Deep Retinal Image Segmentation with Regularization Under Geometric Priors
• RetiFluidNet: A Self-Adaptive and Multi-Attention Deep Convolutional Network for Retinal OCT Fluid Segmentation
• Factorizer: A Scalable Interpretable Approach to Context Modeling for Medical Image Segmentation
• Automatic Segmentation and Visualization of Choroid in OCT with Knowledge Infused Deep Learning
• Learning With Context Feedback Loop for Robust Medical Image Segmentation
• DoubleU-NetPlus: A Novel Attention and Context Guided Dual U-Net with Multi-Scale Residual Feature Fusion Network for Semantic Segmentation of Medical Images
• Deep Co-supervision and Attention Fusion Strategy for Automatic COVID-19 Lung Infection Segmentation on CT Images
• Robust Retinal Vessel Segmentation from a Data Augmentation Perspective
• Local Label Point Correction for Edge Detection of Overlapping Cervical Cells
• Learning Fuzzy Clustering for SPECT/CT Segmentation via Convolutional Neural Networks
• DSCA: A Digital Subtraction Angiography Sequence Dataset and Spatio-Temporal Model for Cerebral Artery Segmentation
• Interpretable machine learning for time-to-event prediction in medicine and healthcare
• HST-MRF: Heterogeneous Swin Transformer with Multi-Receptive Field for Medical Image Segmentation
• Multiclass Segmentation using Teeth Attention Modules for Dental X-ray Images
• Towards Counterfactual and Contrastive Explainability and Transparency of DCNN Image Classifiers
• OCTAve: 2D en face Optical Coherence Tomography Angiography Vessel Segmentation in Weakly-Supervised Learning with Locality Augmentation
• FourierNet: Shape-Preserving Network for Henle's Fiber Layer Segmentation in Optical Coherence Tomography Images
• Multi-Slice Dense-Sparse Learning for Efficient Liver and Tumor Segmentation
• Automated skin lesion segmentation using multi-scale feature extraction scheme and dual-attention mechanism
• An Analysis of the Influence of Transfer Learning When Measuring the Tortuosity of Blood Vessels
• FAS-UNet: A Novel FAS-driven Unet to Learn Variational Image Segmentation
• RA V-Net: Deep learning network for automated liver segmentation
• 3D Distance-color-coded Assessment of PCI Stent Apposition via Deep-learning-based Three-dimensional Multi-object Segmentation
• LGPS: A Lightweight GAN-Based Approach for Polyp Segmentation in Colonoscopy Images
• Universal Vessel Segmentation for Multi-Modality Retinal Images
• OMSN and FAROS: OCTA Microstructure Segmentation Network and Fully Annotated Retinal OCTA Segmentation Dataset
• Innovative Quantitative Analysis for Disease Progression Assessment in Familial Cerebral Cavernous Malformations
• KeyCLD: Learning Constrained Lagrangian Dynamics in Keypoint Coordinates from Images