VGBNet: a disease diagnosis model based on local and global information fusion. (2nd May 2023)

• Record Type: Journal Article
• Authors: Li, Yong; Zhao, Xinyu; Ma, Manfu; Zhang, Qiang; Jia, Hai; Wang, Xia
• Is Part Of: International journal of computing science and mathematics. Volume 17:Number 2(2023)
• Page Start: 107
• Page End: 122
• Publication Date: 2023-05-02
• Subjects: unbalanced data set -- disease prediction -- GCN -- graph convolutional neural network -- attention mechanism; Mathematics -- Periodicals; Computer science -- Periodicals; Mathematics -- Data processing -- Periodicals; 510.285
• Languages: English
• ISSNs: 1752-5055

Abstract

There are significant differences in the data volume of different types of diseases in the electronic medical record (EMR) data. In addition, the mainstream auxiliary diagnosis and prediction models have two shortcomings. They either only pay attention to the local features of medical records and ignore the global features, or only pay attention to the global features and ignore the local features. In response to these problems, we propose a method of fusion random resampling to balance the data set, Using graph convolutional neural network (GCN) to extract global features, combined with a bidirectional self-attention network (BERT), a VGBNet model is proposed to link local and global features to achieve diagnosis and prediction of diseases. A large number of experiments show that compared with the state-of-the-art models, the VGBNet model has improved F1 value and accuracy. This is of great significance to the precise auxiliary diagnosis.