LesionMix data enhancement and entropy minimization for semi-supervised lesion segmentation of lung cancer

Abstract

Determining the location and contour of the lesion is a crucial prerequisite for medical diagnosis, subsequent personalized treatment plan and prognostic prediction of lung cancer. Semi-supervised learning and data augmentation methods facilitate deep learning to be used in many fields of medical imaging. In this paper, we introduce a novel data enhancement technique called LesionMix. This method involves extracting and reusing lesions from a limited amount of labeled CT data, thereby enhancing the efficiency of utilizing those labeled data. Meanwhile, we propose a two-stage semi-supervised training strategy called Entropy Minimization LesionMix (EMLM). In the first stage, features containing lesion contour information are rapidly learned through LesionMix data augmentation. Entropy minimization strategy optimizes the model parameters to alleviate overfitting as much as possible in the second stage and improves prediction confidence. Our proposed method is validated on the public dataset LIDC-IDRI and the in-house dataset GDPHLUAD. Extensive experiments demonstrate that our method achieves promising performance and outperforms seven state-of-the-art semi-supervised models; Moreover, ablation experiments validate the effectiveness of various aspects of our approach.