Conquering Data Variations in Resolution: A Slice-Aware Multi-Branch Decoder Network

Abstract

Fully convolutional neural networks have made promising progress in joint liver and liver tumor segmentation. Instead of following the debates over 2D versus 3D networks (for example, pursuing the balance between large-scale 2D pretraining and 3D context), in this paper, we novelly identify the wide variation in the ratio between intra- and inter-slice resolutions as a crucial obstacle to the performance. To tackle the mismatch between the intra- and inter-slice information, we propose a slice-aware 2.5D network that emphasizes extracting discriminative features utilizing not only in-plane semantics but also out-of-plane coherence for each separate slice. Specifically, we present a slice-wise multi-input multi-output architecture to instantiate such a design paradigm, which contains a Multi-Branch Decoder (MD) with a Slice-centric Attention Block (SAB) for learning slice-specific features and a Densely Connected Dice (DCD) loss to regularize the inter-slice predictions to be coherent and continuous. Based on the aforementioned innovations, we achieve state-of-the-art results on the MICCAI 2017 Liver Tumor Segmentation (LiTS) dataset. Besides, we also test our model on the ISBI 2019 Segmentation of THoracic Organs at Risk (SegTHOR) dataset, and the result proves the robustness and generalizability of the proposed method in other segmentation tasks.

arXiv commentPublished by IEEE TMI

References in corpus (13)

• Rethinking Atrous Convolution for Semantic Image Segmentation
• Attention U-Net: Learning Where to Look for the Pancreas
• Fully Convolutional Networks for Semantic Segmentation
• The Liver Tumor Segmentation Benchmark (LiTS)
• Med3D: Transfer Learning for 3D Medical Image Analysis
• MS-Net: Multi-Site Network for Improving Prostate Segmentation with Heterogeneous MRI Data
• nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentation
• Reinventing 2D Convolutions for 3D Images
• H-DenseUNet: Hybrid Densely Connected UNet for Liver and Tumor Segmentation from CT Volumes
• Hierarchical Convolutional-Deconvolutional Neural Networks for Automatic Liver and Tumor Segmentation
• Neural Network-Based Automatic Liver Tumor Segmentation With Random Forest-Based Candidate Filtering
• Thickened 2D Networks for Efficient 3D Medical Image Segmentation
• 3D Anisotropic Hybrid Network: Transferring Convolutional Features from 2D Images to 3D Anisotropic Volumes

Cited by in corpus (3)

• The Liver Tumor Segmentation Benchmark (LiTS)
• Simultaneous Alignment and Surface Regression Using Hybrid 2D-3D Networks for 3D Coherent Layer Segmentation of Retina OCT Images
• Noisy Labels are Treasure: Mean-Teacher-Assisted Confident Learning for Hepatic Vessel Segmentation