A manifold learning regularization approach to enhance 3D CT image-based lung nodule classification

• Published in: International journal for computer assisted radiology and surgery 2020-02, Vol.15 (2), p.287-295
• Main Authors: Ren, Ying, Tsai, Min-Yu, Chen, Liyuan, Wang, Jing, Li, Shulong, Liu, Yufei, Jia, Xun, Shen, Chenyang
• Format: Article
• Language: English
• Subjects: Artificial neural networks; Classification; Computed tomography; Computer Imaging; Computer Science; Data structures; Deep learning; Diagnosis; Health Informatics; Image classification; Image enhancement; Image reconstruction; Imaging; Machine learning; Manifolds (mathematics); Medical imaging; Medicine; Medicine & Public Health; Nodules; Original Article; Pattern Recognition and Graphics; Radiology; Regularization; Representations; Surgery; Training; Vision
• ISSN: 1861-6410; 1861-6429
• DOI: 10.1007/s11548-019-02097-8

Purpose Diagnosis of lung cancer requires radiologists to review every lung nodule in CT images. Such a process can be very time-consuming, and the accuracy is affected by many factors, such as experience of radiologists and available diagnosis time. To address this problem, we proposed to develop a deep learning-based system to automatically classify benign and malignant lung nodules. Methods The proposed method automatically determines benignity or malignancy given the 3D CT image patch of a lung nodule to assist diagnosis process. Motivated by the fact that real structure among data is often embedded on a low-dimensional manifold, we developed a novel manifold regularized classification deep neural network (MRC-DNN) to perform classification directly based on the manifold representation of lung nodule images. The concise manifold representation revealing important data structure is expected to benefit the classification, while the manifold regularization enforces strong, but natural constraints on network training, preventing over-fitting. Results The proposed method achieves accurate manifold learning with reconstruction error of ~ 30 HU on real lung nodule CT image data. In addition, the classification accuracy on testing data is 0.90 with sensitivity of 0.81 and specificity of 0.95, which outperforms state-of-the-art deep learning methods. Conclusion The proposed MRC-DNN facilitates an accurate manifold learning approach for lung nodule classification based on 3D CT images. More importantly, MRC-DNN suggests a new and effective idea of enforcing regularization for network training, possessing the potential impact to a board range of applications.