Combination of UNet++ and ResNeSt to classify chronic inflammation of the choledochal cystic wall in patients with pancreaticobiliary maljunction

Abstract

Objectives: The aim of this study was to establish an automatic classification model for chronic inflammation of the choledoch wall using deep learning with CT images in patients with pancreaticobiliary maljunction (PBM). Methods: CT images were obtained from 76 PBM patients, including 61 cases assigned to the training set and 15 cases assigned to the testing set. The region of interest (ROI) containing the choledochal lesion was extracted and segmented using the UNet++ network. The degree of severity of inflammation in the choledochal wall was initially classified using the ResNeSt network. The final classification result was determined per decision rules. Grad-CAM was used to explain the association between the classification basis of the network and clinical diagnosis. Results: Segmentation of the lesion on the common bile duct wall was roughly obtained with the UNet++ segmentation model and the average value of Dice coefficient of the segmentation model in the testing set was 0.839 ± 0.150, which was verified through fivefold cross-validation. Inflammation was initially classified with ResNeSt18, which resulted in accuracy = 0.756, sensitivity = 0.611, specificity = 0.852, precision = 0.733, and area under curve (AUC) = 0.711. The final classification sensitivity was 0.8. Grad-CAM revealed similar distribution of inflammation of the choledochal wall and verified the inflammation classification. Conclusions: By combining the UNet++ network and the ResNeSt network, we achieved automatic classification of chronic inflammation of the choledoch in PBM patients and verified the robustness through cross-validation performed five times. This study provided an important basis for classification of inflammation severity of the choledoch in PBM patients. Advances in knowledge: We combined the UNet++ network and the ResNeSt network to achieve automatic classification of chronic inflammation of the choledoch in PBM. These results provided an important basis for classification of choledochal inflammation in PBM and for surgical therapy.