A drug word vector conversion method in electronic medical record based on attention mechanism
10.16781/j.0258-879x.2020.10.1129
- VernacularTitle: 一种基于注意力机制的电子病历中药物词向量转化方法
- Author:
Qing-Hua WANG
1
Author Information
1. Department of Medical Informatics, School of Medicine, Nantong University
- Publication Type:Journal Article
- Keywords:
Attention mechanism;
Electronic health records;
Language concept word vector;
Systemic lupus erythematosus
- From:
Academic Journal of Second Military Medical University
2020;41(10):1129-1135
- CountryChina
- Language:Chinese
-
Abstract:
Objective To propose a drug word vector conversion model based on attention mechanism named Drug2vec for generating vectorized representation of drug information, and to compare the vector conversion effect with Word2vec and Med2vec. Methods Using the attention mechanism to capture the roles of medical entities on the central word, we proposed a Drug2vec model to convert medical entities in unstructured electronic medical records into vectors. Using the systemic lupus erythematosus (SLE) dataset of 14 219 patients and 963 drug entities, we tested the effect of the drug vectors generated by Drug2vec and compared it with the widely used language concept space vector conversion models Word2vec and Med2vec. Results In the SLE dataset, the accuracy of drug vectors generated by Drug2vec was higher than those of Word2vec and Med2vec models. The rank results of the similarity of drugs showed that the drug vectors generated by Drug2vec were consistent with the clinician's medication order. Conclusion Drug2vec model can more accurately modify central drug entities using contextual entities, producing more precise drug vectors.