Construction of a predictive early warning model based on machine learning neural network for prognosis of patients with traumatic brain injury

构建基于机器学习神经网络的创伤性脑损伤患者预后预测早期预警模型

阅读:2

Abstract

OBJECTIVES: The analysis of prognostic regression of patients in the regional treatment programme for severe trauma can improve the survival rate and quality of life of patients. The aim of this study is to construct an accurate and effective prognostic prediction model for the optimization and development of the regional trauma care network. METHODS: We firstly extracted the clinical data of patients admitted to the regional treatment programme for severe trauma in our hospital during the period from January 2020 to December 2022. The criterion weighting method was adopted to comprehensively evaluated the AIS scores of the cumulative patients in different parts of the body. Based on the regression, the patients were divided into cured group, improved group and poor prognosis group. Based on the dependent variables, the included influencing factors were subjected to univariate analysis, multivariate analysis, and prediction model construction and comparison study. Genetic algorithm was used to solve the planning model; combined with the results of unifactorial analysis and Xgboost, RF was used to screen the features, and the interpretable model (SHAP) and column charts were used to verify the effectiveness of the screened features. RESULTS: After feature screening and interpretable model validation, 11 indicators such as the main diagnostic score, AIS score and albumin were ultimately included as the important influencing factors of outcome variables, among which albumin was the more important protective factor, and the diagnostic score and AIS score were the more important risk factors. In the comparative study of categorical prediction models, the RF-Transformer-LSTM model achieved the most excellent prediction effect, the accuracy rate of the model test set was 0.9556, the precision rate was 0.9615, the TPR was 0.9474, the TNR was 0.9619, F1 value of 0.9544 as well as AUC value of 0.9271, and in the construction of the three-classification model, the accuracy of the model test set reached 0.9310. CONCLUSION: We constructed RF-Transformer-LSTM prediction model has high prediction accuracy and good interpretability in practical applications, which can provide strong support for the optimisation of regional trauma treatment strategies.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。