A top-down supervised learning approach to hierarchical multi-label classification in networks

Applied Network Science

Miguel Romero¹  Camilo Rocha¹  Jorge Finke¹ 

[1] Department of Electronics and Computer Science, Pontificia Universidad Javeriana, Calle 18 N 118-250, 760031, Cali, Colombia;

关键词: Hierarchical classification;    Supervised learning;    XGBoost;    Top-down approach;    Oryza sativa;    Gene function prediction;   

DOI  :  10.1007/s41109-022-00445-3

来源: Springer

【 摘 要 】

Node classification is the task of inferring or predicting missing node attributes from information available for other nodes in a network. This paper presents a general prediction model to hierarchical multi-label classification, where the attributes to be inferred can be specified as a strict poset. It is based on a top-down classification approach that addresses hierarchical multi-label classification with supervised learning by building a local classifier per class. The proposed model is showcased with a case study on the prediction of gene functions for Oryza sativa Japonica, a variety of rice. It is compared to the Hierarchical Binomial-Neighborhood, a probabilistic model, by evaluating both approaches in terms of prediction performance and computational cost. The results in this work support the working hypothesis that the proposed model can achieve good levels of prediction efficiency, while scaling up in relation to the state of the art.

【 授权许可】

CC BY