Multi-task deep learning network to predict future macrovascular invasion in hepatocellular carcinoma
Sirui Fu, Haoran Lai, Qiyang Li, Yao Liu, Jiawei Zhang, Jianwen Huang, Xiumei Chen, Chongyang Duan, Xiaoqun Li, Tao Wang, Xiaofeng He, Jianfeng Yan, Ligong Lu, Meiyan Huang
Abstract
BACKGROUND: Models predicting future macrovascular invasion in hepatocellular carcinoma are constructed to assist timely interventions. METHODS: A total of 366 HCC cases were retrospectively collected from five Chinese hospitals between April 2007 and November 2016: the training dataset comprised 281 patients from four hospitals; the external validation dataset comprised 85 patients from another hospital. Multi-task deep learning network-based models were constructed to predict future macrovascular invasion. The discrimination, calibration, and decision curves were compared to identify the best model. We compared the time to macrovascular invasion and overall survival using the best model and related image heterogeneity scores (H-score). Then, we determined the need for a segmentation subnet or the replacement deep learning algorithm by logistic regression in screening clinical/radiological factors. Finally, an applet was constructed for future application. FINDINGS: Similar results were achieved when the patients were subdivided by the H-score. The subnet for segmentation and end-to-end deep learning algorithms improved the performance of the model. INTERPRETATION: Our multi-task deep learning network-based model successfully predicted future macrovascular invasion. In high-risk populations, besides the current first-line treatments, more therapies may be explored for macrovascular invasion.