Multi-Label Causal Feature Selection


  • Xingyu Wu University of Science and Technology of China
  • Bingbing Jiang Hangzhou Normal University
  • Kui Yu Hefei University of Technology
  • Huanhuan Chen University of Science and Technology of China
  • Chunyan Miao Nanyang Technological University



Multi-label feature selection has received considerable attentions during the past decade. However, existing algorithms do not attempt to uncover the underlying causal mechanism, and individually solve different types of variable relationships, ignoring the mutual effects between them. Furthermore, these algorithms lack of interpretability, which can only select features for all labels, but cannot explain the correlation between a selected feature and a certain label. To address these problems, in this paper, we theoretically study the causal relationships in multi-label data, and propose a novel Markov blanket based multi-label causal feature selection (MB-MCF) algorithm. MB-MCF mines the causal mechanism of labels and features first, to obtain a complete representation of information about labels. Based on the causal relationships, MB-MCF then selects predictive features and simultaneously distinguishes common features shared by multiple labels and label-specific features owned by single labels. Experiments on real-world data sets validate that MB-MCF could automatically determine the number of selected features and simultaneously achieve the best performance compared with state-of-the-art methods. An experiment in Emotions data set further demonstrates the interpretability of MB-MCF.




How to Cite

Wu, X., Jiang, B., Yu, K., Chen, H., & Miao, C. (2020). Multi-Label Causal Feature Selection. Proceedings of the AAAI Conference on Artificial Intelligence, 34(04), 6430-6437.



AAAI Technical Track: Machine Learning