Medical Triage Chatbot Diagnosis Improvement via Multi-relational Hyperbolic Graph Neural Network

Medical triage chatbot is widely used in pre-diagnosis by asking symptom and medical history-related questions. Information collected from patients through an online chatbot system is often incomplete and imprecise, and thus it's essentially hard to achieve precise triaging. In this paper, we propose Multi-relational Hyperbolic Diagnosis Predictor (MHDP) --- a novel multi-relational hyperbolic graph neural network-based approach, to build a disease predictive model. More specifically, in MHDP, we generate a heterogeneous graph consisting of symptoms, patients, and diagnoses nodes, and then derive node representations by aggregating neighborhood information recursively in the hyperbolic space. Experiments conducted on two real-world datasets demonstrate that the proposed MHDP approach surpasses state-of-the-art baselines.

[1]  Douwe Kiela,et al.  Hyperbolic Graph Neural Networks , 2019, NeurIPS.

[2]  Douwe Kiela,et al.  Poincaré Embeddings for Learning Hierarchical Representations , 2017, NIPS.

[3]  Le Song,et al.  GRAM: Graph-based Attention Model for Healthcare Representation Learning , 2016, KDD.

[4]  Max Welling,et al.  Modeling Relational Data with Graph Convolutional Networks , 2017, ESWC.

[5]  Qun Liu,et al.  HyperText: Endowing FastText with Hyperbolic Geometry , 2020, FINDINGS.

[6]  Yujia Li,et al.  Learning the Graphical Structure of Electronic Health Records with Graph Convolutional Transformer , 2020, AAAI.

[7]  Jimeng Sun,et al.  MiME: Multilevel Medical Embedding of Electronic Health Records for Predictive Healthcare , 2018, NeurIPS.

[8]  Yanfang Ye,et al.  Hyperbolic Graph Attention Network , 2019, IEEE Transactions on Big Data.

[9]  Fenglong Ma,et al.  KAME: Knowledge-based Attention Model for Diagnosis Prediction in Healthcare , 2018, CIKM.

[10]  Gerald Sommer,et al.  A hyperbolic multilayer perceptron , 2000, Proceedings of the IEEE-INNS-ENNS International Joint Conference on Neural Networks. IJCNN 2000. Neural Computing: New Challenges and Perspectives for the New Millennium.

[11]  GRAM: A Framework for Dynamically Mixing Precisions in GPU Applications , 2021, ACM Trans. Archit. Code Optim..

[12]  Jure Leskovec,et al.  Inductive Representation Learning on Large Graphs , 2017, NIPS.

[13]  Christopher R'e,et al.  Low-Dimensional Hyperbolic Knowledge Graph Embeddings , 2020, ACL.

[14]  Yubo Chen,et al.  HyperCore: Hyperbolic and Co-graph Representation for Automatic ICD Coding , 2020, ACL.

[15]  Maksims Volkovs,et al.  HGCF: Hyperbolic Graph Convolution Networks for Collaborative Filtering , 2021, WWW.

[16]  Xian Wu,et al.  MNN: Multimodal Attentional Neural Networks for Diagnosis Prediction , 2019, IJCAI.

[17]  Alexandros Kalousis,et al.  Hyperbolic Knowledge Graph Embeddings for Knowledge Base Completion , 2019, ESWC.

[18]  KAME , 2018, Proceedings of the 27th ACM International Conference on Information and Knowledge Management.

[19]  Xian Wu,et al.  MHM: Multi-modal Clinical Data based Hierarchical Multi-label Diagnosis Prediction , 2020, SIGIR.

[20]  Philip S. Yu,et al.  Heterogeneous Similarity Graph Neural Network on Electronic Health Records , 2020, 2020 IEEE International Conference on Big Data (Big Data).

[21]  George A. Anastassiou,et al.  Multivariate hyperbolic tangent neural network approximation , 2011, Comput. Math. Appl..

[22]  Lars Schmidt-Thieme,et al.  BPR: Bayesian Personalized Ranking from Implicit Feedback , 2009, UAI.

[23]  Gao Cong,et al.  HyperML: A Boosting Metric Learning Approach in Hyperbolic Space for Recommender Systems , 2018, WSDM.

[24]  Fenglong Ma,et al.  Dipole: Diagnosis Prediction in Healthcare via Attention-based Bidirectional Recurrent Neural Networks , 2017, KDD.

[25]  Peter Szolovits,et al.  MIMIC-III, a freely accessible critical care database , 2016, Scientific Data.