Knowledge-routed Automatic Diagnosis with Heterogeneous Patient-oriented Graph

Zhiang Li,Tong Ruan
DOI: https://doi.org/10.1109/access.2024.3412419
IF: 3.9
2024-01-01
IEEE Access
Abstract:Automated diagnosis, as a temporary medical supplement, has gained significant attention in research in recent years. Existing methods employ sequence generation approaches to inquire about symptoms and diagnose diseases. However, these methods ignore the fact that: 1) doctors utilize their past experience and similar cases to aid in diagnosis in real-world scenarios; 2) doctors inquire about key symptoms that serve as vital diagnostic evidence within limited conversations. To address these issues, we propose an end-to-end model KDPoG. Firstly, in addition to use the symptom and attribute embedding, we propose patient-oriented graph enhanced representation learning, which is built by a patient-oriented graph and learned with heterogeneous graph convolution networks. Furthermore, based on the encoder built with gated attention units, we propose knowledge-guided attention mechanism learning, which incorporates conditional probabilities of co-occurrence between symptom pairs. Finally, we utilize two linear layers as the classification module to achieve symptom probing and disease diagnosis. We conduct extensive experiments on four public datasets, which demonstrate that our proposed model outperforms the state-of-the-art methods. We achieve an average absolute improvement of over 2% in disease diagnosis accuracy. Particularly, on the Muzhi-10 dataset, we observe an absolute improvement of over 14.7% in symptom recall rate.
What problem does this paper attempt to address?