Zhou Leilei, Zhang Zuoheng, Chen Yu-Chen, Zhao Zhen-Yu, Yin Xin-Dao, Jiang Hong-Bing
Department of Medical Equipment, Nanjing First Hospital, Nanjing Medical University, Nanjing 210006, China.
State Key Laboratory of Bioelectronics, Jiangsu Key Laboratory for Bio materials and Devices, School of Biological Science and Medical Engineering, Southeast University, Nanjing 210096, China.
Transl Oncol. 2019 Feb;12(2):292-300. doi: 10.1016/j.tranon.2018.10.012. Epub 2018 Dec 17.
To investigate the effect of transfer learning on computed tomography (CT) images for the benign and malignant classification on renal tumors and to attempt to improve the classification accuracy by building patient-level models.
One hundred ninety-two cases of renal tumors were collected and identified by pathologic diagnosis within 15 days after enhanced CT examination (66% male, 70% malignant renal tumors, average age of 62.27 ± 12.26 years). The InceptionV3 model pretrained by the ImageNet dataset was cross-trained to perform this classification. Five image-level models were established for each of the Slice, region of interest (ROI), and rectangular box region (RBR) datasets. Then, two patient-level models were built based on the optimal image-level models. The network's performance was evaluated through analysis of the receiver operating characteristic (ROC) and five-fold cross-validation.
In the image-level models, the test results of model trained on the Slice dataset [accuracy (ACC) = 0.69 and Matthews correlation coefficient (MCC) = 0.45] were the worst. The corresponding results on the ROI dataset (ACC = 0.97 and MCC = 0.93) were slightly better than those on the RBR dataset (ACC = 0.93 and MCC = 0.85) when freezing the weights before the mixed6 layer. Compared with the image-level models, both patient-level models could discriminate better (ACC increased by 2%-5%) on the RBR and Slice datasets.
Deep learning can be used to classify benign and malignant renal tumors from CT images. Our patient-level models could benefit from 3D data to improve the accuracy.
研究迁移学习对肾肿瘤计算机断层扫描(CT)图像进行良恶性分类的效果,并尝试通过构建患者水平模型提高分类准确率。
收集192例肾肿瘤病例,在增强CT检查后15天内通过病理诊断进行确诊(男性占66%,恶性肾肿瘤占70%,平均年龄62.27±12.26岁)。对由ImageNet数据集预训练的InceptionV3模型进行交叉训练以执行此分类。针对切片、感兴趣区域(ROI)和矩形框区域(RBR)数据集分别建立了5个图像水平模型。然后,基于最优的图像水平模型构建了2个患者水平模型。通过分析受试者工作特征(ROC)曲线和五折交叉验证来评估网络性能。
在图像水平模型中,在切片数据集上训练的模型测试结果[准确率(ACC)=0.69,马修斯相关系数(MCC)=0.45]最差。在混合6层之前冻结权重时,ROI数据集上的相应结果(ACC=0.97,MCC=0.93)略优于RBR数据集上的结果(ACC=0.93,MCC=0.85)。与图像水平模型相比,两个患者水平模型在RBR和切片数据集上的辨别能力都更好(ACC提高了2%-5%)。
深度学习可用于从CT图像对肾肿瘤的良恶性进行分类。我们的患者水平模型可受益于三维数据以提高准确率。