Heng Chung, MW; Liu, J; Tissot, H; (2020) Clinical Knowledge Graph Embedding Representation Bridging the Gap between Electronic Health Records and Prediction Models. In: Proceedings of the 2019 18th IEEE International Conference On Machine Learning And Applications (ICMLA). (pp. pp. 1448-1453). IEEE: Boca Raton, FL, USA.

Preview

Text 2019g-ICMLA-Clinical Knowledge Graph Embedding Representation.pdf - Accepted Version Download (318kB) | Preview

Abstract

Learning knowledge embedding representation is an increasingly important technology. However, the choice of hyperparameters is seldom justified and usually relies on exhaustive search. Understanding the effect of hyperparameter combinations on embedding quality is crucial to avoid the inefficient process and enhance practicality of embedding representation along subsequent machine learning applications. This work focuses on translational embedding models for multi-relational categorized data in the clinical domain. We trained and evaluated models with different combinations of hyperparameters on two clinical datasets. We contrasted the results by comparing metric distributions and fitting a random forest regression model. Classifiers were trained to assess embedding representation quality. Finally, clustering was tested as a validation protocol. We observed consistent patterns of hyperparameter preference and identified those that achieved better results respectively. However, results show different patterns regarding link prediction, which is taken as strong evidence that traditional evaluation protocol used for open-domain data does not necessarily lead to the best embedding representation for categorized data.

Download activity - last month

Export as CSVJSONXML

Download activity - last 12 months

Export as CSVJSONXML

Downloads by country - last 12 months

Export as CSVXMLJSON

Archive Staff Only

View Item