关键词:
创伤性颅脑损伤
机器学习
卷积神经网络
支持向量机
支持向量回归机
摘要:
目的当前,成人和儿童创伤性脑损伤的治疗和护理是一个难以解决的全球健康问题。对创伤性颅脑损伤患者的预后和住院时间进行预测可以使得临床治疗效果获得提高,并有助于实现精准医疗,从而大大减少社会医疗负担。随着现代社会计算机技术的蓬勃发展,使得将新的机器学习方法应用于创伤性颅脑损伤(traumatic brain injury,TBI)领域成为可能,已有相关研究表明,计算机方法可以用于患者的相关临床结果的预测研究,准确率较高,对于确定临床治疗的预后和成本效益可能很有价值。本研究共收集了1112位临床诊断为TBI患者的相关临床资料并对其进行回顾性分析,结合多种机器学习(machine-learning,ML)方法和国内外相关研究结果,建立混合模型,用于预测成人和儿童创伤性颅脑损伤的预后和住院时间。方法我们设法收集了2017年5月至2022年5月期间安徽医科大学第二附属医院神经外科中心收治并诊断为TBI患者的相关临床信息,通过筛选和随机分割,其中80%用于训练模型,20%用于测试模型。对于两种预测结果我们分别对比构建了目前国内外研究常用且精度较高的四种机器学习模型,并且使用五次交叉验证方法对模型进行了反复训练和测试,以避免出现过度拟合的情况。在机器学习模型中,11种自变量被用作输入变量,用于评估患者预后的GOS评分和患者的住院时间被用作输出变量。模型训练完成后,我们利用五轮交叉验证方法获得了每个机器学习模型的误差并进行比较,以选择最佳预测模型。最后,为了进一步验证我们的实验结果,我们收集筛选并利用2021年6月至2022年2月期间在安徽医科大学第一附属医院神经外科中心治疗的TBI患者的临床数据对所有模型进行了外部测试。结果最终建立的卷积神经网络-支持向量机(convolutional neural network-support vector machine,CNN-SVM)混合模型预测作为TBI患者预后的GOS分类结果的准确率(accuracy,ACC)在测试集和外部验证集中分别为93%和93.69%,曲线下面积(area under curve,AUC)在测试集和外部验证集中分别为94.68%和94.32%。最终建立的卷积神经网络-支持向量回归(convolutional neural network-supportvector regression,CNN-SVR)混合模型预测TBI患者住院时间的平均绝对百分比误差(mean absolute percentage error,MAPE)在测试集和外部验证集中的结果分别为10.72%和10.44%,该混合预测模型在测试集和外部验证集的决定系数(R2)分别为0.93和0.92。我们的混合模型与单独建立的反向传播神经网络模型、卷积神经网络模型和支持向量机模型相比,获得的预测结果最优,并具有很高的可信度。结论本研究证明了通过结合多种新型机器学习方法建立的两个混合模型的临床效用,可以用于准确预测成人和儿童创伤性颅脑损伤的预后和住院时间。这些模型的应用可以减轻医生在评估创伤性颅脑损伤患者时的负担,协助临床医生进行医疗决策,并为以后的相关研究者提供了一定的参考。