Deep learning model based on contrast-enhanced MRI for predicting post-surgical survival in patients with hepatocellular carcinoma

ObjectiveTo develop a deep learning model based on contrast-enhanced magnetic resonance imaging (MRI) data to predict post-surgical overall survival (OS) in patients with hepatocellular carcinoma (HCC).MethodsThis bi-center retrospective study included 564 surgically resected patients with HCC and divided them into training (326), testing (143), and external validation (95) cohorts. This study used a three-dimensional convolutional neural network (3D-CNN) ResNet to learn features from the pretreatment MR images (T1WIpre, late arterial phase, and portal venous phase) and got the deep learning score (DL score). Three cox regression models were established separately using the DL score (3D-CNN model), clinical features (clinical model), and a combination of above (combined model). The concordance index (C-index) was used to evaluate model performance.ResultsWe trained a 3D-CNN model to get DL score from samples. The C-index of the 3D-CNN model in predicting 5-year OS for the training, testing, and external validation cohorts were 0.746, 0.714, and 0.698, respectively, and were higher than those of the clinical model, which were 0.675, 0.674, and 0.631, respectively (P = 0.009, P = 0.204, and P = 0.092, respectively). The C-index of the combined model for testing and external validation cohorts was 0.750 and 0.723, respectively, significantly higher than the clinical model (P = 0.017, P = 0.016) and the 3D-CNN model (P = 0.029, P = 0.036).ConclusionsThe combined model integrating the DL score and clinical factors showed a higher predictive value than the clinical and 3D-CNN models and may be more useful in guiding clinical treatment decisions to improve the prognosis of patients with HCC.