Digital Health Innovation and Informatics
PD 14 - Digital Health Information & Informatics - Poster Discussion
1117 - A Multi-Task CNN Model for Autosegmentation of Prostate Patients
Tuesday, October 23
4:45 PM - 4:51 PM
Location: Room 217 A/B
Vasant Kearney, PhD
University of California, San Francisco
UCSF: physics resident: Employee, Physics Resident: Employee
Nimble Therapy, LLC: Partnership
A Multi-Task CNN Model for Autosegmentation of Prostate Patients
V. Kearney1, J. Chan1, M. Descovich1, S. S. Yom2, and T. D. Solberg1; 1University of California San Francisco, Department of Radiation Oncology, San Francisco, CA, 2University of California, San Francisco, San Francisco, CA
Purpose/Objective(s): Transfer learning with deep convolutional neural networks (CNNs) has been successful in increasing the prediction accuracy of 2D classification tasks, by utilizing open source training sets such as ImageNet. Currently, there are no large public 3D segmentation data sets, so CNN models have to rely on limited training data, which can lead to poor generalization. One alternative is to create a core shared model trained on a variety of separate but similar tasks, known as Multi-Task Learning (MTL). MTL has the ability to train on a richer and more diverse learning environment, which allows for better prediction accuracy and improved training efficiency. The present study suggests a MTL-CNN model as a superior alternative to traditional CNN models for autosegmentation of prostate patients.
Materials/Methods: This study evaluates the Dice coefficients between the physician-contoured ground truths (GT) and the 2D-CNN, 3D-CNN, and multi-task learning CNN (MTL-CNN) predictions. The planning CT and structures of 100 prostate patients were used in this study. Training, validation, and testing were done following Kaggle competition rules, where 80 patients were used for training, 10 patients were used for internal validation, and 10 patients were used to report final results. The 2D-CNN utilized an adaptation of U-Net, the 3D-CNN was based on a commercial 3D fully convolutional neural network architecture, and the MTL-CNN used the same core architecture as 3D-CNN but with two additional convolutional blocks per structure. The MTL-CNN was trained on all four structures simultaneously. The training time was ~11 hours, ~35 hours, and ~10 hours for the 2D-CNN, 3D-CNN, and MTL-CNN respectively using a 4x data augmentation scheme for all 4 structures. All algorithms were implemented in Pytorch using two Nvidia 1080 Ti GPUs. Due to proximity and GPU memory limitations, this study was restricted to the urethra, CTV, penile bulb, and bladder. Comparator p-values were reported using Mann-Whitney U test.
Results: The 2D-CNN had Dice scores of 81.6 ± 2.2, 89.6 ± 1.9, 61.6 ± 1.74, and 41 ± 17.8 for the CTV, bladder, penile bulb, and urethra respectively. The 3D-CNN had Dice scores of 84.4 ± 2.15, 90.4 ± 2.1, 68.8 ± 1.3, and 60.6 ± 6.5 for the CTV, bladder, penile bulb, and urethra respectively. The MTL-CNN had Dice scores of 87.6 ± 1.9, 91.8 ± 1.94, 72.2 ± 1.72, and 68 ± 4.3. P-values of MTL-CNN versus 3D-CNN were 0.029, 0.175, 0.007, and 0.036 for the CTV, bladder, penile bulb, and urethra respectively.
Conclusion: This study has demonstrated that multi-task learning is capable of achieving superior training efficiency and prediction accuracy than traditional deep learning methods for CNN-based autosegmentation. Largest mean differences were seen for the most clinically challenging structures.
Author Disclosure: V. Kearney: Partnership; Nimble Therapy, LLC. J. Chan: None. M. Descovich: Research Grant; Accuray Inc., Sunnyvale CA. Honoraria; Accuray Inc., Sunnyvale CA. Travel Expenses; Accuray Inc., Sunnyvale CA. Physics Residency Program Director; University of California San Francisco. S.S. Yom: Research Grant; Genentech, Merck, Bristol-Myers Squibb. royalty; UpToDate, Springer. Honoraria; ASTRO. Chair; Am Radium Society-Am College of Radiology. Treasurer; American Radium Society. T.D. Solberg: Speaker's Bureau; Brainlab. Partnership; Global Radiosurgery, LLC. Deputy Editor-in-Chief; Journal of Applied Clinical Physics.