基于增量学习的肺炎克雷伯菌碳青霉烯类耐药风险预测

Incremental learning-based risk prediction for carbapenems resistance in Klebsiella pneumoniae

  • 摘要:
    目的 利用增量学习框架下的 IXGBoost 模型,对肺炎克雷伯菌碳青霉烯类耐药性进行早期预测。
    方法 收集2015年1月-2024年12月大连理工大学附属中心医院35 060例肺炎克雷伯菌感染患者的病历数据,经过数据清洗和预处理后,提取显著的耐药风险相关特征变量。数据集按8∶2的比例随机分为训练集和测试集。初始批次基于2015-2020年数据,随后用2021-2024年每周新增数据进行增量训练。选取支持向量机、决策树、随机森林、逻辑回归、XGBoost、LightGBM模型进行对比。模型性能通过K折交叉验证在测试集进行评估。
    结果 IXGBoost模型不但能实现了对数据的实时学习和预测,其精确率(0.982 3)、召回率(0.981 3)、F1分数(0.981 8)和AUC(0.842 1)等性能指标均优于其他对比模型。
    结论 本研究证实了增量学习方法在预测肺炎克雷伯菌碳青霉烯类耐药性方面的有效性,揭示了影响耐药性产生的重要风险因素,对于临床预防及深入理解耐药性产生的根本机制具有重要意义。

     

    Abstract:
    OBJECTIVE  To achieve early prediction for carbapenems resistance of Klebsiella pneumoniae based on the IXGBoost model under an incremental learning framework.
    METHODS  Medical records of 35 060 patients infected with K. pneumoniae at the Central Hospital of Dalian University of Technology from Jan. 2015 to Dec. 2024 were collected. After data cleaning and preprocessing, significant characteristic variables related to the risk of drug resistance were extracted. The dataset was randomly split into training and test sets at an 8:2 ratio. The initial batch was based on data from 2015 to 2020, followed by incremental training with weekly new data from 2021 to 2024. Models, including support vector machine, decision tree, random forest, logistic regression, XGBoost and LightGBM, were selected for comparison. Model performance was evaluated on the test set through K-fold cross-validation.
    RESULTS  The IXGBoost model not only enabled real-time learning and prediction for data, but also outperformed other models in performance indicators, including precision (0.9823), recall rate (0.9813), F1-score (0.9818) and AUC (0.8421).
    CONCLUSIONS  This study confirms the effectiveness of incremental learning in predicting the carbapenems resistance of K. pneumoniae, identifies key risk factors influencing resistance development, and is of great significance for clinical prevention and understanding the underlying mechanisms of drug resistance.

     

/

返回文章
返回