Semi-supervised multi-task learning for lung cancer diagnosis
  Early detection of lung nodules is of great importance in lung cancer
screening. Existing research recognizes the critical role played by CAD systems
in early detection and diagnosis of lung nodules. However, many CAD systems,
which are used as cancer detection tools, produce a lot of false positives (FP)
and require a further FP reduction step. Furthermore, guidelines for early
diagnosis and treatment of lung cancer are consist of different shape and
volume measurements of abnormalities. Segmentation is at the heart of our
understanding of nodules morphology making it a major area of interest within
the field of computer aided diagnosis systems. This study set out to test the
hypothesis that joint learning of false positive (FP) nodule reduction and
nodule segmentation can improve the computer aided diagnosis (CAD) systems'
performance on both tasks. To support this hypothesis we propose a 3D deep
multi-task CNN to tackle these two problems jointly. We tested our system on
LUNA16 dataset and achieved an average dice similarity coefficient (DSC) of 91%
as segmentation accuracy and a score of nearly 92% for FP reduction. As a proof
of our hypothesis, we showed improvements of segmentation and FP reduction
tasks over two baselines. Our results support that joint training of these two
tasks through a multi-task learning approach improves system performance on
both. We also showed that a semi-supervised approach can be used to overcome
the limitation of lack of labeled data for the 3D segmentation task.
