用机器学习预测细菌耐药性,还能解释为什么这样判断。
Predicting Multi-Drug Resistance in Bacterial Isolates Through Performance Comparison and LIME-based Interpretation of Classification Models
- 用五种模型分析临床数据和药敏信息预测多重耐药性。
- XGBoost和LightGBM表现最佳,准确率超90%。
- 用LIME解释结果,找出关键耐药药物,适合临床医生使用。
抗菌药物耐药性,尤其是多重耐药(MDR)的上升,给临床决策带来严峻挑战,因治疗选择有限且传统药敏检测耗时。本研究提出一个可解释的机器学习框架,基于临床特征和抗生素敏感性模式预测细菌分离株的MDR。评估了五种分类模型:逻辑回归、随机森林、AdaBoost、XGBoost和LightGBM。模型在包含9,714个菌株的标注数据集上训练,耐药性以抗生素类别级别编码,以捕捉符合MDR定义的跨类耐药模式。性能评估包括准确率、F1分数、AUC-ROC和马修斯相关系数。集成模型,特别是XGBoost和LightGBM,在所有指标上均表现更优。为解决临床透明度问题,采用局部可解释模型无关解释(LIME)生成实例级解释。LIME识别出喹诺酮类、复方新诺明、粘菌素、氨基糖苷类和呋喃类耐药是预测MDR最强的贡献因素,与已知生物学机制一致。结果表明,结合高性能模型与局部可解释性,既能保证准确性,又能提供可操作的临床洞察,支持更早识别MDR,增强对机器学习辅助临床决策的信任。
原文摘要 · Abstract (English)
The rise of Antimicrobial Resistance, particularly Multi-Drug Resistance (MDR), presents a critical challenge for clinical decision-making due to limited treatment options and delays in conventional susceptibility testing. This study proposes an interpretable machine learning framework to predict MDR in bacterial isolates using clinical features and antibiotic susceptibility patterns. Five classification models were evaluated, including Logistic Regression, Random Forest, AdaBoost, XGBoost, and LightGBM. The models were trained on a curated dataset of 9,714 isolates, with resistance encoded at the antibiotic family level to capture cross-class resistance patterns consistent with MDR definitions. Performance assessment included accuracy, F1-score, AUC-ROC, and Matthews Correlation Coefficient. Ensemble models, particularly XGBoost and LightGBM, demonstrated superior predictive capability across all metrics. To address the clinical transparency gap, Local Interpretable Model-agnostic Explanations (LIME) was applied to generate instance-level explanations. LIME identified resistance to quinolones, Co-trimoxazole, Colistin, aminoglycosides, and Furanes as the strongest contributors to MDR predictions, aligning with known biological mechanisms. The results show that combining high-performing models with local interpretability provides both accuracy and actionable insights for antimicrobial stewardship. This framework supports earlier MDR identification and enhances trust in machine learning-assisted clinical decision support.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。