Abstract:PurposeDiagnosis 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.MethodsThe 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.ResultsThe 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.ConclusionThe 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.

Joint Nuclear-Norm Nonlinear Manifold Learning And Robust Classification By Linear Embedding

Learning Exemplar-Represented Manifolds in Latent Space for Classification.

A Novel Image Classification Method Based on Manifold Learning and Gaussian Mixture Model

Markov-Lipschitz Deep Learning

Learning nonlinear manifolds based on mixtures of localized linear manifolds under a self-organizing framework

Manifold constrained joint sparse learning via non-convex regularization

Joint Learning of Latent Similarity and Local Embedding for Multi-View Clustering

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

Robust Grassmann manifold convex hull collaborative representation learning and its kernel extension for image set analysis

Neural Manifold Clustering and Embedding

Robust Manifold Matrix Factorization for Joint Clustering and Feature Extraction.

Joint Metric Learning-Based Class-Specific Representation for Image Set Classification

A joint manifold leaning-based framework for heterogeneous upstream data fusion

Joint learning with nonnegative matrix factorization and multinomial logistic regression

Manifold Mapping Machine

Nearest Linear Manifold Classification

Joint & Progressive Learning from High-Dimensional Data for Multi-Label Classification

Multi-Manifold Deep Metric Learning For Image Set Classification

Nearest Manifold Approach for Face Recognition

Patterns classification of nonlinear multi-dimensional time series based on manifold learning

Image retrieval using nonlinear manifold embedding