Lung nodule malignancy prediction using multi-task convolutional neural network

Xiuli Li; Yueying Kao; Wei Shen; Xiang Li; Guotong Xie

doi:10.1117/12.2253836

3 March 2017 Lung nodule malignancy prediction using multi-task convolutional neural network

Xiuli Li, Yueying Kao, Wei Shen, Xiang Li, Guotong Xie

Author Affiliations +

Proceedings Volume 10134, Medical Imaging 2017: Computer-Aided Diagnosis; 1013424 (2017) https://doi.org/10.1117/12.2253836
Event: SPIE Medical Imaging, 2017, Orlando, Florida, United States

Abstract

In this paper, we investigated the problem of diagnostic lung nodule malignancy prediction using thoracic Computed Tomography (CT) screening. Unlike most existing studies classify the nodules into two types benign and malignancy, we interpreted the nodule malignancy prediction as a regression problem to predict continuous malignancy level. We proposed a joint multi-task learning algorithm using Convolutional Neural Network (CNN) to capture nodule heterogeneity by extracting discriminative features from alternatingly stacked layers. We trained a CNN regression model to predict the nodule malignancy, and designed a multi-task learning mechanism to simultaneously share knowledge among 9 different nodule characteristics (Subtlety, Calcification, Sphericity, Margin, Lobulation, Spiculation, Texture, Diameter and Malignancy), and improved the final prediction result. Each CNN would generate characteristic-specific feature representations, and then we applied multi-task learning on the features to predict the corresponding likelihood for that characteristic. We evaluated the proposed method on 2620 nodules CT scans from LIDC-IDRI dataset with the 5-fold cross validation strategy. The multitask CNN regression result for regression RMSE and mapped classification ACC were 0.830 and 83.03%, while the results for single task regression RMSE 0.894 and mapped classification ACC 74.9%. Experiments show that the proposed method could predict the lung nodule malignancy likelihood effectively and outperforms the state-of-the-art methods. The learning framework could easily be applied in other anomaly likelihood prediction problem, such as skin cancer and breast cancer. It demonstrated the possibility of our method facilitating the radiologists for nodule staging assessment and individual therapeutic planning.

Conference Presentation

Citation Download Citation

Xiuli Li, Yueying Kao, Wei Shen, Xiang Li, and Guotong Xie "Lung nodule malignancy prediction using multi-task convolutional neural network", Proc. SPIE 10134, Medical Imaging 2017: Computer-Aided Diagnosis, 1013424 (3 March 2017); https://doi.org/10.1117/12.2253836

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

;

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE