首页|Knowledge Graph Enhanced Transformers for Diagnosis Generation of Chinese Medicine
Knowledge Graph Enhanced Transformers for Diagnosis Generation of Chinese Medicine
扫码查看
点击上方二维码区域,可以放大扫码查看
原文链接
NETL
NSTL
万方数据
Chinese medicine(CM)diagnosis intellectualization is one of the hotspots in the research of CM modernization.The traditional CM intelligent diagnosis models transform the CM diagnosis issues into classification issues,however,it is difficult to solve the problems such as excessive or similar categories.With the development of natural language processing techniques,text generation technique has become increasingly mature.In this study,we aimed to establish the CM diagnosis generation model by transforming the CM diagnosis issues into text generation issues.The semantic context characteristic learning capacity was enhanced referring to Bidirectional Long Short-Term Memory(BILSTM)with Transformer as the backbone network.Meanwhile,the CM diagnosis generation model Knowledge Graph Enhanced Transformer(KGET)was established by introducing the knowledge in medical field to enhance the inferential capability.The KGET model was established based on 566 CM case texts,and was compared with the classic text generation models including Long Short-Term Memory sequence-to-sequence(LSTM-seq2seq),Bidirectional and Auto-Regression Transformer(BART),and Chinese Pre-trained Unbalanced Transformer(CPT),so as to analyze the model manifestations.Finally,the ablation experiments were performed to explore the influence of the optimized part on the KGET model.The results of Bilingual Evaluation Understudy(BLEU),Recall-Oriented Understudy for Gisting Evaluation 1(ROUGE1),ROUGE2 and Edit distance of KGET model were 45.85,73.93,54.59 and 7.12,respectively in this study.Compared with LSTM-seq2seq,BART and CPT models,the KGET model was higher in BLEU,ROUGE1 and ROUGE2 by 6.00-17.09,1.65-9.39 and 0.51-17.62,respectively,and lower in Edit distance by 0.47-3.21.The ablation experiment results revealed that introduction of BILSTM model and prior knowledge could significantly increase the model performance.Additionally,the manual assessment indicated that the CM diagnosis results of the KGET model used in this study were highly consistent with the practical diagnosis results.In conclusion,text generation technology can be effectively applied to CM diagnostic modeling.It can effectively avoid the problem of poor diagnostic performance caused by excessive and similar categories in traditional CM diagnostic classification models.CM diagnostic text generation technology has broad application prospects in the future.
Chinese medicine diagnosisknowledge graph enhanced transformertext generation
WANG Xin-yu、YANG Tao、GAO Xiao-yuan、HU Kong-fa
展开 >
School of Artificial Intelligence and Information Technology,Nanjing University of Chinese Medicine,Nanjing(210023),China
School of Information Management,Nanjing University,Nanjing(210023),China
Jiangsu Collaborative Innovation Center of Traditional Chinese Medicine in Prevention and Treatment of Tumor,Nanjing(210023),China
Jiangsu Province Engineering Research Center of TCM Intelligence Health Service,Nanjing(210023),China
展开 >
National Natural Science Foundation of ChinaNational Natural Science Foundation of ChinaKey Research and Development Program of Jiangsu ProvinceChina Postdoctoral FoundationPostdoctoral Research Program of Jiangsu ProvinceQinglan Project of Jiangsu Universities 2021