Proceedings

MCF-Spouse: A Multi-Label Causal Feature Selection Method with Optimal Spouses Discovery

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 5914-5922.

https://doi.org/10.24963/ijcai.2025/658

Abstract

Multi-label causal feature selection has garnered considerable attention for its ability to identify the most informative features while accounting for the causal dependencies between labels and features. However, previous work often overlooks the unique contributions of labels to the target variables in multi-label settings, focusing instead on prioritizing feature variables. Moreover, existing methods typically rely on traditional Markov Blanket (MB) discovery to construct an initial MB, which often fails to explore the most valuable form of spouse variables to feature selection in multi-label scenarios, leading to significant computational overhead due to redundant Conditional Independence (CI) tests required for spouse search. To address these challenges, we propose the Multi-label Causal Feature Selection Method with Optimal Spouses Discovery, MCF-Spouse, which leverages mutual information to quantify the contributions of both labels and features, ensuring the retention of the most informative variables in multi-label settings. Moreover, we systematically analyzes all potential forms of spouse variables to identify the optimal spouse case, significantly reducing the spouse search space and alleviating the time overhead associated with CI tests. Experiments conducted on diverse real-world datasets demonstrate that MCF-Spouse consistently outperforms state-of-the-art methods across multiple metrics, offering a scalable and interpretable solution for multi-label causal feature selection.

  1. [Aliferis et al., 2003] Constantin F Aliferis, Ioannis Tsamardinos, and Alexander Statnikov. Hiton: a novel markov blanket algorithm for optimal variable selection. In AMIA annual symposium proceedings, volume 2003, page 21. American Medical Informatics Association, 2003.
  2. [Aliferis et al., 2010a] Constantin F Aliferis, Alexander Statnikov, Ioannis Tsamardinos, Subramani Mani, and Xenofon D Koutsoukos. Local causal and markov blan- ket induction for causal discovery and feature selection for classification part i: algorithms and empirical evaluation. Journal of Machine Learning Research, 11(1), 2010.
  3. [Aliferis et al., 2010b] Constantin F Aliferis, Alexander Statnikov, Ioannis Tsamardinos, Subramani Mani, and Xenofon D Koutsoukos. Local causal and markov blan- ket induction for causal discovery and feature selection for classification part ii: analysis and extensions. Journal of Machine Learning Research, 11(1), 2010.
  4. [Ben-Gal, 2008] Irad Ben-Gal. Bayesian networks. Encyclo- pedia of statistics in quality and reliability, 2008.
  5. [Cai and Zhu, 2018] Zhiling Cai and William Zhu. Multi- label feature selection via feature manifold learning and sparsity regularization. International journal of machine learning and cybernetics, 9:1321-1334, 2018.
  6. [Chu et al., 2023] Zhixuan Chu, Ruopeng Li, Stephen Rath- bun, and Sheng Li. Continual causal inference with in- cremental observational data. In 2023 IEEE 39th Inter- national Conference on Data Engineering (ICDE), pages 3430-3439. IEEE, 2023.
  7. [Elisseeff and Weston, 2001] André Elisseeff and Jason We- ston. A kernel method for multi-labelled classification. Advances in neural information processing systems, 14, 2001.
  8. [Faraji et al., 2024] Mohammad Faraji, Seyed Amjad Seyedi, Fardin Akhlaghian Tab, and Reza Mahmoodi. Multi-label feature selection with global and local label correlation. Expert Systems with Applications, 246:123198, 2024.
  9. [Gao et al., 2017] Tian Gao, Kshitij Fadnis, and Murray Campbell. Local-to-global bayesian network structure learning. In International Conference on Machine Learn- ing, pages 1193-1202. PMLR, 2017.
  10. [Gonc ¸alves et al., 2013] Eduardo Corrêa Gonc ¸alves, Alexandre Plastino, and Alex A Freitas. A genetic algorithm for optimizing the label ordering in multi-label classifier chains. In 2013 IEEE 25th international confer- ence on tools with artificial intelligence, pages 469-476. IEEE, 2013.
  11. [Gonzalez-Lopez et al., 2020] Jorge Gonzalez-Lopez, Se- bastián Ventura, and Alberto Cano. Distributed multi-label feature selection using individual mutual information mea- sures. Knowledge-Based Systems, 188:105052, 2020.
  12. [Guo et al., 2023] Xianjie Guo, Kui Yu, Lin Liu, Peipei Li, and Jiuyong Li. Adaptive skeleton construction for accu- rate dag learning. IEEE Transactions on Knowledge and Data Engineering, 2023.
  13. [Guo et al., 2024] Xianjie Guo, Kui Yu, Lin Liu, Jiuyong Li, Jiye Liang, Fuyuan Cao, and Xindong Wu. Progressive skeleton learning for effective local-to-global causal struc- ture learning. IEEE Transactions on Knowledge and Data Engineering, 2024.
  14. [Huang et al., 2018] Rui Huang, Weidong Jiang, and Guan- gling Sun. Manifold-based constraint laplacian score for multi-label feature selection. Pattern Recognition Letters, 112:346-352, 2018.
  15. [Jian et al., 2016] Ling Jian, Jundong Li, Kai Shu, and Huan Liu. Multi-label informed feature selection. In IJCAI, vol- ume 16, pages 1627-33, 2016. IJCAI
  16. [Kraskov et al., 2004] Alexander Kraskov, Harald Stögbauer, and Peter Grassberger. Estimating mutual information. Physical Review E-Statistical, Nonlinear, and Soft Matter Physics, 69(6):066138, 2004.
  17. [Lee and Kim, 2015] Jaesung Lee and Dae-Won Kim. Mu- tual information-based multi-label feature selection using interaction information. Expert Systems with Applications, 42(4):2013-2025, 2015.
  18. [Li et al., 2024] Yonghao Li, Liang Hu, and Wanfu Gao. Multi-label feature selection with high-sparse personalized and low-redundancy shared common features. Information Processing & Management, 61(3):103633, 2024.
  19. [Ling et al., 2022a] Zhaolong Ling, Bo Li, Yiwen Zhang, Qingren Wang, Kui Yu, and Xindong Wu. Causal feature selection with efficient spouses discovery. IEEE Transac- tions on Big Data, 9(2):555-568, 2022.
  20. [Ling et al., 2022b] Zhaolong Ling, Ying Li, Yiwen Zhang, Kui Yu, Peng Zhou, Bo Li, and Xindong Wu. A light causal feature selection approach to high-dimensional data. IEEE Transactions on Knowledge and Data Engi- neering, 2022.
  21. [Ling et al., 2024a] Zhaolong Ling, Bo Li, Yiwen Zhang, Peng Zhou, Xingyu Wu, Yuee Huang, Kui Yu, and Xin- dong Wu. Causal discovery using weight-based condi- tional independence test. ACM Transactions on Knowl- edge Discovery from Data, 19(1):1-24, 2024.
  22. [Ling et al., 2024b] Zhaolong Ling, Jingxuan Wu, Yiwen Zhang, Peng Zhou, Xingyu Wu, Kui Yu, and Xindong Wu. Label-aware causal feature selection. IEEE Transactions on Knowledge and Data Engineering, 2024.
  23. [Liu et al., 2020] Jinghua Liu, Yuwen Li, Wei Weng, Jia Zhang, Baihua Chen, and Shunxiang Wu. Feature selec- tion for multi-label learning with streaming label. Neuro- computing, 387:268-278, 2020.
  24. [Ma et al., 2012] Zhigang Ma, Feiping Nie, Yi Yang, Jasper RR Uijlings, and Nicu Sebe. Web image annotation via subspace-sparsity collaborated feature selection. IEEE Transactions on Multimedia, 14(4):1021-1030, 2012.
  25. [Ma et al., 2025] Lin Ma, Liang Hu, Yonghao Li, Weiping Ding, and Wanfu Gao. Mi-mcf: A mutual information- based multilabel causal feature selection. IEEE Trans- actions on Neural Networks and Learning Systems, 36(6):9864-9878, 2025.
  26. [Masegosa and Moral, 2012] Andrés R Masegosa and Serafín Moral. A bayesian stochastic search method for discovering markov boundaries. Knowledge-Based Systems, 35:211-223, 2012.
  27. [Pearl, 2014] Judea Pearl. Probabilistic reasoning in intel- ligent systems: networks of plausible inference. Elsevier, 2014.
  28. [Pellet and Elisseeff, 2008] Jean-Philippe Pellet and André Elisseeff. Using markov blankets for causal structure learning. Journal of Machine Learning Research, 9(7), 2008.
  29. [Pena et al., 2007] Jose M Pena, Roland Nilsson, Johan Björkegren, and Jesper Tegnér. Towards scalable and data efficient learning of markov boundaries. International Journal of Approximate Reasoning, 45(2):211-232, 2007.
  30. [Pereira et al., 2018] Rafael B Pereira, Alexandre Plastino, Bianca Zadrozny, and Luiz HC Merschmann. Categoriz- ing feature selection methods for multi-label classification. Artificial intelligence review, 49:57-78, 2018.
  31. [Read et al., 2008] Jesse Read, Bernhard Pfahringer, and Ge- off Holmes. Multi-label classification using ensembles of pruned sets. In 2008 eighth IEEE international conference on data mining, pages 995-1000. IEEE, 2008.
  32. [Shao et al., 2013] Huan Shao, GuoZheng Li, GuoPing Liu, and YiQin Wang. Symptom selection for multi-label data of inquiry diagnosis in traditional chinese medicine. Sci- ence China Information Sciences, 56:1-13, 2013.
  33. [Spirtes et al., 2001] Peter Spirtes, Clark Glymour, and Richard Scheines. Causation, prediction, and search. MIT press, 2001.
  34. [Statnikov et al., 2013] Alexander Statnikov, Nikita I Lytkin, Jan Lemeire, and Constantin F Aliferis. Algorithms for discovery of multiple markov boundaries. Journal of Ma- chine Learning Research, 14(Feb):499-566, 2013.
  35. [Tsamardinos et al., 2003a] Ioannis Tsamardinos, Con- stantin F Aliferis, and Alexander Statnikov. Time and sample efficient discovery of markov blankets and direct causal relations. In Proceedings of the ninth ACM SIGKDD international conference on Knowledge discovery and data mining, pages 673-678, 2003.
  36. [Tsamardinos et al., 2003b] Ioannis Tsamardinos, Con- stantin F Aliferis, Alexander R Statnikov, and Er Stat- nikov. Algorithms for large scale markov blanket discovery. In FLAIRS, volume 2, pages 376-81, 2003.
  37. [Tsamardinos et al., 2006] Ioannis Tsamardinos, Laura E Brown, and Constantin F Aliferis. The max-min hill- climbing bayesian network structure learning algorithm. Machine learning, 65:31-78, 2006.
  38. [Wu et al., 2020] Xingyu Wu, Bingbing Jiang, Kui Yu, Huanhuan Chen, and Chunyan Miao. Multi-label causal feature selection. In Proceedings of the AAAI conference on artificial intelligence, volume 34, pages 6430-6437, 2020.
  39. [Wu et al., 2022] Xingyu Wu, Bingbing Jiang, Yan Zhong, and Huanhuan Chen. Multi-target markov boundary dis- covery: Theory, algorithm, and application. IEEE Trans- actions on Pattern Analysis and Machine Intelligence, 45(4):4964-4980, 2022.
  40. [Wu et al., 2023a] Xingyu Wu, Bingbing Jiang, Xiangyu Wang, Taiyu Ban, and Huanhuan Chen. Feature selection in the data stream based on incremental markov bound- ary learning. IEEE Transactions on Neural Networks and Learning Systems, 34(10):6740-6754, 2023.
  41. [Wu et al., 2023b] Xingyu Wu, Bingbing Jiang, Tianhao Wu, and Huanhuan Chen. Practical markov boundary learning without strong assumptions. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 37, pages 10388-10398, 2023.
  42. [Xu et al., 2016] Jianhua Xu, Jiali Liu, Jing Yin, and Chengyu Sun. A multi-label feature extraction algo- rithm via maximizing feature variance and feature-label dependence simultaneously. Knowledge-Based Systems, 98:172-184, 2016.
  43. [Yu et al., 2021] Kui Yu, Mingzhu Cai, Xingyu Wu, Lin Liu, and Jiuyong Li. Multilabel feature selection: a local causal structure learning approach. IEEE Transactions on Neural Networks and Learning Systems, 34(6):3044-3057, 2021.
  44. [Zhang and Zhou, 2007] Min-Ling Zhang and Zhi-Hua Zhou. Ml-knn: A lazy learning approach to multi-label learning. Pattern recognition, 40(7):2038-2048, 2007.
  45. [Zhang et al., 2019a] Jia Zhang, Zhiming Luo, Candong Li, Changen Zhou, and Shaozi Li. Manifold regularized dis- criminative feature selection for multi-label learning. Pat- tern Recognition, 95:136-150, 2019.
  46. [Zhang et al., 2019b] Ping Zhang, Guixia Liu, and Wanfu Gao. Distinguishing two types of labels for multi-label feature selection. Pattern recognition, 95:72-82, 2019.
  47. [Zhang et al., 2020] Ping Zhang, Wanfu Gao, Juncheng Hu, and Yonghao Li. Multi-label feature selection based on high-order label correlation assumption. Entropy, 22(7):797, 2020.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards