Automatic Emergency Diagnosis with Knowledge-Based Tree Decoding

Automatic diagnosis based on clinical notes is critical especially in the emergency department, where a fast and professional result is vital in assuring proper and timely treatment. Previous works formalize this task as plain text classification and fail to utilize the medically significant tree structure of International Classification of Diseases (ICD) coding system. Besides, external medical knowledge is rarely used before, and we explore it by extracting relevant materials from Wikipedia or Baidupedia. In this paper, we propose a knowledge-based tree decoding model (K-BTD), and the inference procedure is a top-down decoding process from the root node to leaf nodes. The stepwise inference procedure enables the model to give support for decision at each step, which visualizes the diagnosis procedure and adds to the interpretability of final predictions. Experiments on real-world data from the emergency department of a large-scale hospital indicate that the proposed model outperforms all baselines in both micro-F1 and macro-F1, and reduce the semantic distance dramatically.

Authors

• Ke Wang School of Astronautics, Beihang University; Tianmushan Laboratory, Beihang University
• Xuyan Chen
• Ning Chen
• Ting Chen BNRist, Department of Computer Science and Technology, Tsinghua University

Keywords

• AI Ethics: Explainability
• Multidisciplinary Topics and Applications: Biology and Medicine
• Natural Language Processing: NLP Applications and Tools
• Natural Language Processing: Text Classification