Multi-classification Algorithm Based on Truncated Gaussian Prior and Variational Bayesian

Xing TIAN,Huan-huan CHEN
DOI: https://doi.org/10.3969/j.issn.1000-1220.2019.04.001
2019-01-01
Abstract:As one paradigm of classification algorithms,the sparse kernel method has gained a great success in the field of machine learning due to its good interpretability and wide applicability in recent decades. Probabilistic classification vector machine is a repre-sentative in this class. It is built from a truncated Gaussian prior,which not only enables a probabilistic output,but also guarantees the stability of learned kernel parameters. However,these algorithms are proposed for the binary cases and cannot be directly applied to the multi-class classification problems. In the Bayesian framework,this paper proposes a truncated Gaussian multi-class model that can solve the multi-class problem directly,as well as a variational Bayesian-based optimization algorithm to solve this model. This model uses truncated Gaussian prior for the model parameters,so it can better use the corresponding label information of the base samples during the training procedure. The number of weights not increasing with the number of dataset categories not only avoids overfitting but also lessens the time-space consumptions of the optimization algorithm. The update of parameters is constrained within each class. Therefore,there is no interference between different classes. This property not only makes its time complexity superior to other Bayes-ian algorithms,but also ensures that when the dataset is imbalanced,this model will not ignore the minority class. Several experiments on multiple datasets have shown that the model and algorithm presented in this paper are superior to the baseline algorithms in terms of classification error rate and AUC,especially when the number of dataset categories is large.
What problem does this paper attempt to address?