Traditional Chinese medicine syndrome type identification method based on graph attention network

Abstract

The invention discloses a traditional Chinese medicine syndrome type identification method based on a graph attention network, and the method comprises the steps: standardizing a plurality of medical case data, and constructing a training set, namely corpus data; respectively establishing a symptom set and a syndrome set for all non-repeated symptoms and syndromes in the corpus; every two symptoms in each medical case data belonging to the training set being connected to serve as nodes, calculating point mutual information between the two symptoms in the symptom set, and updating nodes in a graph by using a graph attention network; weighting the updated nodes by using an attention mechanism to obtain feature vectors of symptoms; inputting the feature vectors into a linear layer for classification to obtain the probability of each syndrome; calculating a loss function in combination with the probability of each syndrome type and the real condition in the training set; performing back propagation according to the loss function, and completing model iteration; and inputting symptom information to be identified into the trained model to obtain a symptom type identification result. According to the invention, the accuracy of identification of the syndrome type is effectively improved.

Description

technical field [0001] The invention relates to the field of TCM syndrome type identification system, and relates to a TCM syndrome type identification method based on graph attention network. Background technique [0002] Traditional Chinese medicine has a history of more than 3,000 years, providing an important guarantee for the medical and health of the Chinese people. Diagnosis and treatment of TCM can be divided into three processes: syndrome differentiation, legislation, and formulating prescriptions. Syndrome differentiation is the process by which doctors comprehensively judge the patient's current status and obtain syndrome types based on the information of the four diagnosis methods, which is a key step in determining the entire TCM diagnosis and treatment. The traditional TCM syndrome identification method first presets the decision weight of each symptom for each syndrome type in advance, then calculates the score of each syndrome type, and takes the syndrome typ...

AI Technical Summary

Problems solved by technology

[0003] Aiming at the above-mentioned deficiencies in the prior art, the invention provides a TCM syndrome type identification method based on graph-attention network, which solves the problem of low accuracy of traditional method syndrome type identification results

Images