Page 307 - 《软件学报》2021年第12期
P. 307

邓成龙  等:基于随机森林的宫颈鳞癌放化疗疗效预测                                                        3971


         预测结果最低.
         4.4.3    不同预测算法对比
             由于目前宫颈癌疗效预测方法较少,因此,为验证本文提出的放化疗疗效预测算法优于其他算法,选取目前
         图像分类准确率较高的两种算法(Inception-ResNet-v2 和 Inception-v4)、文献[17]方法与本文算法在训练数据集
         及测试数据集上进行实验对比.图 10、图 11 分别展示了 4 种算法在训练集及测试集上预测放化疗疗效准确率.
















                    Fig.10    Prediction results obtained by different classification algorithms on training set
                                    图 10   不同分类算法在训练集上预测结果















                      Fig.11    Prediction results obtained by different classification algorithms on test set
                                    图 11   不同分类算法在测试集上预测结果

             根据图 11 可知:本文利用随机森林算法预测宫颈鳞癌放化疗疗效的 AUC 值达到 0.921,分别高于其他两种
         神经网络分类算法.Inception-ResNet-v2 和 Inception-v4 模型虽然分类性能优越,但在数据量较少的情况下分类
         效果欠佳.文献[17]只提取了纹理特征,导致描述肿瘤的特征数量及种类不足,使得 LASSO-Logistic 建立的宫颈
         癌治疗敏感性模型预测疗效结果的 AUC 值仅为 0.7867,远低于本文采用的预测模型.

         5    总   结

             本文针对部分宫颈鳞癌患者对放化疗不敏感的问题,提出了基于随机森林算法的疗效预测模型.在该模型
         中,为提高 MR 图像肿瘤区域的分割精度,采用了在小样本下分割性能较好的 U-net 模型.同时,为防止 U-net 模
         型训练阶段出现梯度消失情况,在 U-net 模型中加入了残差学习.此外,为训练泛化能力较强的随机森林模型,实
         验对提取的图像特征进行筛选,并调整了随机森林中决策树的数量以及训练最大特征数量.实验结果表明:本文
         提出的预测模型在分割肿瘤区域敏感性达到 0.801,预测放化疗疗效的准确度达到 0.921.
             本文未来工作包括:第一,开发基于不同权重类型图像的预测模型,包括 T1W1,DWI 等 MR 图像,综合运用多
         种权重类型的图像可以得到更全面的诊断信息,进一步提高宫颈鳞癌放化疗疗效预测性能;第二,未来将收集更
         多病例数,尤其是部分少见病理类型,进一步丰富实验数据集来训练预测模型,提升模型预测能力.
   302   303   304   305   306   307   308   309   310   311   312