A label ranking method based on Gaussian mixture model

作者:

Highlights:

摘要

Label ranking studies the issue of learning a model that maps instances to rankings over a finite set of predefined labels. In order to relieve the cost of memory and time during training and prediction, we propose a novel approach for label ranking problem based on Gaussian mixture model in this paper. The key idea of the approach is to divide the label ranking training data into multiple clusters using clustering algorithm, and each cluster is described by a Gaussian prototype. Then, a Gaussian mixture model is introduced to model the mapping from instances to rankings. Finally, a predicted ranking is obtained with maximum posterior probability. In the experiments, we compare our method with two state-of-the-art label ranking approaches. Experimental results show that our method is fully competitive in terms of predictive accuracy. Moreover, the proposed method also provides a measure of the reliability of the corresponding predicted ranking.

论文关键词:Machine learning,Label ranking,Multi-label learning,Gaussian mixture model,Clustering

论文评审过程:Received 10 October 2013, Revised 12 July 2014, Accepted 30 August 2014, Available online 16 September 2014.

论文官网地址:https://doi.org/10.1016/j.knosys.2014.08.029