Maximum mutual information regularized classification

J.J.-Y. Wang, Y. Wang, S. Zhao, X. Gao
Engineering Applications of Artificial Intelligence, 37, 1-8, (2015)

Maximum mutual information regularized classification

Keywords

Pattern classification, Maximum mutual information, Entropy, Gradient descend

Abstract

​In this paper, a novel pattern classification approach is proposed by regularizing the classifier learning to maximize mutual information between the classification response and the true class label. We argue that, with the learned classifier, the uncertainty of the true class label of a data sample should be reduced by knowing its classification response as much as possible. The reduced uncertainty is measured by the mutual information between the classification response and the true class label. To this end, when learning a linear classifier, we propose to maximize the mutual information between classification responses and true class labels of training samples, besides minimizing the classification error and reducing the classifier complexity. An objective function is constructed by modeling mutual information with entropy estimation, and it is optimized by a gradient descend method in an iterative algorithm. Experiments on two real world pattern classification problems show the significant improvements achieved by maximum mutual information regularization.

Code

DOI: 10.1016/j.engappai.2014.08.009

Sources

Website PDF

See all publications 2015