Lobenet: a global position reservation and fissure-aware convolutional neural network for pulmonary lobe segmentation

Mingjian Chen,Yun Gu,Yulei Qin,Hao Zheng,Jie Yang
2020-01-01
Abstract:Pulmonary lobe segmentation is vital for clinical analysis and preoperative planning, which is essential for Computer-Aided Diagnosis systems (CADs). The fissures, separating the lung parenchyma into five pulmonary lobes, are hard to detect by the classical method with combining anatomical information. The deep learning method based on convolutional neural networks (CNNs) has made great success in pulmonary lobe segmentation. As we know, the position of the pulmonary voxel is helpful for pulmonary lobe segmentation. However, 3D CNN can not take the whole scan, only smaller cubes cropped from the scan as input due to the limitation of GPU memory, which will make the position information lost. In this paper, we proposed a 3D residual CNN called LobeNet for pulmonary lobe segmentation with global position reservation and fissure-aware property. Our LobeNet makes up for the loss of position information by incorporating the 3D global position as high-level features making the network converge to a better local minimum. Moreover, we use a fissure-aware training strategy to lead the network to concentrate on the fissures, which forces the predicted border of pulmonary lobes more accurate. Meanwhile, the focal loss is leveraged for hard example mining. Our proposed network is validated on a public dataset and achieved the state-of-the-art performance with a 93.35% dice coefficient. The ablation study illustrates the global position and fissure-aware training strategy can produce a more accurate segmentation result.
What problem does this paper attempt to address?