Context-Aware U-Net for Biomedical Image Segmentation

Jiaxu Leng,Ying Liu,Tianlin Zhang,Pei Quan,Zhenyu Cui
DOI: https://doi.org/10.1109/bibm.2018.8621512
2018-01-01
Abstract:Biomedical image segmentation is challenging for the lack of training data. U-Net has achieved good performance with small scale datasets through skip connections to merge features of low-level layers and high-level layers. However, it does not fully exploit the valuable contexts, which limits its segmentation performance. To enhance traditional U-Net not fully considering contextual information, we develop a context-aware U-Net, which aims to capture valuable contexts and improves the segmentation performance. Specially, we design a context transfer module that learns the rich contexts and produces context feature maps corresponding to input convolution feature maps. The proposed context transfer module is lightweight and it is easy to embed in any existing networks for image segmentation. In this paper, we present a context-aware U-Net with our context transfer module and conduct experiments to evaluate its performance on the ISBI challenge for segmentation of neuronal structures in electron microscopic stacks. Experimental results show that our proposed context-aware U-Net achieves state-of-the-art performance on the ISBI challenge.
What problem does this paper attempt to address?