Boundary Refinement Using Adversarial Learning: Application to Lung Tumor Segmentation in Thoracic MRI

Pei Zhou,Houjin Chen,Yahui Peng,Yanfeng Li
DOI: https://doi.org/10.2139/ssrn.3990566
2021-01-01
SSRN Electronic Journal
Abstract:Lung tumor segmentation in thoracic MRI images is challenging due to the significant variation in shape and location, as well as the discontinuity problem (including blurred boundaries and the discontinuity within regions). We propose an effective segmentation scheme based on adversarial learning for lung tumor segmentation. An improved Unet servers as the generator for intensity-based coarse segmentation, and then co-trained with the discriminator in the proposed adversarial scheme for refinement. Specifically, we fill back the tumor segmentation into the modified input without tumors as the pseudo input to be discriminated against the original input. Compared with traditional approaches, the proposed scheme emphasizes segmentation errors and preserves the global structures simultaneously for effective and stable training. Besides, we introduce an adaptive L1 loss to learn the intensity information for further boundary refinement. We evaluated our method on 76 T2-weighted MRI scans. Experimental results achieve improved performance compared with other traditional adversarial learning-based segmentation methods.
What problem does this paper attempt to address?