Validating the representation of distance between infarct diseases using Word2Vec word embedding

Abstract

AbstractObjectiveTo determine if inter-disease distances between word embedding vectors using the picot-and-cluster strategy (PCS) are a valid quantitative representation of similar disease groups in a limited domain.Materials and MethodsAbstracts were extracted from the Ichushi-Web database and subjected to morphological analysis and training using the Word2Vec. From this, word embedding vectors were obtained. For words including “infarction”, we calculated the cophenetic correlation coefficient (CCC) as an internal validity measure and the adjusted rand index (ARI), normalized mutual information (NMI), and adjusted mutual information (AMI) with ICD-10 codes as the external validity measures. This was performed for each combination of metric and hierarchical clustering method.ResultsSeventy-one words included “infarction”, of which 38 diseases matched the ICD-10 standard with the appearance of 21 unique ICD-10 codes. The CCC was most significant at 0.8690 (metric and method: euclidean and centroid), while the AMI was maximal at 0.4109 (metric and method: cosine and correlation, and average and weighted). The NMI and ARI were maximal at 0.8463 and 0.3593, respectively (metric and method: cosine and complete).DiscussionThe metric and method that maximized the internal validity measure were different from those that maximized the external validity measures; both produced different results. The Cosine distance should be used when considering ICD-10, and the Euclidean distance when considering the frequency of word occurrence.ConclusionThe distributed representation, when trained by Word2Vec on the “infarction” domain from a Japanese academic corpus, provides an objective inter-disease distance used in PCS.