Explainable Diagnosis Prediction through Neuro-Symbolic Integration.

Journal: AMIA Joint Summits on Translational Science proceedings. AMIA Joint Summits on Translational Science
Published Date:

Abstract

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 weights and thresholds. Our models, particularly Mmulti-pathway and Mcomprehensive, 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 ofprecision 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.

Authors

  • Qiuhao Lu
    McWilliams School of Biomedical Informatics, University of Texas Health Science Center, Houston, TX, USA.
  • Rui Li
    Department of Oncology, Xiyuan Hospital, China Academy of Chinese Medical Science, Beijing, China.
  • Elham Sagheb
    Department of Artificial Intelligence and Informatics, Mayo Clinic, 200 First St SW, Rochester, MN, 55905, USA.
  • Andrew Wen
    Department of Medical Informatics and Clinical Epidemiology, Oregon Health & Science University, Portland, OR, USA.
  • Jinlian Wang
    McWilliams School of Biomedical Informatics, University of Texas Health Science Center, Houston, TX, USA.
  • Liwei Wang
    Department of Health Sciences Research, Mayo Clinic, Rochester, MN, USA.
  • Jungwei W Fan
    Department of Artificial Intelligence and Informatics, Mayo Clinic, Rochester, MN, USA.
  • Hongfang Liu
    Department of Artificial Intelligence & Informatics, Mayo Clinic, Rochester, MN, United States.

Keywords

No keywords available for this article.