Diagnostic Classification Of Lung Nodules Using 3D Neural Networks
This work addresses early lung cancer diagnosis for medical imaging, but it is incremental as it builds on existing 3D CNN methods with hybrid improvements.
The paper tackled the problem of diagnostic classification between benign and malignant lung nodules in CT images using 3D neural networks, achieving state-of-the-art classification accuracy with the proposed 3D multi-output DenseNet on the LIDC-IDRI dataset.
Lung cancer is the leading cause of cancer-related death worldwide. Early diagnosis of pulmonary nodules in Computed Tomography (CT) chest scans provides an opportunity for designing effective treatment and making financial and care plans. In this paper, we consider the problem of diagnostic classification between benign and malignant lung nodules in CT images, which aims to learn a direct mapping from 3D images to class labels. To achieve this goal, four two-pathway Convolutional Neural Networks (CNN) are proposed, including a basic 3D CNN, a novel multi-output network, a 3D DenseNet, and an augmented 3D DenseNet with multi-outputs. These four networks are evaluated on the public LIDC-IDRI dataset and outperform most existing methods. In particular, the 3D multi-output DenseNet (MoDenseNet) achieves the state-of-the-art classification accuracy on the task of end-to-end lung nodule diagnosis. In addition, the networks pretrained on the LIDC-IDRI dataset can be further extended to handle smaller datasets using transfer learning. This is demonstrated on our dataset with encouraging prediction accuracy in lung nodule classification.