Automatic detecting hepatic hemangioma on enhanced CT images using Faster R-CNN

LI Bo,JIA Kefeng,GAO Zhongsong,KANG Xiaodong,LIU Hanqing,WANG Jiao,WANG Xiaotian
DOI: https://doi.org/10.13929/j.issn.1672-8475.2023.05.012
2023-01-01
Abstract:Objective To construct an automatic detection system of hepatic hemangioma on enhanced CT images using faster regional convolutional neural network(Faster R-CNN), and to observe its performance for detecting hepatic hemangioma. Methods A total of 128 patients(2 304 enhanced CT images) with hepatic hemangioma diagnosed with abdominal enhanced CT were enrolled. The patients were divided into training set(102 cases, 1 836 enhanced CT images) and test set(26 cases, 468 enhanced CT images) at the ratio of 8∶2. Faster R-CNN was used to construct an automatic hepatic hemangioma detection system for enhanced CT images. Based on the migration learning scheme, Resnet50 pre-trained classification network was taken as the basic skeleton of feature extraction module, while region proposal net was applied to extract the features of enhanced CT images in training set. Then boundary box classification regression module was performed to output the exact position coordinates and category probability scores of the predicted border. During the training process of training set, the Loss curve was drawn to evaluate the training efficacy and stability of the model, while stochastic gradient descent method was used as the optimizer to optimize parameters so as to improve the performance of the model. The mean average precision(mAP) was used to evaluate the performance of the automatic detection system for detecting hepatic hemangiomas on enhanced CT images in test set. Results Loss curve of the automatic detection system showed a rapid decreasing trend, suggesting that the model had good learning ability and stable prediction performance. When the iteration epoch was 40-80, the mAP of the automatic detection system for detecting hepatic hemangioma on 468 enhanced CT images in test set was 0.962—0.973, with small fluctuation, indicating that the model had converged and its automatic detecting effect was good. Conclusion The constructed automatic detection system could effectively detect hepatic hemangioma on enhanced CT images.
What problem does this paper attempt to address?