基于卷积神经网络的人工智能烧伤深度识别模型的建立及测试效果

目的建立基于卷积神经网络的人工智能烧伤深度识别模型并测试其效果。方法在本诊断试验评价研究中,收集中南大学湘雅医院(下称笔者单位)2010年1月—2019年12月收治的符合入选标准的221例烧伤患者伤后48 h内创面照片484张,采用随机数字编号。采用图像查看软件圈出目标创面,由笔者单位烧伤整形科3名具有5年以上专科工作经验的主治医师判断烧伤深度,用不同颜色标记浅Ⅱ度、深Ⅱ度或Ⅲ度烧伤后,按224×224像素的尺寸切割得到完整大小的图像块5637张。采用图片生成器将3种深度烧伤图像块均扩充至10000张后,将每种烧伤深度图像块按7.0∶1.5∶1.5比例分为训练集、验证集和测试集。在Keras 2.2.4 Python 2.8.0版本下,采用卷积神经网络中的残差网络ResNet-50构建人工智能烧伤深度识别模型,输入训练集进行训练,利用验证集对模型进行调整、优化。利用测试集测试构建的模型识别各类烧伤深度的准确率,计算精确率、召回率及F1指数;通过降维工具tSNE将测试结果降维可视化生成二维tSNE云图,观察各类烧伤深度分布情况;根据模型对3种烧伤深度识别的敏感度及特异度,绘制出相应受试者工作特征(ROC)曲线,计算ROC曲线下面积。结果(1)经测试集测试,人工智能烧伤深度识别模型识别浅Ⅱ度、深Ⅱ度、Ⅲ度烧伤的精确率分别为84%(1095/1301)、81%(1215/1499)、82%(1395/1700),召回率分别为73%(1095/1500)、81%(1215/1500)、93%(1395/1500),F1指数分别为0.78、0.81、0.87。(2)tSNE云图显示,人工智能烧伤深度识别模型测试集测试结果中不同烧伤深度之间总体重叠较少,其中浅Ⅱ度与深Ⅱ度、深Ⅱ度与Ⅲ度烧伤之间重叠相对较多,而浅Ⅱ度与Ⅲ度烧伤之间重叠相对较少。(3)人工智能烧伤深度识别模型识别3种烧伤深度的ROC曲线下面积均≥0.94。结论采用ResNet-50网络建立的人工智能烧伤深度识别模型可较准确地识别烧伤患者早期创面照片中烧伤深度,特别是浅Ⅱ度与Ⅲ度烧伤,有望用于临床烧伤深度辅助诊断,提高诊断准确率。