arXiv:2410.01855cs.LGcs.AI2024-10被引 26

用逻辑神经网络让糖尿病诊断模型既准又可解释。

Explainable Diagnosis Prediction through Neuro-Symbolic Integration

  • 结合医学规则与可学习阈值,构建可解释的神经符号模型。
  • 在糖尿病预测中准确率达80.52%,AUROC达0.8457。
  • 适合需要透明决策的临床AI场景,如医疗辅助诊断。

诊断预测是医疗领域关键任务,及时准确识别疾病可显著改善患者预后。传统机器学习与深度学习模型虽表现优异,但缺乏可解释性,难以满足临床需求。本文探索神经符号方法,特别是逻辑神经网络(LNNs),构建可解释的诊断预测模型。通过将领域知识以逻辑规则形式融入模型,并结合可学习阈值,设计出 $M_{\text{multi-pathway}}$ 与 $M_{\text{comprehensive}}$ 模型。在糖尿病预测案例中,其准确率最高达80.52%,AUROC达0.8457,优于逻辑回归、SVM和随机森林等传统模型。模型中的权重与阈值可直接反映特征贡献,提升可解释性且不牺牲预测能力。结果表明,神经符号方法有望弥合医疗AI中精度与可解释性的差距。未来工作将拓展至更大更多样化的数据集,验证其在多种疾病与人群中的适用性。

原文摘要 · Abstract (English)

Diagnosis prediction is a critical task in healthcare, where timely and accurate identification of medical conditions can significantly impact patient outcomes. Traditional machine learning and deep learning models have achieved notable success in this domain but often lack interpretability which is a crucial requirement in clinical settings. In this study, we explore the use of neuro-symbolic methods, specifically Logical Neural Networks (LNNs), to develop explainable models for diagnosis prediction. Essentially, we design and implement LNN-based models that integrate domain-specific knowledge through logical rules with learnable thresholds. Our models, particularly $M_{\text{multi-pathway}}$ and $M_{\text{comprehensive}}$, demonstrate superior performance over traditional models such as Logistic Regression, SVM, and Random Forest, achieving higher accuracy (up to 80.52\%) and AUROC scores (up to 0.8457) in the case study of diabetes prediction. The learned weights and thresholds within the LNN models provide direct insights into feature contributions, enhancing interpretability without compromising predictive power. These findings highlight the potential of neuro-symbolic approaches in bridging the gap between accuracy and explainability in healthcare AI applications. By offering transparent and adaptable diagnostic models, our work contributes to the advancement of precision medicine and supports the development of equitable healthcare solutions. Future research will focus on extending these methods to larger and more diverse datasets to further validate their applicability across different medical conditions and populations.

可解释AI医疗诊断神经符号

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。