Publications

Evaluating the Effectiveness of Margin Parameter when Learning Knowledge Embedding Representation for Domain-specific Multi-relational Categorized Data

 

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.

 

URL
https://arxiv.org/abs/1912.10264

 

DOI
10.48550/arXiv.1912.10264

 

LaTeX
@inproceedings{Chung2020StarAI,
	title     = {Evaluating the Effectiveness of Margin Parameter when Learning Knowledge Embedding Representation for Domain-specific Multi-relational Categorized Data},
	booktitle = {StarAI 2020 - Ninth International Workshop on Statistical Relational AI},
	author    = {Matthew Wai Heng Chung and Hegler Tissot},
	publisher = {AAAI}
	venue     = {New York, USA},
	month     = {Feb},
	year      = {2020},
	url       = {http://www.starai.org/2020/}
}