Feature selection for multi-label classification problems

  • Authors:
  • Gauthier Doquire;Michel Verleysen

  • Affiliations:
  • Université catholique de Louvain, Machine Learning Group - ICTEAM, Louvain-la-Neuve, Belgium;Université catholique de Louvain, Machine Learning Group - ICTEAM, Louvain-la-Neuve, Belgium

  • Venue:
  • IWANN'11 Proceedings of the 11th international conference on Artificial neural networks conference on Advances in computational intelligence - Volume Part I
  • Year:
  • 2011

Quantified Score

Hi-index 0.00

Visualization

Abstract

This paper proposes the use of mutual information for feature selection in multi-label classification, a surprisingly almost not studied problem. A pruned problem transformation method is first applied, transforming the multi-label problem into a single-label one. A greedy feature selection procedure based on multidimensional mutual information is then conducted. Results on three databases clearly demonstrate the interest of the approach which allows one to sharply reduce the dimension of the problem and to enhance the performance of classifiers.