Disease Diagnosis Based on Multiple Semantic Relationship Prompt Subgraph

Jing Xie,Xin Li,Ye Yuan,Yi Guan,Xitong Guo,Yi Lin
DOI: https://doi.org/10.1109/bibm58861.2023.10385346
2023-01-01
Abstract:Existing disease diagnosis models are often trained on clinical data such as electronic medical records (EMRs), which are lack of knowledge guidance. In this paper, we propose a new disease diagnosis model based on Multiple Semantic Relationship Prompt Subgraph (MSRPS), which can efficiently incorporate medical knowledge and utilize pretrained language models (PLMs) to achieve reasonable and accurate diagnosis results. Firstly, the MSRPS is extracted from the medical knowledge graph based on the patient’s personalized condition. Then graph encoder is used to generate a continuous prompt learning template based on graph neural network (GNN). Finally, a disease diagnosis model with prompt learning template under pre-trained and prompt paradigm is constructed to predict diagnosis results. Compared with the fine-tuning approach, this method needs fewer trainable parameters and less training data but achieve better performance. Experiments are conduct on two multi-label disease diagnosis datasets: ChineseEMR-50 and MIMIC-III-50. Results demonstrate that our model can be used in various pre-trained models and achieve the state-of-the-art results.
What problem does this paper attempt to address?