Pre-trained deep convolutional neural networks for the segmentation of malignant pleural mesothelioma tumor on CT scans

Eyjolfur Gudmundsson; Christopher M. Straus; Samuel G. Armato III

doi:10.1117/12.2512974

13 March 2019 Pre-trained deep convolutional neural networks for the segmentation of malignant pleural mesothelioma tumor on CT scans

Eyjolfur Gudmundsson, Christopher M. Straus, Samuel G. Armato III

Author Affiliations +

Proceedings Volume 10950, Medical Imaging 2019: Computer-Aided Diagnosis; 109503J (2019) https://doi.org/10.1117/12.2512974
Event: SPIE Medical Imaging, 2019, San Diego, California, United States

Abstract

Pre-trained deep convolutional neural networks (CNNs) have shown promise in the training of deep CNNs for medical imaging applications. The purpose of this study was to investigate the use of partially pre-trained deep CNNs for the segmentation of malignant pleural mesothelioma tumor on CT scans. Four network configurations were investigated: (1) VGG16/U-Net network with pre-trained layers fixed during training, (2) VGG16/U-Net network with pre-trained layers fine-tuned during training, (3) VGG16/U-Net network with all except the first two pre-trained layers fine-tuned during training, and (4) a standard U-Net architecture trained from scratch. Deep CNNs were trained separately for tumor segmentation in left and right hemithoraces using 4259 and 6441 contoured axial CT sections, respectively. A test set of 61 CT sections from 16 patients excluded from training was used to evaluate segmentation performance; the Dice similarity coefficient (DSC) was calculated between computer-generated and reference segmentations provided by two radiologists and one radiology resident. Median DSC on the test set was 0.739 (range 0.328–0.920), 0.772 (range 0.342–0.949), 0.777 (range 0.216–0.946), and 0.758 (range 0.099–0.943) across all observers for network configurations (1), (2), (3) and (4) above, respectively. The median DSC achieved with configuration (3) when compared with the standard U-Net trained from scratch was found to be significantly higher for two out of three observers. A fine-tuned VGG16/U-Net deep CNN showed significantly higher overlap with two out of three observers when compared with a standard U-Net trained from scratch for the segmentation of malignant pleural mesothelioma tumor.

Citation Download Citation

Eyjolfur Gudmundsson, Christopher M. Straus, and Samuel G. Armato III "Pre-trained deep convolutional neural networks for the segmentation of malignant pleural mesothelioma tumor on CT scans", Proc. SPIE 10950, Medical Imaging 2019: Computer-Aided Diagnosis, 109503J (13 March 2019); https://doi.org/10.1117/12.2512974

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

Includes PDF, HTML & Video, when available