Automatic Emergency Diagnosis with Knowledge-Based Tree Decoding

Automatic Emergency Diagnosis with Knowledge-Based Tree Decoding

Ke Wang, Xuyan Chen, Ning Chen, Ting Chen

Proceedings of the Twenty-Ninth International Joint Conference on Artificial Intelligence
Main track. Pages 3407-3414. https://doi.org/10.24963/ijcai.2020/471

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.
Keywords:
Multidisciplinary Topics and Applications: Biology and Medicine
AI Ethics: Explainability
Natural Language Processing: Text Classification
Natural Language Processing: NLP Applications and Tools