论文信息 - TransResUNet: Improving U-Net Architecture for Robust Lungs Segmentation in Chest X-rays

TransResUNet: Improving U-Net Architecture for Robust Lungs Segmentation in Chest X-rays

Medical image segmentation is regarded as an important component in a computer-aided diagnosis (CAD) system as it directly affects overall system performance. In this paper, we propose a new fully convolutional encoder-decoder model for lung segmentation named TransResUNet. We developed this architecture improving the state-of-the-art U-Net model. As part of the improvement to the classical U-Net, we introduced a pre-trained encoder, a special skip connection and a post-processing module in the proposed architecture. As a result, the proposed model outperformed the baseline U-Net model by 97.6% vs 94.9% considering the dice coefficient and 98.5% vs. 96.8% in terms of accuracy. The proposed TransResUNet achieved this feat with about 24% fewer parameters than the baseline U-Net. The implementation (based on Keras) of our proposed model is publicly available at https://sakibreza.github.io/TransResUNet with additional resources.

Sakib Reza | Ohida Binte Amin | M.M.A. Hashem

[1] Shanq-Jang Ruan,et al. Low order adaptive region growing for lung segmentation on plain chest radiographs , 2018, Neurocomputing.

[2] Mohammad Sohel Rahman,et al. MultiResUNet : Rethinking the U-Net Architecture for Multimodal Biomedical Image Segmentation , 2019, Neural Networks.

[3] Taimur Hassan,et al. Fully Convolutional Neural Network for Lungs Segmentation from Chest X-Rays , 2018, ICIAR.

[4] Li Fei-Fei,et al. ImageNet: A large-scale hierarchical image database , 2009, CVPR.

[5] Andrew Zisserman,et al. Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[6] Alexey Shvets,et al. TernausNetV2: Fully Convolutional Network for Instance Segmentation , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[7] Thomas Brox,et al. U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[8] Clement J. McDonald,et al. Lung Segmentation in Chest Radiographs Using Anatomical Atlases With Nonrigid Registration , 2014, IEEE Transactions on Medical Imaging.