Automated pulmonary nodule detection using 3D deep convolutional neural networks

Hao Tang,Daniel R. Kim,Xiaohui Xie
DOI: https://doi.org/10.48550/arXiv.1903.09876
2019-03-24
Abstract:Early detection of pulmonary nodules in computed tomography (CT) images is essential for successful outcomes among lung cancer patients. Much attention has been given to deep convolutional neural network (DCNN)-based approaches to this task, but models have relied at least partly on 2D or 2.5D components for inherently 3D data. In this paper, we introduce a novel DCNN approach, consisting of two stages, that is fully three-dimensional end-to-end and utilizes the state-of-the-art in object detection. First, nodule candidates are identified with a U-Net-inspired 3D Faster R-CNN trained using online hard negative mining. Second, false positive reduction is performed by 3D DCNN classifiers trained on difficult examples produced during candidate screening. Finally, we introduce a method to ensemble models from both stages via consensus to give the final predictions. By using this framework, we ranked first of 2887 teams in Season One of Alibaba's 2017 TianChi AI Competition for Healthcare.
Computer Vision and Pattern Recognition,Artificial Intelligence,Machine Learning,Neural and Evolutionary Computing
What problem does this paper attempt to address?
This paper aims to solve the problem of early detection of pulmonary nodules in computed tomography (CT) images, which is crucial for the successful treatment of lung cancer patients. Although deep convolutional neural networks (DCNN) have received extensive attention in this task, existing models often rely in part on 2D or 2.5D components to process essentially 3D data. This paper proposes a brand - new DCNN method, which is completely three - dimensional end - to - end and utilizes state - of - the - art object detection techniques to improve the accuracy and efficiency of pulmonary nodule detection. Specifically, the paper proposes a two - stage framework: 1. **Candidate Nodule Detection**: Use 3D Faster R - CNN inspired by U - Net to identify nodule candidate regions, and the online hard negative mining technique is adopted during the training process. 2. **False Positive Reduction**: Train the initially screened difficult samples through a 3D DCNN classifier to reduce false positive results. Finally, the paper also introduces a model integration method. By combining the model outputs of the two stages through a consensus mechanism, the final prediction results are generated. This method ranked first in the first season of Alibaba's Tianchi AI Medical Competition in 2017, demonstrating its effectiveness and superior performance in practical applications.