Combination of Active Transfer Learning and Natural Language Processing to Improve Liver Volumetry Using Surrogate
Brett Marinelli1, Martin Kang1, Michael Martini1
1Departments of Radiology (B.M., M.K., M.M., J.T.), Orthopedic Surgery (S.C.), and Neurological Surgery (A.B.C., E.K.O.), Mount Sinai Health System, 1468 Madison Ave, Annenberg Building, 8th Floor, New York, NY 10029; and Department of Medicine, California Pacific Medical Center, San Francisco, Calif (J.R.Z.).
Purpose:
To determine if weakly supervised learning with surrogate metrics and active transfer learning can hasten clinical deployment of deep learning models.
Materials And Methods:
By leveraging Liver Tumor Segmentation (LiTS) challenge 2017 public data (n = 131 studies), natural language processing of reports, and an active learning method, a model was trained to segment livers on 239 retrospectively collected portal venous phase abdominal CT studies obtained between January 1, 2014, and December 31, 2016. Absolute volume differences between predicted and originally reported liver volumes were used to guide active learning and assess accuracy. Overall survival based on liver volumes predicted by this model (n = 34 patients) versus radiology reports and Model for End-Stage Liver Disease with sodium (MELD-Na) scores was assessed. Differences in absolute liver volume were compared by using the paired Student t test, Bland-Altman analysis, and intraclass correlation; survival analysis was performed with the Kaplan-Meier method and a Mantel-Cox test.
Results:
Data from patients with poor liver volume prediction (n = 10) with a model trained only with publicly available data were incorporated into an active learning method that trained a new model (LiTS data plus over- and underestimated active learning cases [LiTS-OU]) that performed significantly better on a held-out institutional test set (absolute volume difference of 231 vs 176 mL, P = .0005). In overall survival analysis, predicted liver volumes using the best active learning-trained model (LiTS-OU) were at least comparable with liver volumes extracted from radiology reports and MELD-Na scores in predicting survival.
Conclusion:
Active transfer learning using surrogate metrics facilitated deployment of deep learning models for clinically meaningful liver segmentation at a major liver transplant center.© RSNA, 2019Supplemental material is available for this article.
