Advances in Feature Selection with Mutual Information

Michel Verleysen (1) , Fabrice Rossi (2) , Damien François (3)

1 DICE - MLG - Dispositifs Intégrés et Circuits Electroniques Machine Learning Group
2 LTCI - Laboratoire Traitement et Communication de l'Information
3 CSAM - Centre for Systems Engineering and Applied Mechanics

The selection of features that are relevant for a prediction or classification problem is an important problem in many domains involving high-dimensional data. Selecting features helps fighting the curse of dimensionality, improving the performances of prediction or classification methods, and interpreting the application. In a nonlinear context, the mutual information is widely used as relevance criterion for features and sets of features. Nevertheless, it suffers from at least three major limitations: mutual information estimators depend on smoothing parameters, there is no theoretically justified stopping criterion in the feature selection greedy procedure, and the estimation itself suffers from the curse of dimensionality. This chapter shows how to deal with these problems. The two first ones are addressed by using resampling techniques that provide a statistical basis to select the estimator parameters and to stop the search procedure. The third one is addressed by modifying the mutual information criterion into a measure of how features are complementary (and not only informative) for the problem at hand.

Mots clés: Feature Selection; Mutual Information; Resampling; Forward selection

Domaines: Apprentissage [cs.LG]; Théorie de l'information [cs.IT]; Théorie de l'information et codage [math.IT]

Michel Verleysen, Fabrice Rossi, Damien François. Advances in Feature Selection with Mutual Information. Villmann, Th.; Biehl, M.; Hammer, B.; Verleysen, M. Similarity-Based Clustering, Springer Berlin / Heidelberg, pp.52-69, 2009, Lecture Notes in Computer Science, ⟨10.1007/978-3-642-01805-3_4⟩. ⟨hal-00413154⟩