The Graph-based Mutual Attentive Network for Automatic Diagnosis

The Graph-based Mutual Attentive Network for Automatic Diagnosis

Quan Yuan, Jun Chen, Chao Lu, Haifeng Huang

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

The automatic diagnosis has been suffering from the problem of inadequate reliable corpus to train a trustworthy predictive model. Besides, most of the previous deep learning based diagnosis models adopt the sequence learning techniques (CNN or RNN), which is difficult to extract the complex structural information, e.g. graph structure, between the critical medical entities. In this paper, we propose to build the diagnosis model based on the high-standard EMR documents from real hospitals to improve the accuracy and the credibility of the resulting model. Meanwhile, we introduce the Graph Convolutional Network into the model that alleviates the sparse feature problem and facilitates the extraction of structural information for diagnosis. Moreover, we propose the mutual attentive network to enhance the representation of inputs towards the better model performance. The evaluation conducted on the real EMR documents demonstrates that the proposed model is more accurate compared to the previous sequence learning based diagnosis models. The proposed model has been integrated into the information systems in over hundreds of primary health care facilities in China to assist physicians in the diagnostic process.
Keywords:
Multidisciplinary Topics and Applications: Biology and Medicine
Machine Learning Applications: Bio/Medicine
Natural Language Processing: NLP Applications and Tools